Yifei Wang
I am a Member of Technical Staff at Amazon AGI SF Lab. I help build the lab’s next-generation general-agent training effort from scratch, shaping its agent RL system, harness design, data pipeline, and post-training strategy.
I was a postdoc at MIT CSAIL (2023-2025), advised by Stefanie Jegelka. I received my Ph.D. in Applied Mathematics from Peking University, advised by Yisen Wang, Zhouchen Lin, and Jiansheng Yang. I also completed my B.S. and B.A. at Peking University.
In research, I am generally interested in reasoning, representation learning, and self-supervised learning. My work has received 5 best paper awards and has been featured by MIT News and Anthropic. I serve as an Area Chair for ICLR, ICML and NeurIPS.
news
| April, 2026 | Four papers were accepted to ICML 2026, covering self-supervised rewards for RL (SSL4RL), single-stage sparse multi-vector retrieval (SSR), agentic graph reasoning (GRASP), and data scale-difficulty interplay. One latent visual reasoning work, Monet, was accepted to CVPR 2026. |
|---|---|
| January, 2026 | Six papers were accepted to ICLR 2026, explaining scaling laws of CoT length, showing AR models rival diffusion models for any-order generation, and exploring real-world benefits of sparsity via ultra-sparse embeddings, sparse feature attention, and predicting LLM transferability. |
| December, 2025 | Our paper G1 has received the Best Paper Award at the NeurIPS 2025 NPGML Workshop. |
| August, 2025 | I gave an invited talk Two New Dimensions of Sparsity for Scaling LLMs at Google DeepMind on sparse long-context training (ICLR 2025) and sparse embedding (ICML 2025 Oral). |
| June, 2025 | Our ICML 2025 paper was featured in an MIT News article, Unpacking the bias of large language models, where we identified and theoretically proved the root causes of position bias in Transformers. |
| June, 2025 | I gave an invited talk at the ASAP Seminar on Your Next-Token Prediction and Transformers Are Biased for Long-Context Modeling—see the recording at YouTube. |
selected papers (see full publication)
- ICMLSSL4RL: Revisiting Self-supervised Learning as Intrinsic Reward for Visual-Language ReasoningICML, 2026
- ICLR
Best Paper Runner-up When More is Less: Understanding Chain-of-Thought Length in LLMsICLR, 2026🏆 Best Paper Runner-up Award at ICLR 2025 Workshop on Reasoning and Planning for LLMs - NeurIPS
Best Paper Award
at ICML-W’24A Theoretical Understanding of Self-Correction through In-context AlignmentIn NeurIPS, 2024🏆 Best Paper Award at ICML 2024 ICL Workshop
We proposed the first theoretical explanation of how LLM self-correction works (as in OpenAI o1) and showed its effectiveness against social bias and jailbreak attacks. - ICLR
JMLR Chaos is a Ladder: A New Theoretical Understanding of Contrastive Learning via Augmentation OverlapIn ICLR, 2022A new augmentation overlap theory for understanding the generalization of contrastive learning. Cited over 150 times. Extended version was accepted at JMLR.