Hailin Zhang (张海林)
Email: z.hl [AT] pku.edu.cn
I am a 5th-year PhD student at the School of Computer Science, Peking University, under the supervision of Professor Bin Cui. My research interests primarily lie in the field of MLSys (Machine Learning System), with a focus on large-scale LLM (Large Language Model), DLRM (Deep Learning Recommendation Model), IR (Information Retrieval), and general distributed computing.
I am currently on the job market. Please feel free to reach out if you have openings.Education
PhD, major in Computer Science
Peking University 2020-NowBS, major in Computer Science; BEc, double major in Economics
Peking University 2016-2020
Publications
Publications in reverse chronological order of acceptance date. *
represents co-first author.
2025
- MEMO: Fine-grained Tensor Management For Ultra-long Context LLM Training. PDF
Pinxue Zhao, Hailin Zhang, Fangcheng Fu, Xiaonan Nie, Qibin Liu, Fang Yang, Yuanbo Peng, Dian Jiao, Shuaipeng Li, Jinbao Xue, Yangyu Tao, Bin Cui.
SIGMOD 2025, CCF-A.
2024
Malleus: Straggler-Resilient Hybrid Parallel Training of Large-scale Models via Malleable Data and Model Parallelization. PDF
Haoyang Li, Fangcheng Fu, Hao Ge, Sheng Lin, Xuanyu Wang, Jiawen Niu, Yujie Wang, Hailin Zhang, Xiaonan Nie, Bin Cui.
Preprint.PQCache: Product Quantization-based KVCache for Long Context LLM Inference. PDF
Hailin Zhang, Xiaodong Ji, Yilin Chen, Fangcheng Fu, Xupeng Miao, Xiaonan Nie, Weipeng Chen, Bin Cui.
Preprint.Retrieval-Augmented Generation for AI-Generated Content: A Survey. PDF
Penghao Zhao*, Hailin Zhang*, Qinhan Yu, Zhengren Wang, Yunteng Geng, Fangcheng Fu, Ling Yang, Wentao Zhang, Jie Jiang, Bin Cui.
Preprint.Surge Phenomenon in Optimal Learning Rate and Batch Size Scaling. PDF
Shuaipeng Li*, Penghao Zhao*, Hailin Zhang*, Xingwu Sun, Hao Wu, Dian Jiao, Weiyan Wang, Chengjun Liu, Zheng Fang, Jinbao Xue, Yangyu Tao, Bin Cui, Di Wang.
NeurIPS 2024, CCF-A.Enabling Parallelism Hot Switching for Efficient Training of Large Language Models. PDF
Hao Ge, Fangcheng Fu, Haoyang Li, Xuanyu Wang, Sheng Lin, Yujie Wang, Xiaonan Nie, Hailin Zhang, Xupeng Miao, Bin Cui.
SOSP 2024, CCF-A.A Unified Framework for Mining Batch and Periodic Batch in Data Streams. PDF
Zirui Liu, Xiangyuan Wang, Yuhan Wu, Tong Yang, Kaicheng Yang, Hailin Zhang, Yaofeng Tu, Bin Cui.
IEEE Transactions on Knowledge and Data Engineering.
TKDE 2024, CCF-A.CAFE: Towards Compact, Adaptive, and Fast Embedding for Large-scale Recommendation Models. PDF
Hailin Zhang*, Zirui Liu*, Boxuan Chen, Yikai Zhao, Tong Zhao, Tong Yang, Bin Cui.
ACM SIGMOD International Conference on Management of Data.
SIGMOD 2024, CCF-A.Experimental Analysis of Large-scale Learnable Vector Storage Compression. PDF
Hailin Zhang, Penghao Zhao, Xupeng Miao, Yingxia Shao, Zirui Liu, Tong Yang, Bin Cui.
International Conference on Very Large Data Bases.
VLDB 2024, CCF-A.
2023
Model-enhanced Vector Index. PDF
Hailin Zhang, Yujing Wang, Qi Chen, Ruiheng Chang, Ting Zhang, Ziming Miao, Yingyan Hou, Yang Ding, Xupeng Miao, Haonan Wang, Bochen Pang, Yuefeng Zhan, Hao Sun, Weiwei Deng, Qi Zhang, Fan Yang, Xing Xie, Mao Yang, Bin Cui.
Conference on Neural Information Processing Systems.
NeurIPS 2023, CCF-A.Galvatron: Efficient Transformer Training over Multiple GPUs Using Automatic Parallelism. PDF
Xupeng Miao, Yujie Wang, Youhe Jiang, Chunan Shi, Xiaonan Nie, Hailin Zhang, Bin Cui.
International Conference on Very Large Data Bases.
VLDB 2023, CCF-A.
2022
Hetu: A Highly Efficient Automatic Parallel Distributed Deep Learning System. PDF
Xupeng Miao, Xiaonan Nie, Hailin Zhang, Tong Zhao, Bin Cui.
Science China Information Sciences.
SCIS 2022, CCF-A.HET-GMP: A Graph-based System Approach to Scaling Large Embedding Model Training. PDF
Xupeng Miao, Yining Shi, Hailin Zhang, Xin Zhang, Xiaonan Nie, Zhi Yang, Bin Cui.
ACM SIGMOD International Conference on Management of Data.
SIGMOD 2022, CCF-A.HET: Scaling out Huge Embedding Model Training via Cache-enabled Distributed Framework. PDF
Xupeng Miao*, Hailin Zhang*, Yining Shi, Xiaonan Nie, Zhi Yang, Yangyu Tao, Bin Cui.
International Conference on Very Large Data Bases.
VLDB 2022, CCF-A, Best Scalable Data Science Paper!
Systems
- Hetu
- 2021 Synced Machine Intelligence TOP-10 Open Source Awards.
- Pop SOTA!List for AI Developers 2021.
- Outstanding Award & Champion of 2021 CCF BDCI Contest.