I am a senior undergraduate student at Peking University, majoring in Computational Mathematics and double majoring in Computer Science, where I am fortunate to be advised by Prof. Di He. Previously, I worked as a research intern at Princeton University, advised by Prof. Chi Jin, and University of Washington, advised by Prof. Simon Shaolei Du. I have also interned at Shanghai AI Lab.

My research interests broadly lie in machine learning and artificial intelligence, with a particular focus on generative AI. I am guided by the belief that meaningful research should identify important problems and/or deliver valuable outcomes — whether through theory, algorithms, data, architectures, or systems — with the ultimate goal of advancing AI in ways that can solve real-world problems and be beneficial to the whole society.

I am seeking PhD opportunities starting in Fall 2026. Please feel free to reach out!
You can find me at: gengyihan [at] stu [dot] pku [dot] edu [dot] cn.

Publications

* stands for equal contribution or core contributor. You can also refer to my Google Scholar.

• Goedel-Prover-V2: Scaling Formal Theorem Proving with Scaffolded Data Synthesis and Self-Correction, Technical Report
  Yong Lin*, Shange Tang*, Bohan Lyu*, Ziran Yang*, Jui-Hui Chung*, Haoyu Zhao*, Lai Jiang*, Yihan Geng*, Jiawei Ge, Jingruo Sun, Jiayun Wu, Jiri Gesi, Ximing Lu, David Acuna, Kaiyu Yang, Hongzhou Lin*, Yejin Choi, Danqi Chen, Sanjeev Arora, Chi Jin*
  [Paper] [Blog] [Code] [Models]

• Ineq-Comp: Benchmarking Human-Intuitive Compositional Reasoning in Automated Theorem Proving on Inequalities, Preprint
  Haoyu Zhao, Yihan Geng, Shange Tang, Yong Lin, Bohan Lyu, Hongzhou Lin, Chi Jin, Sanjeev Arora
  [Paper] [Code and Dataset] [Huggingface]

• Theoretical Benefit and Limitation of Diffusion Language Model, Preprint
  Guhao Feng*, Yihan Geng*, Jian Guan, Wei Wu, Liwei Wang, Di He
  [Paper] [Video] [Slides]

• Lean Workbook: A large-scale Lean problem set formalized from natural language math problems, Neurips 2024
  Huaiyuan Ying, Zijian Wu, Yihan Geng, Zheng Yuan, Dahua Lin, Kai Chen
  [Paper] [Code] [Dataset]

Selected Awards and Honors

• 2023-2024: Samsung Scholarship
• 2023-2024: Peking University Ding Shisun - Gui Linlin Outstanding Student Scholarship
• 2022-2023: The Third Prize of Peking University Scholarship
• 2022-2024: Merit Student of Peking University * 2
• 2021: Chinese Mathematical Olympiad (CMO) Silver Medalist
• 2021: China Girl’s Mathematical Olympiad (CGMO) Gold Medalist