Youhui Bai

According to our database, Youhui Bai authored at least 17 papers between 2017 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

nnScaler-M: Constraint-Guided and Placement-Aware Parallelization Plan Generation for Deep Learning Training.

Jiaan Zhu, Qinghe Wang, Long Zhao, Chizheng Fang, Zewen Jin, Youhui Bai, Zhiqi Lin, Cheng Li

IEEE Trans. Parallel Distributed Syst., July, 2026

Accelerating Long-Tail Generation in Synchronous RLHF Training via Adaptive Tensor Parallelism.

Long Zhao, Qinghe Wang, Jiaan Zhu, Youhui Bai, Zewen Jin, Chaoyi Ruan, Shengnan Wang, Cheng Li

CoRR, May, 2026

AdaCluster: Adaptive Query-Key Clustering for Sparse Attention in Video Generation.

Haoyue Tan, Shengnan Wang, Yulin Qiao, Juncheng Zhang, Youhui Bai, Ping Gong, Zewen Jin, Cheng Li

CoRR, April, 2026

Lagom: Unleashing the Power of Communication and Computation Overlapping for Distributed LLM Training.

Guanbin Xu, ZhenGuo Xu, Yuzhe Li, Youhui Bai, Ping Gong, Chaoyi Ruan, Cheng Li

CoRR, February, 2026

SMIDT: High-Performance Inference Framework for MoE Models with Dynamic Top-K Routing.

Zewen Jin, Shen Fu, Chengjie Tang, Youhui Bai, Shengnan Wang, Jiaan Zhu, Chizheng Fang, Ping Gong, Cheng Li

Proceedings of the Fortieth AAAI Conference on Artificial Intelligence, 2026

2025

CLO: Efficient LLM Inference System with CPU-Light KVCache Offloading via Algorithm-System Co-Design.

Jiawei Yi, Ping Gong, Youhui Bai, Jiaqi Ruan, Shengnan Wang, Pengcheng Wang, Haibo Wang, Weiguang Wang, Xia Zhu, Feng Wu, Cheng Li

CoRR, November, 2025

A Generic, High-Performance, Compression-Aware Framework for Data Parallel DNN Training.

Hao Wu, Shiyi Wang, Youhui Bai, Cheng Li, Quan Zhou, Jun Yi, Feng Yan, Ruichuan Chen, Yinlong Xu

IEEE Trans. Parallel Distributed Syst., July, 2025

Efficient Long-Context LLM Inference via KV Cache Clustering.

Jie Hu, Shengnan Wang, Yutong He, Ping Gong, Jiawei Yi, Juncheng Zhang, Youhui Bai, Renhai Chen, Gong Zhang, Cheng Li, Kun Yuan

CoRR, June, 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

CoRR, February, 2025

HATA: Trainable and Hardware-Efficient Hash-Aware Top-k Attention for Scalable Large Model Inference.

Ping Gong, Jiawei Yi, Shengnan Wang, Juncheng Zhang, Zewen Jin, Ouxiang Zhou, Ruibo Liu, Guanbin Xu, Youhui Bai, Bowen Ye, Kun Yuan, Tong Yang, Gong Zhang, Renhai Chen, Feng Wu, Cheng Li

Proceedings of the Findings of the Association for Computational Linguistics, 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

Proceedings of the Thirty-Ninth AAAI Conference on Artificial Intelligence, 2025

2024

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, Sen Wang, Renhai Chen, Hua Xu, Hongwei Sun

CoRR, 2024

2023

A Survey on Auto-Parallelism of Large-Scale Deep Learning Training.

Peng Liang, Yu Tang, Xiaoda Zhang, Youhui Bai, Teng Su, Zhiquan Lai, Linbo Qiao, Dongsheng Li

IEEE Trans. Parallel Distributed Syst., August, 2023

MPress: Democratizing Billion-Scale Model Training on Multi-GPU Servers via Memory-Saving Inter-Operator Parallelism.

Quan Zhou, Haiquan Wang, Xiaoyan Yu, Cheng Li, Youhui Bai, Feng Yan, Yinlong Xu

Proceedings of the IEEE International Symposium on High-Performance Computer Architecture, 2023

2021

Efficient Data Loader for Fast Sampling-Based GNN Training on Large Graphs.

Youhui Bai, Cheng Li, Zhiqi Lin, Yufei Wu, Youshan Miao, Yunxin Liu, Yinlong Xu

IEEE Trans. Parallel Distributed Syst., 2021

Gradient Compression Supercharged High-Performance Data Parallel DNN Training.

Youhui Bai, Cheng Li, Quan Zhou, Jun Yi, Ping Gong, Feng Yan, Ruichuan Chen, Yinlong Xu

Proceedings of the SOSP '21: ACM SIGOPS 28th Symposium on Operating Systems Principles, 2021

2017

PDS: An I/O-Efficient Scaling Scheme for Parity Declustered Data Layout.

Zhipeng Li, Yinlong Xu, Yongkun Li, Chengjin Tian, Youhui Bai

Proceedings of the 46th International Conference on Parallel Processing, 2017