Research interests
Stochastic / Bilevel / Decentralized Optimization
Subspace Methods
AI for Mathematics
Beijing, China
Featured project
A reference for what optimization algorithms can achieve.
An interactive reference for lower and upper complexity bounds for optimization algorithms, with assumptions, paper references, and expert validation. The current release covers decentralized optimization.
Selected publications & preprints
2026
Preprint
Yutong He* , Daibo Li* , Guohong Li* , Jiahe Geng* , Zhengyang Huang* , Can Ren* , Zekun Zhang, Yifan Liu, Shuchen Zhu, Hengrui Zhang , Boao Kong, Ming Sun, Shu Li, Chenyi Li, Jiang Hu, Kun Yuan, Zaiwen Wen, Pingwen Zhang
A collaborative agent system for mathematical research, from literature synthesis and theoretical reasoning to experiments.
2026
Preprint
Hengrui Zhang
* , Boao Kong* , Engao Zhang* , Kun Yuan
Randomized subspaces reduce the memory cost of bilevel optimization while retaining convergence guarantees.
2026
Preprint
Hengrui Zhang
* , Boao Kong* , Jiahe Geng, Zhengyang Huang, Kun Yuan
A unified view of decentralized Muon, connecting communication design with convergence and structural limitations.
2026
Preprint
Boao Kong* , Hengrui Zhang
* , Kun Yuan
How perturbations propagate through forward and backward passes, and when stochastic gradient descent still converges.
* Equal contribution.
Education & experience
Education
2026 — Present
Peking University
M.Sc. in Big Data School of Mathematical Sciences
2022 — 2026
Sichuan University
B.S. in Mathematics and Applied Mathematics School of Mathematics GPA: 3.90 / 4.0 · Rank: 1 / 40
Research experience
Jul 2024 — Present
AI for Mathematics · BICMR
Mathematical formalization and theorem proving in Lean 4. Outstanding Participant Award and Special Prize.
2025 — Present
Optimization research · Peking University
Working with Prof. Kun Yuan on randomized subspace methods, compositional error propagation, and OptBound.
Feb — May 2025
Optimization-based RLHF
Research with Prof. Ju Ren at Tsinghua University on optimization formulations and convergence in RLHF.
Selected honors
2026
Provincial Outstanding Graduate · Sichuan
2024
National First Prize · China Undergraduate Mathematical Contest in Modeling
2023, 2024
National Scholarship