Tianhao Wang (王天浩)

Ph.D. student, Department of Statistics and Data Science, Yale University.


Department of S&DS
24 Hillhouse Avenue
New Haven, CT 06511

I am a fifth year Ph.D. student in the Department of Statistics and Data Science at Yale University. I am very fortunate to be advised by Prof. Zhou Fan. I am broadly interested in various aspects of statistics and machine learning theory, with a recent focus on AMP algorithms and implicit bias of optimization algorithms.

Prior to Yale, I obtained my Bachelor’s degree in mathematics with a dual degree in computer science at University of Science and Technology of China.


Selected recent papers by topics (*: equal contribution)

Approximate Message Passing algorithms
  1. Universality of Approximate Message Passing algorithms and tensor networks
    Tianhao WangXinyi Zhong, and Zhou Fan
    arXiv preprint arXiv:2206.13037, 2022
  2. Approximate Message Passing for orthogonally invariant ensembles: Multivariate non-linearities and spectral initialization
    Xinyi Zhong*,  Tianhao Wang*, and Zhou Fan
    arXiv preprint arXiv:2110.02318, 2021
Implicit bias of optimization algorithms
  1. Fast Mixing of Stochastic Gradient Descent with Normalization and Weight Decay
    Zhiyuan LiTianhao Wang, and Dingli Yu
    In Advances in Neural Information Processing Systems (NeurIPS), 2022
  2. Implicit bias of gradient descent on reparametrized models: On equivalence to mirror descent
    Zhiyuan Li*,  Tianhao Wang*, Jason D. Lee, and Sanjeev Arora
    In Advances in Neural Information Processing Systems (NeurIPS), 2022
    Abridged version accepted for a contributed talk to ICML 2022 Workshop on Continuous time methods for machine learning
  3. What happens after SGD reaches zero loss?--A mathematical framework
    Zhiyuan LiTianhao Wang, and Sanjeev Arora
    In International Conference on Learning Representations (ICLR), 2022  (Spotlight)
Online learning and reinforcement learning
  1. Learn to match with no regret: Reinforcement learning in Markov matching markets
    Yifei MinTianhao WangRuitu XuZhaoran Wang, Michael I Jordan, and Zhuoran Yang
    In Advances in Neural Information Processing Systems (NeurIPS), 2022  (Oral)
  2. A simple and provably efficient algorithm for asynchronous federated contextual linear bandits
    Jiafan He*,  Tianhao Wang*, Yifei Min*, and Quanquan Gu
    In Advances in Neural Information Processing Systems (NeurIPS), 2022
  3. Learning stochastic shortest path with linear function approximation
    Yifei MinJiafan HeTianhao Wang, and Quanquan Gu
    In International Conference on Machine Learning (ICML), 2022
  4. Variance-aware off-policy evaluation with linear function approximation
    Yifei Min*,  Tianhao Wang*, Dongruo Zhou, and Quanquan Gu
    In Advances in neural information processing systems (NeurIPS), 2021
  5. Provably efficient reinforcement learning with linear function approximation under adaptivity constraints
    Tianhao Wang*, Dongruo Zhou*, and Quanquan Gu
    In Advances in Neural Information Processing Systems (NeurIPS), 2021
Orbit recovery model
  1. Maximum likelihood for high-noise group orbit estimation and single-particle cryo-EM
    Zhou Fan, Roy R Lederman, Yi SunTianhao Wang, and Sheng Xu
    arXiv preprint arXiv:2107.01305, 2021
  2. Likelihood landscape and maximum likelihood estimation for the discrete orbit recovery model
    Zhou FanYi SunTianhao Wang, and Yihong Wu
    Communications on Pure and Applied Mathematics, 2022