Sharpness-Aware Minimization Revisited: Weighted Sharpness as a Regularization Term. - Yun Yue; Jiadi Jiang; Zhiling Ye; Ning Gao; Yongchao Liu; Ke Zhang (KDD, 2023)

AGD: an Auto-switchable Optimizer using Stepwise Gradient Difference for Preconditioning Matrix. - Yun Yue; Zhiling Ye; Jiadi Jiang; Yongchao Liu; Ke Zhang (NeurIPS, 2023)

EDiT: A Local-SGD-Based Efficient Distributed Training Method for Large Language Models. - Jialiang Cheng; Ning Gao; Yun Yue; Zhiling Ye; Jiadi Jiang; Jian Sha (ICLR, 2025)

Learning to Align, Aligning to Learn: A Unified Approach for Self-Optimized Alignment. - Haowen Wang; Yun Yue; Zhiling Ye; Shuowen Zhang; Lei Fan; Jiaxin Liang; Jiadi Jiang; Cheng Wei; Jingyuan Deng; Xudong Han; Ji Li; Chunxiao Guo; et al. (CoRR, 2025)