About Me

I am a Ph.D. student in the RISELab (former AMPLab) and Math Department at University of California at Berkeley. I am advised by Michael Mahoney. My research interest lies in computing statistics, optimization and machine learning. Currently, I am interested in leveraging tools from randomized linear algebra to provide efficient and scalable solutions for large-scale optimization and learning problems. I am also working on the theory and application of deep learning. Before joining UC Berkeley, I recieved my B.S. in Math from Zhiyuan College at Shanghai Jiao Tong University.


  • Hessian-based Analysis of Large Batch Training and Robustness to Adversaries.
    Z Yao, A Gholami, Q Lei, K Keutzer, MW Mahoney
    arXiv preprint arXiv
  • Inexact non-convex Newton-type methods.
    Z Yao, P Xu, F Roosta-Khorasani, MW Mahoney
    arXiv preprint arXiv
  • A hybrid adaptive MCMC algorithm in function spaces.
    Q Zhou, Z Hu, Z Yao, J Li
    arXiv preprint arXiv
    SIAM/ASA Journal on Uncertainty Quantification 5 (1), 621-639
  • On an adaptive preconditioned Crank–Nicolson MCMC algorithm for infinite dimensional Bayesian inference.
    Z Hu, Z Yao, J Li
    arXiv preprint arXiv
    Journal of Computational Physics 332, 492-503
  • A TV-Gaussian prior for infinite-dimensional Bayesian inverse problems and its numerical implementation.
    Z Yao, Z Hu, J Li
    arXiv preprint arXiv
    Inverse Problems 32 (7), 075006