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 :-)
zhayi{at}microsoft.com
y{dot}zhang{at}cs{dot}princeton{dot}edu

Manuscripts
Sanjeev Arora, Yi Zhang
Publications
author lists are alphabetical, unless labeled by '#'
(#)Yi Zhang, Arushi Gupta, Nikunj Saunshi, Sanjeev Arora
ICLR 2022
Spotlight
(#) 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
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
Misc.
Blog posts featuring my works
Teaching
-
Teaching Assistant for COS 324 Intro to Machine Learning, Fall 2018.
-
Teaching Assistant for COS 445 Economics and Computation, Spring 2019