Demo Data from SIGIR 2026 / reviewers in the wild / expert
Yu Shen 0003
dblp:48/4462-3
· DBLP profile ↗
22ranked-venue papers
4as first author
22since 2021 · last 2025
0000-0001-6503-6504ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Databases, data management, data science and information retrieval · 14 · 2 first-author · 14 since 2021Artificial intelligence and machine learning · 11 · 3 first-author · 11 since 2021Graphics, computer vision, multimedia, augmented reality and games · 2 · 1 first-author · 2 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 2 since 2021Systems, architecture and hardware · 1 · 1 since 2021
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
14 papers |
Graph learning · 39% Optimization for machine learning · 31% Efficient and distributed learning · 27% | |
| Computer architecture, parallel and distributed computing, and storage systems
5 papers |
Cloud and datacenter computing · 42% Hardware accelerators and domain-specific architectures · 14% Interconnection networks and networks-on-chip · 14% | |
| Databases, data mining, and information retrieval
7 papers |
Database system architecture and tuning · 43% Machine learning and data management · 31% Data models and query languages · 16% |
Topics — the 30 heaviest of 51, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Optimization for machine learning
hyperparameter optimization |
3.2 | 6 | 2022 | Hyper-Tune: Towards Efficient Hyper-parameter Tuning at Scale · Proc. VLDB Endow. 2022 TransBO: Hyperparameter Optimization via Two-Phase Transfer Learning · KDD 2022 Transfer Learning based Search Space Design for Hyperparameter Tuning · KDD 2022 |
Machine learning › Graph learning
graph neural network |
2.6 | 5 | 2022 | NAFS: A Simple yet Tough-to-beat Baseline for Graph Representation Learning · ICML 2022 Deep and Flexible Graph Neural Architecture Search · ICML 2022 Grain: Improving Data Efficiency of Graph Neural Networks via Diversified Influence Maximization · Proc. VLDB Endow. 2021 |
Machine learning › Optimization for machine learning › model-based optimization
bayesian optimization |
1.7 | 4 | 2022 | Transfer Learning based Search Space Design for Hyperparameter Tuning · KDD 2022 OpenBox: A Generalized Black-box Optimization Service · KDD 2021 MFES-HB: Efficient Hyperband with Multi-Fidelity Quality Measurements · AAAI 2021 |
Machine learning and data management
automated machine learning |
1.2 | 2 | 2023 | VolcanoML: speeding up end-to-end AutoML via scalable search space decomposition · VLDB J. 2023 VolcanoML: Speeding up End-to-End AutoML via Scalable Search Space Decomposition · Proc. VLDB Endow. 2021 |
Machine learning › Graph learning › graph neural network
graph neural architecture search |
1.1 | 2 | 2022 | PaSca: A Graph Neural Architecture Search System under the Scalable Paradigm · WWW 2022 Deep and Flexible Graph Neural Architecture Search · ICML 2022 |
Machine learning › Efficient and distributed learning › automated machine learning
neural architecture search |
1.0 | 3 | 2023 | ProxyBO: Accelerating Neural Architecture Search via Bayesian Optimization with Zero-Cost Proxies · AAAI 2023 TransBO: Hyperparameter Optimization via Two-Phase Transfer Learning · KDD 2022 Transfer Learning based Search Space Design for Hyperparameter Tuning · KDD 2022 |
Data models and query languages › natural language interface › natural language interface to database
text-to-SQL |
0.9 | 1 | 2025 | SiriusBI: A Comprehensive LLM-powered Solution for Data Analytics in Business Intelligence · Proc. VLDB Endow. 2025 |
Cloud and datacenter computing
cluster resource management and scheduling |
0.9 | 1 | 2025 | A-Tune-Online: Efficient and QoS-Aware Online Configuration Tuning for Dynamic Workloads · ICDE 2025 |
Electronic design automation
design space exploration |
0.9 | 1 | 2025 | Theseus: Exploring Efficient Wafer-Scale Chip Design for Large Language Models · IEEE Trans. Comput. Aided Des. Integr. Circuits Syst. 2025 |
Interconnection networks and networks-on-chip
network-on-chip design |
0.9 | 1 | 2025 | Theseus: Exploring Efficient Wafer-Scale Chip Design for Large Language Models · IEEE Trans. Comput. Aided Des. Integr. Circuits Syst. 2025 |
Cloud and datacenter computing › configuration tuning
online tuning |
0.9 | 1 | 2025 | A-Tune-Online: Efficient and QoS-Aware Online Configuration Tuning for Dynamic Workloads · ICDE 2025 |
Performance modeling and evaluation
workload characterization |
0.9 | 1 | 2025 | A-Tune-Online: Efficient and QoS-Aware Online Configuration Tuning for Dynamic Workloads · ICDE 2025 |
Cloud and datacenter computing › cluster resource management and scheduling
cluster resource management |
0.8 | 2 | 2023 | Towards General and Efficient Online Tuning for Spark · Proc. VLDB Endow. 2023 Hyper-Tune: Towards Efficient Hyper-parameter Tuning at Scale · Proc. VLDB Endow. 2022 |
Machine learning › Efficient and distributed learning › automated machine learning › neural architecture search
zero-cost proxy |
0.7 | 1 | 2023 | ProxyBO: Accelerating Neural Architecture Search via Bayesian Optimization with Zero-Cost Proxies · AAAI 2023 |
Database system architecture and tuning › configuration tuning
automatic parameter tuning |
0.7 | 1 | 2023 | Towards General and Efficient Online Tuning for Spark · Proc. VLDB Endow. 2023 |
Database system architecture and tuning
configuration tuning |
0.7 | 1 | 2023 | Rover: An Online Spark SQL Tuning Service via Generalized Transfer Learning · KDD 2023 |
Database system architecture and tuning › database tuning
online tuning |
0.7 | 1 | 2023 | Towards General and Efficient Online Tuning for Spark · Proc. VLDB Endow. 2023 |
Machine learning › Efficient and distributed learning
automated machine learning |
0.6 | 1 | 2022 | DivBO: Diversity-aware CASH for Ensemble Learning · NeurIPS 2022 |
Machine learning › Optimization for machine learning › hyperparameter optimization
combined algorithm selection and hyperparameter optimization |
0.6 | 1 | 2022 | DivBO: Diversity-aware CASH for Ensemble Learning · NeurIPS 2022 |
Machine learning › Efficient and distributed learning › distributed training
distributed training systems |
0.6 | 1 | 2022 | Hyper-Tune: Towards Efficient Hyper-parameter Tuning at Scale · Proc. VLDB Endow. 2022 |
Machine learning › Kernel, tree and ensemble methods
ensemble learning |
0.6 | 1 | 2022 | DivBO: Diversity-aware CASH for Ensemble Learning · NeurIPS 2022 |
Machine learning › Graph learning
graph representation learning |
0.6 | 1 | 2022 | NAFS: A Simple yet Tough-to-beat Baseline for Graph Representation Learning · ICML 2022 |
Machine learning › Graph learning › graph neural network
message passing |
0.6 | 1 | 2022 | Deep and Flexible Graph Neural Architecture Search · ICML 2022 |
Machine learning › Graph learning
network embedding |
0.6 | 1 | 2022 | NAFS: A Simple yet Tough-to-beat Baseline for Graph Representation Learning · ICML 2022 |
Machine learning › Graph learning › graph neural network › deep graph neural network
over-smoothing |
0.6 | 1 | 2022 | NAFS: A Simple yet Tough-to-beat Baseline for Graph Representation Learning · ICML 2022 |
Machine learning › Graph learning › graph neural network
scalable graph neural network |
0.6 | 1 | 2022 | PaSca: A Graph Neural Architecture Search System under the Scalable Paradigm · WWW 2022 |
Machine learning › Efficient and distributed learning › automated machine learning › neural architecture search
search space design |
0.6 | 1 | 2022 | Transfer Learning based Search Space Design for Hyperparameter Tuning · KDD 2022 |
Medical and health informatics › clinical diagnosis
disease classification |
0.6 | 1 | 2022 | AutoDC: an automatic machine learning framework for disease classification · Bioinform. 2022 |
Machine learning › Efficient and distributed learning
active learning |
0.5 | 1 | 2021 | ALG: Fast and Accurate Active Learning Framework for Graph Convolutional Networks · SIGMOD Conference 2021 |
Machine learning › Optimization for machine learning
black-box optimization |
0.5 | 1 | 2021 | OpenBox: A Generalized Black-box Optimization Service · KDD 2021 |
Methods — techniques the papers use, named apart from their topics
bayesian optimization · 5.2transfer learning · 2.3multi-fidelity optimization · 1.6meta-learning · 1.3approximate gradient descent · 1.3adaptive sub-space generation · 1.3feature smoothing · 1.1ensemble · 1.1warm-start · 0.9semantic intermediate representation · 0.9multi-fidelity bayesian optimization · 0.9lower confidence bound · 0.9large language model · 0.9fine-tuning · 0.9zero-cost proxies · 0.7weighted acquisition function · 0.6two-stage feature selection · 0.6multi-armed bandit · 0.6
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2025 | A-Tune-Online: Efficient and QoS-Aware Online Configuration Tuning for Dynamic WorkloadsabstractAutomatic configuration tuning of online services with dynamic workloads has attracted increasing interest. Effective online tuning ensures configurations adapt to workload changes over time to maintain optimal online service performance. To be practical, online tuning must satisfy the dynamicity, efficiency, and Quality of Service (QoS) requirements. However, existing online tuning approaches fail to meet these requirements due to the inability to eliminate negative effects from historical observations. In this paper, we propose A-Tune-Online, an online configuration tuning system that tackles dynamic workloads, delivering superior tuning efficiency, and QoS guarantee simultaneously to a wide range of online scenarios. We identify that restarting the optimization based on explicit workload shift detection is necessary and critical to eliminate negative historical observations. First, to invoke optimization restarts appropriately, we design a multi-stage multi-indicator detection strategy based on heuristic rules and configuration replays. Then, to avoid initial efficiency drop after re-optimization, A-Tune-Online utilizes a similarity-based dual warm start scheme that transfers knowledge from similar historical workloads effectively. Finally, to prevent transient performance degradation from violating QoS guarantee after optimization restart, we leverage lower confidence bound to construct a safety region where each configuration is expected to perform better than the QoS requirement. Empirical study on five tuning scenarios showcases the superiority of A-Tune-Online compared with state-of-art tuning systems. A-Tune-Online achieves an average speedup of 2.90x and 1.72x compared with OnlineTune and DDPG+, respectively. We provide a version of our system in https://github.com/PKU-DAIR/A-Tune-Online. Yu Shen 0003, Beicheng Xu, Yupeng Lu, Huaijun Jiang, Zhipeng Xie, Senbo Fu, Nan Zhang 0004, Yuxin Ren 0001, Ning Jia 0004, Xinwei Hu, Bin Cui 0001 |
ICDE | 1 |
| 2025 | SiriusBI: A Comprehensive LLM-powered Solution for Data Analytics in Business IntelligenceabstractWith the proliferation of Large Language Models (LLMs) in Business Intelligence (BI), existing solutions face critical challenges in industrial deployments: functionality deficiencies from legacy systems failing to meet evolving LLM-era user demands, interaction limitations from single-round SQL generation paradigms inadequate for multi-round clarification, and cost for domain adaptation arising from cross-domain methods migration. We present SiriusBI, a practical LLM-powered BI system addressing the challenges of industrial deployments through three key innovations: (a) An end-to-end architecture integrating multi-module coordination to overcome functionality gaps in legacy systems; (b) A multi-round dialogue with querying mechanism, consisting of semantic completion, knowledge-guided clarification, and proactive querying processes, to resolve interaction constraints in SQL generation; (c) A data-conditioned SQL generation method selection strategy that supports both an efficient one-step Fine-Tuning approach and a two-step method leveraging Semantic Intermediate Representation for low-cost cross-domain applications. Experiments on both real-world datasets and public benchmarks demonstrate the effectiveness of SiriusBI. User studies further confirm that SiriusBI enhances both productivity and user experience. As an independent service on Tencent's data platform, SiriusBI is deployed across finance, advertising, and cloud sectors, serving dozens of enterprise clients. It achieves over 93% accuracy in SQL generation and reduces data analysts' query time from minutes to seconds in real-world applications. Jie Jiang 0015, Haining Xie, Yu Shen 0003, Meng Lei, Yang Li 0106, Chunyou Li, Danqing Huang, Yinjun Wu, Wentao Zhang 0001, Bin Cui 0001, Peng Chen 0021 |
Proc. VLDB Endow. | 4 |
| 2025 | Theseus: Exploring Efficient Wafer-Scale Chip Design for Large Language ModelsabstractThe emergence of the large language model (LLM) poses an exponential growth of demand for computation throughput, memory capacity, and communication bandwidth. Such a demand growth has significantly surpassed the improvement of corresponding chip designs. With the advancement of fabrication and integration technologies, designers have been developing Wafer-Scale Chips (WSCs) to scale up and exploit the limits of computation density, memory capacity, and communication bandwidth at the level of a single chip. Existing solutions have demonstrated the significant advantages of WSCs over traditional designs, showing potential to effectively support LLM workloads. Despite the benefits, exploring the early-stage design space of WSCs for LLMs is a crucial yet challenging task due to the enormous and complicated design space, time-consuming evaluation methods, and inefficient exploration strategies. To address these challenges, we propose Theseus, an efficient WSC design space exploration framework for LLMs. We construct the design space of WSCs with various constraints considering the unique characteristics of WSCs. We propose efficient evaluation methodologies for large-scale NoC-based WSCs and introduce multi-fidelity Bayesian optimization to efficiently explore the design space. Evaluation results demonstrate the efficiency of Theseus that the searched Pareto optimal results outperform GPU cluster and existing WSC designs by up to 62.8%/73.7% in performance (with the same or lower power) and 38.6%/42.4% in power consumption (with the same or higher performance) for LLM training, while improving up to 23.2× and 15.7× for the performance and power of inference tasks. Furthermore, we conduct case studies to address the design tradeoffs in WSCs and provide insights to facilitate WSC designs for LLMs. Jingchen Zhu, Chenhao Xue, Chen Zhang 0001, Yu Shen 0003, Zekang Cheng, Yibo Lin, Wei Hu 0003, Bin Cui 0001, Runsheng Wang, Yun Liang 0001, Guangyu Sun 0003 |
IEEE Trans. Comput. Aided Des. Integr. Circuits Syst. | 6 |
| 2024 | OpenBox: A Python Toolkit for Generalized Black-box OptimizationabstractBlack-box optimization (BBO) has a broad range of applications, including automatic machine learning, experimental design, and database knob tuning. However, users still face challenges when applying BBO methods to their problems at hand with existing software packages in terms of applicability, performance, and efficiency. This paper presents OpenBox, an open-source BBO toolkit with improved usability. It implements user-friendly interfaces and visualization for users to define and manage their tasks. The modular design behind OpenBox facilitates its flexible deployment in existing systems. Experimental results demonstrate the effectiveness and efficiency of OpenBox over existing systems. The source code of OpenBox is available at https://github.com/PKU-DAIR/open-box. Huaijun Jiang, Yu Shen 0003, Yang Li 0106, Beicheng Xu, Sixian Du, Wentao Zhang 0001, Ce Zhang 0001, Bin Cui 0001 |
J. Mach. Learn. Res. | 2 |
| 2023 | ProxyBO: Accelerating Neural Architecture Search via Bayesian Optimization with Zero-Cost ProxiesabstractDesigning neural architectures requires immense manual efforts. This has promoted the development of neural architecture search (NAS) to automate the design. While previous NAS methods achieve promising results but run slowly, zero-cost proxies run extremely fast but are less promising. Therefore, it’s of great potential to accelerate NAS via those zero-cost proxies. The existing method has two limitations, which are unforeseeable reliability and one-shot usage. To address the limitations, we present ProxyBO, an efficient Bayesian optimization (BO) framework that utilizes the zero-cost proxies to accelerate neural architecture search. We apply the generalization ability measurement to estimate the fitness of proxies on the task during each iteration and design a novel acquisition function to combine BO with zero-cost proxies based on their dynamic influence. Extensive empirical studies show that ProxyBO consistently outperforms competitive baselines on five tasks from three public benchmarks. Concretely, ProxyBO achieves up to 5.41× and 3.86× speedups over the state-of-the-art approaches REA and BRP-NAS. Yu Shen 0003, Yang Li 0106, Wentao Zhang 0001, Sen Yang 0004, Ji Liu 0002, Bin Cui 0001 |
AAAI | 1 |
| 2023 | Rover: An Online Spark SQL Tuning Service via Generalized Transfer LearningabstractDistributed data analytic engines like Spark are common choices to process massive data in industry. However, the performance of Spark SQL highly depends on the choice of configurations, where the optimal ones vary with the executed workloads. Among various alternatives for Spark SQL tuning, Bayesian optimization (BO) is a popular framework that finds near-optimal configurations given sufficient budget, but it suffers from the re-optimization issue and is not practical in real production. When applying transfer learning to accelerate the tuning process, we notice two domain-specific challenges: 1) most previous work focus on transferring tuning history, while expert knowledge from Spark engineers is of great potential to improve the tuning performance but is not well studied so far; 2) history tasks should be carefully utilized, where using dissimilar ones lead to a deteriorated performance in production. Yu Shen 0003, Xinyuyang Ren, Yupeng Lu, Huaijun Jiang, Huanyong Xu, Yang Li 0106, Wentao Zhang 0001, Bin Cui 0001 |
KDD | 1 |
| 2023 | Towards General and Efficient Online Tuning for SparkabstractThe distributed data analytic system - Spark is a common choice for processing massive volumes of heterogeneous data, while it is challenging to tune its parameters to achieve high performance. Recent studies try to employ auto-tuning techniques to solve this problem but suffer from three issues: limited functionality, high overhead, and inefficient search. In this paper, we present a general and efficient Spark tuning framework that can deal with the three issues simultaneously. First, we introduce a generalized tuning formulation, which can support multiple tuning goals and constraints conveniently, and a Bayesian optimization (BO) based solution to solve this generalized optimization problem. Second, to avoid high overhead from additional offline evaluations in existing methods, we propose to tune parameters along with the actual periodic executions of each job (i.e., online evaluations). To ensure safety during online job executions, we design a safe configuration acquisition method that models the safe region. Finally, three innovative techniques are leveraged to further accelerate the search process: adaptive sub-space generation, approximate gradient descent, and meta-learning method. We have implemented this framework as an independent cloud service, and applied it to the data platform in Tencent. The empirical results on both public benchmarks and large-scale production tasks demonstrate its superiority in terms of practicality, generality, and efficiency. Notably, this service saves an average of 57.00% memory cost and 34.93% CPU cost on 25K in-production tasks within 20 iterations, respectively. Yang Li 0106, Huaijun Jiang, Yu Shen 0003, Yide Fang, Danqing Huang, Xinyi Zhang 0002, Wentao Zhang 0001, Ce Zhang 0001, Peng Chen 0021, Bin Cui 0001 |
Proc. VLDB Endow. | 3 |
| 2023 | VolcanoML: speeding up end-to-end AutoML via scalable search space decomposition
Yang Li 0106, Yu Shen 0003, Wentao Zhang 0001, Ce Zhang 0001, Bin Cui 0001 |
VLDB J. | 2 |
| 2022 | Deep and Flexible Graph Neural Architecture SearchabstractGraph neural networks (GNNs) have been intensively applied to various graph-based applications. Despite their success, designing good GNN architectures is non-trivial, which heavily relies on lots of human efforts and domain knowledge. Although several attempts have been made in graph neural architecture search, they suffer from the following limitations: 1) fixed pipeline pattern of propagation (P) and (T) transformation operations; 2) restricted pipeline depth of GNN architectures. This paper proposes DFG-NAS, a novel method that searches for deep and flexible GNN architectures. Unlike most existing methods that focus on micro-architecture, DFG-NAS highlights another level of design: the search for macro-architectures of how atomic P and T are integrated and organized into a GNN. Concretely, DFG-NAS proposes a novel-designed search space for the P-T permutations and combinations based on the message-passing dis-aggregation, and defines various mutation strategies and employs the evolutionary algorithm to conduct an efficient and effective search. Empirical studies on four benchmark datasets demonstrate that DFG-NAS could find more powerful architectures than state-of-the-art manual designs and meanwhile are more efficient than the current graph neural architecture search approaches. Wentao Zhang 0001, Zheyu Lin, Yu Shen 0003, Yang Li 0106, Zhi Yang 0001, Bin Cui 0001 |
ICML | 3 |
| 2022 | NAFS: A Simple yet Tough-to-beat Baseline for Graph Representation LearningabstractRecently, graph neural networks (GNNs) have shown prominent performance in graph representation learning by leveraging knowledge from both graph structure and node features. However, most of them have two major limitations. First, GNNs can learn higher-order structural information by stacking more layers but can not deal with large depth due to the over-smoothing issue. Second, it is not easy to apply these methods on large graphs due to the expensive computation cost and high memory usage. In this paper, we present node-adaptive feature smoothing (NAFS), a simple non-parametric method that constructs node representations without parameter learning. NAFS first extracts the features of each node with its neighbors of different hops by feature smoothing, and then adaptively combines the smoothed features. Besides, the constructed node representation can further be enhanced by the ensemble of smoothed features extracted via different smoothing strategies. We conduct experiments on four benchmark datasets on two different application scenarios: node clustering and link prediction. Remarkably, NAFS with feature ensemble outperforms the state-of-the-art GNNs on these tasks and mitigates the aforementioned two limitations of most learning-based GNN counterparts. Wentao Zhang 0001, Zeang Sheng, Yang Li 0106, Yu Shen 0003, Zhi Yang 0001, Bin Cui 0001 |
ICML | 5 |
| 2022 | Transfer Learning based Search Space Design for Hyperparameter TuningabstractThe tuning of hyperparameters becomes increasingly important as machine learning (ML) models have been extensively applied in data mining applications. Among various approaches, Bayesian optimization (BO) is a successful methodology to tune hyperparameters automatically. While traditional methods optimize each tuning task in isolation, there has been recent interest in speeding up BO by transferring knowledge across previous tasks. In this work, we introduce an automatic method to design the BO search space with the aid of tuning history from past tasks. This simple yet effective approach can be used to endow many existing BO methods with transfer learning capabilities. In addition, it enjoys the three advantages: universality, generality, and safeness. The extensive experiments show that our approach considerably boosts BO by designing a promising and compact search space instead of using the entire space, and outperforms the state-of-the-arts on a wide range of benchmarks, including machine learning and deep learning tuning tasks, and neural architecture search. Yang Li 0106, Yu Shen 0003, Huaijun Jiang, Tianyi Bai, Wentao Zhang 0001, Ce Zhang 0001, Bin Cui 0001 |
KDD | 2 |
| 2022 | TransBO: Hyperparameter Optimization via Two-Phase Transfer LearningabstractWith the extensive applications of machine learning models, automatic hyperparameter optimization (HPO) has become increasingly important. Motivated by the tuning behaviors of human experts, it is intuitive to leverage auxiliary knowledge from past HPO tasks to accelerate the current HPO task. In this paper, we propose TransBO, a novel two-phase transfer learning framework for HPO, which can deal with the complementary nature among source tasks and dynamics during knowledge aggregation issues simultaneously. This framework extracts and aggregates source and target knowledge jointly and adaptively, where the weights can be learned in a principled manner. The extensive experiments, including static and dynamic transfer learning settings and neural architecture search, demonstrate the superiority of TransBO over the state-of-the-arts. Yang Li 0106, Yu Shen 0003, Huaijun Jiang, Wentao Zhang 0001, Zhi Yang 0001, Ce Zhang 0001, Bin Cui 0001 |
KDD | 2 |
| 2022 | DivBO: Diversity-aware CASH for Ensemble LearningabstractThe Combined Algorithm Selection and Hyperparameters optimization (CASH) problem is one of the fundamental problems in Automated Machine Learning (AutoML). Motivated by the success of ensemble learning, recent AutoML systems build post-hoc ensembles to output the final predictions instead of using the best single learner. However, while most CASH methods focus on searching for a single learner with the best performance, they neglect the diversity among base learners (i.e., they may suggest similar configurations to previously evaluated ones), which is also a crucial consideration when building an ensemble. To tackle this issue and further enhance the ensemble performance, we propose DivBO, a diversity-aware framework to inject explicit search of diversity into the CASH problems. In the framework, we propose to use a diversity surrogate to predict the pair-wise diversity of two unseen configurations. Furthermore, we introduce a temporary pool and a weighted acquisition function to guide the search of both performance and diversity based on Bayesian optimization. Empirical results on 15 public datasets show that DivBO achieves the best average ranks (1.82 and 1.73) on both validation and test errors among 10 compared methods, including post-hoc designs in recent AutoML systems and state-of-the-art baselines for ensemble learning on CASH problems. Yu Shen 0003, Yupeng Lu, Yang Li 0106, Yaofeng Tu, Wentao Zhang 0001, Bin Cui 0001 |
NeurIPS | 1 |
| 2022 | PaSca: A Graph Neural Architecture Search System under the Scalable ParadigmabstractGraph neural networks (GNNs) have achieved state-of-the-art performance in various graph-based tasks. However, as mainstream GNNs are designed based on the neural message passing mechanism, they do not scale well to data size and message passing steps. Although there has been an emerging interest in the design of scalable GNNs, current researches focus on specific GNN design, rather than the general design space, limiting the discovery of potential scalable GNN models. This paper proposes PaSca, a new paradigm and system that offers a principled approach to systemically construct and explore the design space for scalable GNNs, rather than studying individual designs. Through deconstructing the message passing mechanism, PaSca presents a novel Scalable Graph Neural Architecture Paradigm (SGAP), together with a general architecture design space consisting of 150k different designs. Following the paradigm, we implement an auto-search engine that can automatically search well-performing and scalable GNN architectures to balance the trade-off between multiple criteria (e.g., accuracy and efficiency) via multi-objective optimization. Empirical studies on ten benchmark datasets demonstrate that the representative instances (i.e., PaSca-V1, V2, and V3) discovered by our system achieve consistent performance among competitive baselines. Concretely, PaSca-V3 outperforms the state-of-the-art GNN method JK-Net by 0.4% in terms of predictive accuracy on our large industry dataset while achieving up to 28.3 × training speedups. Wentao Zhang 0001, Yu Shen 0003, Zheyu Lin, Yang Li 0106, Xiaosen Li, Wen Ouyang, Yangyu Tao, Zhi Yang 0001, Bin Cui 0001 |
WWW | 2 |
| 2022 | AutoDC: an automatic machine learning framework for disease classificationabstractMOTIVATION: The emergence of next-generation sequencing techniques opens up tremendous opportunities for researchers to uncover the basic mechanisms of disease at the molecular level. Recently, automatic machine learning (AutoML) frameworks have been employed for genomic and epigenomic data analysis. However, to analyze those high-dimensional data, existing AutoML frameworks suffer from the following issues: (i) they could not effectively filter out the redundant features from the original data, and (ii) they usually obey the rule of feature engineering first and algorithm hyper-parameter tuning later to build the machine learning pipeline, which could lead to sub-optimal outcomes. Thus, it is an urgent need to design a new AutoML framework for high-dimensional omics data analysis. RESULTS: We introduce a new method: AutoDC, a tailored AutoML framework, for different disease classification based on gene expression data. AutoDC designs two novel optimization strategies to improve the performance. One is that AutoDC designs a novel two-stage feature selection method to select the features with high gene contribution scores. The other is that AutoDC proposes a novel optimization method, based on a two-layer Multi-Armed Bandit framework, to jointly optimize the feature engineering, algorithm selection and algorithm hyper-parameter tuning. We apply our framework to two public gene expression datasets. Compared with three state-of-the-art AutoML frameworks, AutoDC could effectively classify diseases with higher predictive accuracy. AVAILABILITY AND IMPLEMENTATION: The data and codes of AutoDC are available at https://github.com/dingdian110/AutoDC. The data underlying this article are available in the article and in its online supplementary material. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online. Yang Li 0106, Yu Shen 0003, Wentao Zhang 0001, Bin Cui 0001 |
Bioinform. | 3 |
| 2022 | Hyper-Tune: Towards Efficient Hyper-parameter Tuning at ScaleabstractThe ever-growing demand and complexity of machine learning are putting pressure on hyper-parameter tuning systems: while the evaluation cost of models continues to increase, the scalability of state-of-the-arts starts to become a crucial bottleneck. In this paper, inspired by our experience when deploying hyper-parameter tuning in a real-world application in production and the limitations of existing systems, we propose Hyper-Tune, an efficient and robust distributed hyper-parameter tuning framework. Compared with existing systems, Hyper-Tune highlights multiple system optimizations, including (1) automatic resource allocation, (2) asynchronous scheduling, and (3) multi-fidelity optimizer. We conduct extensive evaluations on benchmark datasets and a large-scale real-world dataset in production. Empirically, with the aid of these optimizations, Hyper-Tune outperforms competitive hyper-parameter tuning systems on a wide range of scenarios, including XGBoost, CNN, RNN, and some architectural hyper-parameters for neural networks. Compared with the state-of-the-art BOHB and A-BOHB, Hyper-Tune achieves up to 11.2X and 5.1X speedups, respectively. Yang Li 0106, Yu Shen 0003, Huaijun Jiang, Wentao Zhang 0001, Ji Liu 0002, Ce Zhang 0001, Bin Cui 0001 |
Proc. VLDB Endow. | 2 |
| 2021 | MFES-HB: Efficient Hyperband with Multi-Fidelity Quality MeasurementsabstractHyperparameter optimization (HPO) is a fundamental problem in automatic machine learning (AutoML). However, due to the expensive evaluation cost of models (e.g., training deep learning models or training models on large datasets), vanilla Bayesian optimization (BO) is typically computationally infeasible. To alleviate this issue, Hyperband (HB) utilizes the early stopping mechanism to speed up configuration evaluations by terminating those badly-performing configurations in advance. This leads to two kinds of quality measurements: (1) many low-fidelity measurements for configurations that get early-stopped, and (2) few high-fidelity measurements for configurations that are evaluated without being early stopped. The state-of-the-art HB-style method, BOHB, aims to combine the benefits of both BO and HB. Instead of sampling configurations randomly in HB, BOHB samples configurations based on a BO surrogate model, which is constructed with the high-fidelity measurements only. However, the scarcity of high-fidelity measurements greatly hampers the efficiency of BO to guide the configuration search. In this paper, we present MFES-HB, an efficient Hyperband method that is capable of utilizing both the high-fidelity and low-fidelity measurements to accelerate the convergence of HPO tasks. Designing MFES-HB is not trivial as the low-fidelity measurements can be biased yet informative to guide the configuration search. Thus we propose to build a Multi-Fidelity Ensemble Surrogate (MFES) based on the generalized Product of Experts framework, which can integrate useful information from multi-fidelity measurements effectively. The empirical studies on the real-world AutoML tasks demonstrate that MFES-HB can achieve 3.3-8.9x speedups over the state-of-the-art approach --- BOHB. Yang Li 0106, Yu Shen 0003, Jiawei Jiang 0001, Jinyang Gao, Ce Zhang 0001, Bin Cui 0001 |
AAAI | 2 |
| 2021 | OpenBox: A Generalized Black-box Optimization ServiceabstractBlack-box optimization (BBO) has a broad range of applications, including automatic machine learning, engineering, physics, and experimental design. However, it remains a challenge for users to apply BBO methods to their problems at hand with existing software packages, in terms of applicability, performance, and efficiency. In this paper, we build OpenBox, an open-source and general-purpose BBO service with improved usability. The modular design behind OpenBox also facilitates flexible abstraction and optimization of basic BBO components that are common in other existing systems. OpenBox is distributed, fault-tolerant, and scalable. To improve efficiency, OpenBox further utilizes "algorithm agnostic" parallelization and transfer learning. Our experimental results demonstrate the effectiveness and efficiency of OpenBox compared to existing systems. Yang Li 0106, Yu Shen 0003, Wentao Zhang 0001, Yuanwei Chen, Huaijun Jiang, Mingchao Liu 0001, Jiawei Jiang 0001, Jinyang Gao, Wentao Wu 0001, Zhi Yang 0001, Ce Zhang 0001, Bin Cui 0001 |
KDD | 2 |
| 2021 | ROD: Reception-aware Online Distillation for Sparse GraphsabstractGraph neural networks (GNNs) have been widely used in many graph-based tasks such as node classification, link prediction, and node clustering. However, GNNs gain their performance benefits mainly from performing the feature propagation and smoothing across the edges of the graph, thus requiring sufficient connectivity and label information for effective propagation. Unfortunately, many real-world networks are sparse in terms of both edges and labels, leading to sub-optimal performance of GNNs. Recent interest in this sparse problem has focused on the self-training approach, which expands supervised signals with pseudo labels. Nevertheless, the self-training approach inherently cannot realize the full potential of refining the learning performance on sparse graphs due to the unsatisfactory quality and quantity of pseudo labels. Wentao Zhang 0001, Yuezihan Jiang, Yang Li 0106, Zeang Sheng, Yu Shen 0003, Xupeng Miao, Liang Wang 0001, Zhi Yang 0001, Bin Cui 0001 |
KDD | 5 |
| 2021 | ALG: Fast and Accurate Active Learning Framework for Graph Convolutional NetworksabstractGraph Convolutional Networks (GCNs) have become state-of-the-art methods in many supervised and semi-supervised graph representation learning scenarios. In order to achieve satisfactory performance, GCNs require a sufficient amount of labeled data. However, in real-world scenarios, labeled data is often expensive to obtain. Therefore, we propose ALG, a novel Active Learning framework for GCNs, which employs domain-specific intelligence to achieve much higher performance and efficiency compared to the generic AL frameworks. First, by decoupling GCN models, ALG serves as an effective and efficient AL framework for measuring and combining node representativeness and informativeness. Second, by exploiting the characteristic of the reception field in GCNs, ALG considers both the importance and correlation of nodes by proposing a new node selection metric that maximizes the effective reception field (ERF). We prove that this ERF maximization problem is NP-hard and provide an efficient algorithm accompanied with a provable approximation guarantee. The empirical studies on four public datasets demonstrate that ALG can significantly improve both the performance and efficiency of active learning for GCNs. Wentao Zhang 0001, Yu Shen 0003, Yang Li 0106, Lei Chen 0002, Zhi Yang 0001, Bin Cui 0001 |
SIGMOD Conference | 2 |
| 2021 | VolcanoML: Speeding up End-to-End AutoML via Scalable Search Space DecompositionabstractEnd-to-end AutoML has attracted intensive interests from both academia and industry, which automatically searches for ML pipelines in a space induced by feature engineering, algorithm/model selection, and hyper-parameter tuning. Existing AutoML systems, however, suffer from scalability issues when applying to application domains with large, high-dimensional search spaces. We present VOLCANOML, a scalable and extensible framework that facilitates systematic exploration of large AutoML search spaces. VOLCANOML introduces and implements basic building blocks that decompose a large search space into smaller ones, and allows users to utilize these building blocks to compose an execution plan for the AutoML problem at hand. VOLCANOML further supports a Volcano-style execution model - akin to the one supported by modern database systems - to execute the plan constructed. Our evaluation demonstrates that, not only does VOLCANOML raise the level of expressiveness for search space decomposition in AutoML, it also leads to actual findings of decomposition strategies that are significantly more efficient than the ones employed by state-of-the-art AutoML systems such as auto-sklearn. Yang Li 0106, Yu Shen 0003, Wentao Zhang 0001, Jiawei Jiang 0001, Yaliang Li, Bolin Ding, Jingren Zhou 0001, Zhi Yang 0001, Wentao Wu 0001, Ce Zhang 0001, Bin Cui 0001 |
Proc. VLDB Endow. | 2 |
| 2021 | Grain: Improving Data Efficiency of Graph Neural Networks via Diversified Influence MaximizationabstractData selection methods, such as active learning and core-set selection, are useful tools for improving the data efficiency of deep learning models on large-scale datasets. However, recent deep learning models have moved forward from independent and identically distributed data to graph-structured data, such as social networks, e-commerce user-item graphs, and knowledge graphs. This evolution has led to the emergence of Graph Neural Networks (GNNs) that go beyond the models existing data selection methods are designed for. Therefore, we present GRAIN, an efficient framework that opens up a new perspective through connecting data selection in GNNs with social influence maximization. By exploiting the common patterns of GNNs, GRAIN introduces a novel feature propagation concept, a diversified influence maximization objective with novel influence and diversity functions, and a greedy algorithm with an approximation guarantee into a unified framework. Empirical studies on public datasets demonstrate that GRAIN significantly improves both the performance and efficiency of data selection (including active learning and core-set selection) for GNNs. To the best of our knowledge, this is the first attempt to bridge two largely parallel threads of research, data selection, and social influence maximization, in the setting of GNNs, paving new ways for improving data efficiency. Wentao Zhang 0001, Zhi Yang 0001, Yexin Wang, Yu Shen 0003, Yang Li 0106, Liang Wang 0001, Bin Cui 0001 |
Proc. VLDB Endow. | 4 |