Publications

You can also find my articles on my Google Scholar profile.

Beyond Data Schedule: Fast LLM Pretraining on Data Mixtures by Loss-Reweighting

Jiazheng Li, Huaqing Zhang, Jing Xu, Tianle Cai, Jingzhao Zhang

Published in arXiv preprint, 2026

Jiazheng Li, Hongzhou Lin, Hong Lu, Kaiyue Wen, Zaiwen Yang, Jiaxuan Gao, Yi Wu, Jingzhao Zhang

Published in ICLR, Poster, 2026

Weizhi Fei, Hao Shi, Jing Xu, Jingchen Peng, Jiazheng Li, Jingzhao Zhang, Bo Bai, Wei Han, Zhenyuan Chen, Xueyan Niu

Published in ICLR, Poster, 2026

Published in ICML, Poster, 2025

Published in NeurIPS, Spotlight, 2025