PDS: An I/O-Efficient Scaling Scheme for Parity Declustered Data Layout. - Zhipeng Li 005; Yinlong Xu; Yongkun Li 001; Chengjin Tian; Youhui Bai (ICPP, 2017)

Gradient Compression Supercharged High-Performance Data Parallel DNN Training. - Youhui Bai; Cheng Li 001; Quan Zhou; Jun Yi; Ping Gong 009; Feng Yan 001; Ruichuan Chen; Yinlong Xu (SOSP, 2021)

Efficient Data Loader for Fast Sampling-Based GNN Training on Large Graphs. - Youhui Bai; Cheng Li 001; Zhiqi Lin; Yufei Wu 011; Youshan Miao; Yunxin Liu 001; Yinlong Xu (IEEE Trans. Parallel Distributed Syst., 2021)

MPress: Democratizing Billion-Scale Model Training on Multi-GPU Servers via Memory-Saving Inter-Operator Parallelism. - Quan Zhou; Haiquan Wang; Xiaoyan Yu; Cheng Li 001; Youhui Bai; Feng Yan 001; Yinlong Xu (HPCA, 2023)

A Survey on Auto-Parallelism of Large-Scale Deep Learning Training. - Peng Liang 017; Yu Tang; Xiaoda Zhang; Youhui Bai; Teng Su; Zhiquan Lai; Linbo Qiao; Dongsheng Li 001 (IEEE Trans. Parallel Distributed Syst., 2023)

XL3M: A Training-free Framework for LLM Length Extension Based on Segment-wise Inference. - Shengnan Wang; Youhui Bai; Lin Zhang; Pingyi Zhou; Shixiong Zhao; Gong Zhang 001; Sen Wang 004; Renhai Chen; Hua Xu; Hongwei Sun (CoRR, 2024)

BigMac: A Communication-Efficient Mixture-of-Experts Model Structure for Fast Training and Inference. - Zewen Jin; Shengnan Wang; Jiaan Zhu; Hongrui Zhan; Youhui Bai; Lin Zhang; Zhenyu Ming; Cheng Li (AAAI, 2025)

HATA: Trainable and Hardware-Efficient Hash-Aware Top-k Attention for Scalable Large Model Inference. - Ping Gong 009; Jiawei Yi; Shengnan Wang; Juncheng Zhang; Zewen Jin; Ouxiang Zhou; Ruibo Liu; Guanbin Xu; Youhui Bai; Bowen Ye; Kun Yuan; Tong Yang 003; et al. (ACL (Findings), 2025)

BigMac: A Communication-Efficient Mixture-of-Experts Model Structure for Fast Training and Inference. - Zewen Jin; Shengnan Wang; Jiaan Zhu; Hongrui Zhan; Youhui Bai; Lin Zhang; Zhenyu Ming; Cheng Li 001 (CoRR, 2025)

Efficient Long-Context LLM Inference via KV Cache Clustering. - Jie Hu; Shengnan Wang; Yutong He; Ping Gong 009; Jiawei Yi; Juncheng Zhang; Youhui Bai; Renhai Chen; Gong Zhang 001; Cheng Li 001; Kun Yuan (CoRR, 2025)

A Generic, High-Performance, Compression-Aware Framework for Data Parallel DNN Training. - Hao Wu 077; Shiyi Wang; Youhui Bai; Cheng Li 001; Quan Zhou; Jun Yi; Feng Yan 001; Ruichuan Chen; Yinlong Xu (IEEE Trans. Parallel Distributed Syst., 2025)