I am a postdoctoral researcher in the Machine Learning Foundations Group at Microsoft Research Redmond, under Sébastien Bubeck

I obtained my Ph.D. in Computer Science at Princeton University,  advised by Prof. Sanjeev Arora. In 2020, I received the
 Wallace Memorial Fellowship in Engineering, one of the highest awards from 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 the 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


Sanjeev Arora, Yi Zhang


author lists are alphabetical, unless labeled by '#'

(#)Yi Zhang, Arushi Gupta, Nikunj Saunshi, Sanjeev Arora

ICLR 2022


(#) 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