Hailin Zhang

Researcher, Xiaomi Corporation

Joined

July 2022

Names

Hailin Zhang (Preferred)

Emails

****@pku.edu.cn (Confirmed)

****@xiaomi.com (Confirmed)

Personal Links

Career & Education History

Researcher

Xiaomi Corporation (xiaomi.com)

2025 – Present

PhD student

Peking University (pku.edu.cn)

2020 – 2025

Advisors, Relations & Conflicts

PhD Advisor

Bin CUI

2020 – 2025

Expertise

No areas of expertise listed

Publications

SALE : Low-bit Estimation for Efficient Sparse Attention in Long-context LLM Prefilling
Xiaodong Ji, Hailin Zhang, Fangcheng Fu, Bin CUI
- ICLR 2026 Conference Withdrawn Submission
- Readers: Everyone
Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routers
Wenhan Ma, Hailin Zhang, Liang Zhao, Yifan Song, Yudong Wang, Fuli Luo, Zhifang Sui
- Submitted to ICLR 2026
- Readers: Everyone
PQCache: Product Quantization-based KVCache for Long Context LLM Inference
Hailin Zhang, Xiaodong Ji, Yilin Chen, Fangcheng Fu, Xupeng Miao, Xiaonan Nie, Weipeng Chen, Bin Cui
- Proc. ACM Manag. Data 2025
- Readers: Everyone
Malleus: Straggler-Resilient Hybrid Parallel Training of Large-scale Models via Malleable Data and Model Parallelization
Haoyang Li, Fangcheng Fu, Hao Ge, Sheng Lin, Xuanyu Wang, Jiawen Niu, Yujie Wang, Hailin Zhang, Xiaonan Nie, Bin Cui
- Proc. ACM Manag. Data 2025
- Readers: Everyone
CAFE+: Towards Compact, Adaptive, and Fast Embedding for Large-scale Online Recommendation Models
Zirui Liu, Hailin Zhang, Boxuan Chen, Zihan Jiang, Yikai Zhao, Yangyu Tao, Tong Yang, Bin Cui
- ACM Trans. Inf. Syst. 2025
- Readers: Everyone
Efficient and scalable huge embedding model training via distributed cache management
Xupeng Miao, Hailin Zhang, Yining Shi, Xiaonan Nie, Zhi Yang, Yangyu Tao, Jie Jiang, Bin Cui
- VLDB J. 2025
- Readers: Everyone
MEMO: Fine-grained Tensor Management For Ultra-long Context LLM Training
Pinxue Zhao, Hailin Zhang, Fangcheng Fu, Xiaonan Nie, Qibin Liu, Fang Yang, Yuanbo Peng, Dian Jiao, Shuaipeng Li, Jinbao Xue, Yangyu Tao, Bin Cui
- Proc. ACM Manag. Data 2025
- Readers: Everyone
A Unified Framework for Mining Batch and Periodic Batch in Data Streams
Zirui Liu, Xiangyuan Wang, Yuhan Wu, Tong Yang, Kaicheng Yang, Hailin Zhang, Yaofeng Tu, Bin Cui
- IEEE Trans. Knowl. Data Eng. 2024
- Readers: Everyone
Surge Phenomenon in Optimal Learning Rate and Batch Size Scaling
Shuaipeng Li, Penghao Zhao, Hailin Zhang, Samm Sun, Hao Wu, Dian Jiao, Weiyan Wang, Chengjun Liu, Zheng Fang, Jinbao Xue, Yangyu Tao, Bin CUI, Di Wang
- NeurIPS 2024 poster
- Readers: Everyone
CAFE: Towards Compact, Adaptive, and Fast Embedding for Large-scale Recommendation Models
Hailin Zhang, Zirui Liu, Boxuan Chen, Yikai Zhao, Tong Zhao, Tong Yang, Bin Cui
- Proc. ACM Manag. Data 2024
- Readers: Everyone

View all 19 publications

Co-Authors

View all 69 co-authors