Zihao Wang

I am interested in understanding (and optimizing) the mechanisms of modern machine learning.

Currently, I am a Ph.D. student at Stanford Mathematics. I am extremely fortunate to be advised by Prof. Andrea Montanari.

I will be visiting the Flatiron Institute in New York during the summer of 2026. I also collaborate closely with Prof. Jason D. Lee (at Berkeley).

Previously, I received my B.S. in Mathematics from Peking University in 2024. I visited Princeton in 2023. I was fortunate to be advised by Prof. Jason D. Lee and Prof. Lei Wu for my undergraduate research.

I am open to discussions on other interesting topics. If you are interested in connecting with me, please feel free to reach out!

Publications

* indicates equal contribution or alphabetical author order. Representative papers are highlighted.

Phase Transitions for Feature Learning in Neural Networks
Andrea Montanari*, Zihao Wang*
Under review at the Annals of Statistics, arXiv:2602.01434 / paper / slides

Neural Networks Learn Generic Multi-Index Models near Information-Theoretic Limit
Bohan Zhang*, Zihao Wang*, Hengyu Fu, Jason D. Lee
ICLR, 2026, arXiv:2511.15120 / paper

UQ: Assessing Language Models on Unsolved Questions
Fan Nie*, Ken Ziyu Liu*, Zihao Wang, Rui Sun, Wei Liu, Weijia Shi, Huaxiu Yao, Linjun Zhang, Andrew Y. Ng, James Zou, Sanmi Koyejo, Yejin Choi, Percy Liang, Niklas Muennighoff*
arXiv:2508.17580 / paper

Learning Hierarchical Polynomials of Multiple Nonlinear Features with Three-Layer Networks
Hengyu Fu, Zihao Wang, Eshaan Nichani, Jason Lee
ICLR, 2025, arXiv:2411.17201 / paper

The Local Landscape of Phase Retrieval Under Limited Samples
Kaizhao Liu*, Zihao Wang*, Lei Wu
IEEE Transactions on Information Theory, arXiv:2311.15221 / paper

Learning Hierarchical Polynomials with Three-Layer Neural Networks
Zihao Wang, Eshaan Nichani, Jason Lee
ICLR, 2024, arXiv:2311.13774 / paper

Theoretical Analysis of the Inductive Biases in Deep Convolutional Networks
Zihao Wang, Lei Wu
NeurIPS, 2023, arXiv:2305.08404 / paper

Selected Awards

Weiming Scholar, Peking University 2024

Francis Robbins Upton Fellowship (declined), Princeton University 2024

Selected Presentations

Phase transitions for feature learning

ML theory reading group, Stanford April 2026

ML theory reading group, Simons Institute, Berkeley March 2026

MaD seminar, New York University March 2026

CCM seminar, Flatiron Institute March 2026

ML theory reading group, Wharton, UPenn March 2026

ML seminar, Peking University December 2025

ML theory reading group, Tsinghua University December 2025

Miscellaneous

I love travelling and want to explore places off the beaten path. One of my favorite books is The Art of Travel by Alain de Botton. One of my favorite movies is Nomadland by Chloé Zhao.

I am proud to be born and raised in Qingdao, a coastal city where I grew up just a stone's throw from the sea.

Updated at April 2026

Thanks Jon Barron for this amazing template