Background
Experience & education.
Training in computer science, mathematics, and statistics, with research spanning LLM reinforcement learning and the foundations of generalization.
01Research & education
May – Aug 2026
Applied Scientist Intern · Amazon Rufus
Hosts: Zoey Li and Zhenyu Shi
- Developed a Muon recipe for RLVR with no momentum, achieving 29.17% macro-averaged Pass@1 on Qwen3-4B-Base compared with 24.47% for AdamW across three math benchmarks.
- Used spectral diagnostics and checkpoint interventions to study optimization stability and how learned behavior is organized across singular directions and attention projections.
- Investigated RLVR trajectory predictability with random-walk controls. Read the research write-up ↗
Sep 2024 – present
Ph.D. Student · University of Illinois Urbana-Champaign
Computer Science · Advisor: Prof. Tong Zhang
- Developed Adaptive Layerwise Perturbation (ALP) to stabilize off-policy LLM reinforcement learning under policy staleness and training–inference mismatch.
- Proposed DIPNeT for generalization across reasoning and adversarial robustness settings.
- Developed GVM-RAFT, a prompt-specific sample allocation strategy, achieving a 2–4× speedup and improved accuracy over vanilla RAFT.
- Studied feature decoupling in Transformers and their expressiveness for hidden Markov processes, and explained how ensemble methods reduce overadaptation in supervised fine-tuning.
Sep 2022 – Aug 2024
MPhil · Hong Kong University of Science and Technology
Mathematics · Advisor: Prof. Tong Zhang
Studied the harmful effects of benign overfitting on adversarial robustness, and how spurious feature diversification and overparameterization affect out-of-distribution generalization. Supported by the Hong Kong PhD Fellowship.
Sep 2018 – Jun 2022
B.S. in Statistics · University of Science and Technology of China
School of the Gifted Young
Ranked 5/91 in Statistics; outstanding graduate.
02Teaching assistant experience
- CS 441 · Applied Machine LearningSpring 2026
- MATH 2011 · Introduction to Multivariable CalculusFall 2023
- MATH 2421 · ProbabilitySpring 2023, Spring 2024
- Mathematical Analysis B3Fall 2021
- Complex AnalysisSpring 2021
03Selected honors & awards
- Hong Kong PhD Fellowship2022
- Outstanding Graduate of Anhui Province2022
- Outstanding Graduate of USTC2022
- China National Scholarship · top 1.8%2021
- Gold Prize for Outstanding Student Scholarship2020, 2018
04Professional service
Conference reviewer
NeurIPS, ICLR, ICML, AISTATS, AAAI
Journal reviewer
Journal of Machine Learning Research; SIAM Journal on Mathematics of Data Science; IEEE Transactions on Pattern Analysis and Machine Intelligence
05Technical skills
Programming: Python, PyTorch, MATLAB, R, C/C++