HKUST DB4AI Group

About Us

DB4AI (Database for Artificial Intelligence) group is dedicated to investigating data management techniques to optimize AI models and systems. Our research is motivated by new technologies and applications. We are particularly interested in the topics of large-scale graph learning, compilation optimization of DNN operators, efficient streaming learning, as well as other theoretical and engineered problems in AI. We are part of the Department of Computer Science and Engineering at the The Hong Kong University of Science and Technology.

Group Members

Faculty

Lei Chen

Professor

Collaborators

Graduate Students

Yiming Li

Ph.D Student

Jingzhi Fang

Ph.D Student

Jia Li

Ph.D Student

Xin Zhang

Ph.D Student

Qiqi Zhou

Ph.D Student

Mengxuan Li

Ph.D Student

Shihong Gao

Ph.D Student

Research Topics

Scalable GNN Systems

Graph Sampling
Distributed Training
Knowledge Distillation

Compilation Optimization of DNNs

High-Performance Tensor Program Reuse
Optimizing DNN Computation Graph via Graph Substitutions

Learning over Data Streams

Efficient Continual Learning
Scalable Temporal Graph Neural Networks

Publications

Qiqi Zhou, Yanyan Shen, and Lei Chen. "Narrow the lnput Mismatch in Deep Graph Neural Network Distillation.", KDD 2023.
Xin Zhang, Yanyan Shen, Yingxia Shao, and Lei Chen. "DUCATI: A Dual-Cache Training System for Graph Neural Networks on Giant Graphs with the GPU.", SIGMOD 2023.
Jia Li, Yanyan Shen, Lei Chen, and Charles Wang Wai Ng, "SSIN: Self-Supervised Learning for Rainfall Spatial Interpolation.", SIGMOD 2023.
Yiming Li, Yanyan Shen, Lei Chen, and Mingxuan Yuan. "Orca: Scalable Temporal Graph Neural Network Training with Theoretical Guarantees.", SIGMOD 2023.
Yiming Li, Yanyan Shen, Lei Chen, and Mingxuan Yuan. "Zebra: When Temporal Graph Neural Networks Meet Temporal Personalized PageRank.", VLDB 2023.
Jia Li, Yanyan Shen, Lei Chen, and Charles Wang Wai Ng. "Rainfall Spatial Interpolation with Graph Neural Networks.", DASFAA 2023.
Xin Zhang, Yanyan Shen, and Lei Chen. "Feature-Oriented Sampling for Fast and Scalable GNN Training.", ICDM 2022.
Yiming Li, Yanyan Shen, and Lei Chen. "Camel: Managing Data for Efficient Stream Learning.", SIGMOD 2022.
Jingzhi Fang, Yanyan Shen, Yue Wang, and Lei Chen. "ETO: Accelerating Optimization of DNN Operators by High-Performance Tensor Program Reuse.", VLDB 2022.
Yiming Li, Yanyan Shen, and Lei Chen. "Palette: Towards Multi-source Model Selection and Ensemble for Reuse.", ICDE 2021.
Jia Li, Shimin Di, Yanyan Shen, and Lei Chen. "FluxEV: A Fast and Effective Unsupervised Framework for Time-Series Anomaly Detection.", WSDM 2021.
Jingzhi Fang, Yanyan Shen, Yue Wang, and Lei Chen. "Optimizing DNN Computation Graph Using Graph Substitutions.", VLDB 2020.

HKUST DB4AI Group

About Us

Group Members

Faculty

Lei Chen

Collaborators

Yanyan Shen

Yingxia Shao

Yue Wang

Graduate Students

Yiming Li

Jingzhi Fang

Jia Li

Xin Zhang

Qiqi Zhou

Mengxuan Li

Shihong Gao

Research Topics

Publications