About Me

I am a PhD student at the Department of Computer Science at Stanford University advised by Stefano Ermon. My research interest is decision making under uncertainty. My main research questions include:

  • accurate uncertainty quantification for high-stake decision making;

  • tailoring machine learning for downstream decision tasks.

I am also broadly interested in other applications of uncertainty quantification and decision theory, such as Bayesian optimization, sequential decision making, and high dimensional probabilistic models.

I am on the 2021-2022 job market.

Contact: sjzhao at stanford dot edu



Publications by Topics

Uncertainty Quantification and Decision Making under Uncertainty

  • H-divergence: A Decision-Theoretic Probability Discrepancy Measure
    Shengjia Zhao*, Abhishek Sinha*, Yutong He*, Aidan Perreault, Jiaming Song, Stefano Ermon (New preprint)

  • Sample-Efficient Safety Assurances using Conformal Prediction
    Rachel Luo, Shengjia Zhao, Jonathan Kuck, Boris Ivanovic, Silvio Savarese, Edward Schmerling, Marco Pavone (New preprint)

  • Calibrating Predictions to Decisions: A Novel Approach to Multi-Class Calibration
    Shengjia Zhao, Michael P Kim, Roshni Sahoo, Tengyu Ma, Stefano Ermon [arXiv] (Neurips’2021)

  • Reliable Decisions with Threshold Calibration
    Roshni Sahoo, Shengjia Zhao, Alyssa Chen, Stefano Ermon (Neurips’2021)

  • Right Decisions from Wrong Predictions: A Mechanism Design Alternative to Individual Calibration
    Shengjia Zhao, Stefano Ermon [arXiv] (AISTATS’2021 Oral 3.1%)

  • Individual Calibration with Randomized Forecast
    Shengjia Zhao, Tengyu Ma, Stefano Ermon [arXiv] (ICML’2020)

  • A framework for Sample Efficient Interval Estimation with Control Variates
    Shengjia Zhao, Christopher Yeh, Stefano Ermon [arXiv] (AISTATS’2020)

  • A Theory of Usable Information under Computational Constraints
    Yilun Xu, Shengjia Zhao, Jiaming Song, Russell Stewart, Stefano Ermon [arXiv] (ICLR’2020 Oral 1.9%)

  • Adaptive Concentration Inequalities for Sequential Decision Problems
    Shengjia Zhao, Enze Zhou, Ashish Sabharwal, Stefano Ermon [pdf] (NeurIPS’2016)

Generative Models

  • Improved Autoregressive Modeling with Distribution Smoothing
    Chenlin Meng, Jiaming Song, Yang Song, Shengjia Zhao, Stefano Ermon (ICLR’2021 Oral 1.8%) openreview

  • Permutation Invariant Graph Generation via Score-Based Generative Modeling
    Chenhao Niu, Yang Song, Jiaming Song, Shengjia Zhao, Aditya Grover, Stefano Ermon [arXiv] (AISTATS’2020)

  • InfoVAE: Balancing Learning and Inference in Variational Autoencoders
    Shengjia Zhao, Jiaming Song, Stefano Ermon [arXiv] (AAAI’2019)

  • Bias and Generalization in Deep Generative Models: An Empirical Study
    Shengjia Zhao*, Hongyu Ren*, Arianna Yuan, Jiaming Song, Noah Goodman, Stefano Ermon [arXiv] (NeurIPS’2018 Spotlight 3%)

  • A Lagrangian Perspective on Latent Variable Generative Models
    Shengjia Zhao, Jiaming Song, Stefano Ermon [arXiv] (UAI’2018 Oral 8.6%)

  • Learning Hierarchical Features from Generative Models
    Shengjia Zhao, Jiaming Song, Stefano Ermon [arXiv] (ICML’2017)

Improving Classical Algorithms with Learning

  • Adaptive Antithetic Sampling for Variance Reduction
    Hongyu Ren*, Shengjia Zhao*, Stefano Ermon [paper] (ICML’2019)

  • Learning Neural PDE Solvers with Convergence Guarantees
    Jun-Ting Hsieh*, Shengjia Zhao*, Lucia Mirabella, Stefano Ermon [arXiv] (ICLR’2019)

  • A-NICE-MC: Adversarial Training for MCMC
    Jiaming Song, Shengjia Zhao, Stefano Ermon [arXiv] [code] (NeurIPS’2017)

Miscellaneous Topics

  • Privacy Preserving Recalibration under Domain Shift
    Rachel Luo, Shengjia Zhao, Jiaming Song, Jonathan Kuck, Stefano Ermon, Silvio Savarese [arXiv]

  • Cross domain imitation learning
    Kun Ho Kim, Yihong Gu, Jiaming Song, Shengjia Zhao, Stefano Ermon [arXiv] (ICML’2020)

  • Adaptive hashing for model counting
    Jonathan Kuck, Tri Dao, Shengjia Zhao, Burak Burtan, Ashish Sabharwal, Stefano Ermon [paper] (UAI’2020)

  • Learning Controllable Fair Representations
    Jiaming Song, Pratyusha Kalluri, Aditya Grover, Shengjia Zhao, Stefano Ermon [paper] (AISTATS’2019)

  • Amortized Inference Regularization
    Rui Shu, Hung H Bai, Shengjia Zhao, Stefano Ermon [arXiv] (NeurIPS’2018)

  • Closing the Gap Between Short and Long XORs for Model Counting
    Shengjia Zhao, Sorathan Chaturapruek, Ashish Sabharwal, Stefano Ermon [arXiv] (AAAI’2016)



Awards and Fellowships

  • JP Morgan PhD Fellowship (2019)
  • Qualcomm Innovation Fellowship (QInF) (2018)
  • Qualcomm Scholarship (2016)
  • Google Excellence Scholarship (2015)



Teaching and Services

  • Reviewer: NeurIPS (2017, 2019, 2020, 2021), ICLR (2019, 2020, 2021), ICML (2019, 2020, 2021)
  • Organizer: Information Theory and Machine Learning (ITML) Workshop (NeurIPS’2019)
  • Teaching: CS228 Head TA (2019 and 2021)