I'm a fifth-year Ph.D. candidate in Computer Science at Princeton University, extremely fortunate to be advised by Prof. Sanjeev Arora.

I have spent some wonderful time at Google AI and X as a student researcher. I am sincerely honored to be a recipient of the Wallace Memorial Fellowship in Engineering, one of the highest awards of the Graduate School at Princeton.

My research concentrates on Theoretical Machine Learning. I am broadly interested in theory and empirics for machine learning, with a focus on understanding the empirical success of deep learning from a theoretical perspective.

I obtained my Bachelor's degrees from University of Michigan, Ann Arbor, and Shanghai Jiao Tong University. 

As part of our group culture, I spend a significant amount of time making and enjoying coffee :-)


2018-08-23 18.58.41.jpg


Yi Zhang

Sanjeev Arora, Yi Zhang

Escaping Neural Tangent Kernel Regime via Weight-Decay

Yi Zhang, Guy Gur-Ari


author lists are alphabetical, unless labeled by '#'

(#) Zhiyuan Li, Yi Zhang, Sanjeev Arora

ICLR 2021

(#) Yi Zhang*, Orestis Plevrakis*, Simon S. Du, Xingguo Li, Zhao Song, and Sanjeev Arora

NeurIPS 2020

Udaya Ghai, Holden Lee, Karan Singh, Cyril Zhang, and Yi Zhang

COLT 2020

(#) Nikunj Saunshi, Yi Zhang, Mikhail Kohdak, and Sanjeev Arora

ICML 2020

Mark Braverman, Xinyi Chen, Sham M. Kakade, Karthik Narasimhan, Cyril Zhang, and Yi Zhang

ICML 2020

Naman Agarwal, Xinyi Chen, Elad Hazan, Cyril Zhang, and Yi Zhang

ICLR 2020

(#) Rohith Kuditipudi, Xiang Wang, Holden Lee, Yi Zhang , Zhiyuan Li, Wei Hu, Sanjeev Arora, and Rong Ge

NeurIPS 2019

Naman Agarwal, Brian Bullins, Xinyi Chen, Elad Hazan, Karan Singh, Cyril Zhang and Yi Zhang

ICML 2019

Elad Hazan, Holden Lee, Karan Singh, Cyril Zhang, and Yi Zhang

NeurIPS 2018

Oral Presentation

Sanjeev Arora, Rong Ge, Behnam Neyshabur, and Yi Zhang

ICML 2018

Sanjeev Arora, Elad Hazan, Holden Lee, Karan Singh, Cyril Zhang, and Yi Zhang

ICLR 2018 Workshop

Brian Bullins, Cyril Zhang, Yi Zhang

ICLR 2018

Sanjeev Arora, Andrej Risteski, and Yi Zhang

ICLR 2018

Sanjeev Arora, Rong Ge, Yingyu Liang, Tengyu Ma, and Yi Zhang

ICML 2017

(#) Anna Gilbert, Yi Zhang, Kibok Lee, Yuting Zhang, and Honglak Lee

IJCAI 2017

(#) Tai Cheng, Xiao Tong, Yi Zhang, Xiaogang Wang, and Weinan E

ICLR 2016

(#) Scott Reed, Yi Zhang, Yuting Zhang, Honglak Lee

NIPS 2015

Oral Presentation


Blog posts featuring my works



  • Teaching Assistant for COS 324 Intro to Machine Learning, Fall 2018.

  • Teaching Assistant for COS 445 Economics and Computation, Spring 2019


“As far as the laws of mathematics refer to reality, they are not certain;

and as far as they are certain, they do not refer to reality.”

---- Albert Einstein