Implicit Bias of (Stochastic) Gradient Descent for Rank-1 Linear Neural Network. - Bochen Lyu; Zhanxing Zhu (NeurIPS, 2023)

Effects of Momentum in Implicit Bias of Gradient Flow for Diagonal Linear Networks. - Bochen Lyu; He Wang; Zheng Wang 001; Zhanxing Zhu (AAAI, 2025)

Can Language Models Serve as Analogy Annotators? - Xiaojing Zhang; Bochen Lyu (ACL (Findings), 2025)

DyCAST: Learning Dynamic Causal Structure from Time Series. - Yue Cheng; Bochen Lyu; Weiwei Xing; Zhanxing Zhu (ICLR, 2025)

A Solvable Attention for Neural Scaling Laws. - Bochen Lyu; Di Wang; Zhanxing Zhu (ICLR, 2025)

Heavy-Ball Momentum Method in Continuous Time and Discretization Error Analysis. - Bochen Lyu; Xiaojing Zhang; Fangyi Zheng; He Wang; Zheng Wang; Zhanxing Zhu (CoRR, 2025)

Analyzing the Implicit Bias of Adversarial Training From a Generalized Margin Perspective. - Bochen Lyu; Zhanxing Zhu (IEEE Trans. Pattern Anal. Mach. Intell., 2025)