Research by Topic

Theoretical Foundations of Diffusion Models

• Faster Diffusion Models via Higher-Order Approximation.

  • Gen Li∗, Yuchen Zhou∗, Yuting Wei, Yuxin Chen (∗=equal contribution)

• Dimension-free convergence of diffusion models for approximate Gaussian mixtures.

  • Gen Li∗, Changxiao Cai∗, Yuting Wei

• Denoising diffusion probabilistic models are optimally adaptive to unknown low dimensionality.

  • Zhihan Huang, Yuting Wei, Yuxin Chen

• Stochastic Runge-Kutta methods: Provable acceleration of diffusion models.

  • Yuchen Wu, Yuxin Chen, Yuting Wei

• Towards a mathematical theory for consistency training in diffusion models.
  

  • Gen Li∗, Zhihan Huang∗, Yuting Wei

  • AISTATS, 2025

• Theoretical insights for diffusion guidance: A case study for Gaussian mixture models.
  

  • Yuchen Wu, Minshuo Chen, Zihao Li, Mengdi Wang, Yuting Wei

  • ICML, 2024

• Accelerating convergence of score-based diffusion models, provably.
  

  • Gen Li∗, Yu Huang∗, Timofey Efimov, Yuting Wei, Yuejie Chi, Yuxin Chen

  • ICML, 2024

• A sharp convergence theory for the probability flow ODEs of diffusion models.

  • Gen Li, Yuting Wei, Yuejie Chi, Yuxin Chen

• Towards faster non-asymptotic convergence for diffusion-based generative models.
  

  • Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi

High Dimensional Statistics and Learning

• Transformers meet in-context learning: A universal approximation theory.

  • Gen Li∗, Yuchen Jiao∗, Yu Huang, Yuting Wei, Yuxin Chen

• A non-asymptotic distributional theory of approximate message passing for sparse and robust regression.
  

  • Gen Li, Yuting Wei

  • Paper awarded for 2024 ICSA Junior Researcher Award

  • Accepted to the Annals of Statistics, 2025

• Approximate message passing from random initialization with applications to Z2 synchronization.
  

  • Gen Li, Wei Fan, Yuting Wei

  • Proceedings of the National Academy of Sciences (PNAS), 2023

• A non-asymptotic framework for approximate message passing in spiked models.
  

  • Gen Li, Yuting Wei

• Mitigating multiple descents: A model-agnostic framework for risk monotonization.
  

  • Pratik Patil, Arun Kumar Kuchibhotla, Yuting Wei, Alessandro Rinaldo

• Minimum L1 interpolators: Precise asymptotics and multiple descent.
  

  • Yue Li, Yuting Wei

• The Lasso with general Gaussian designs with applications to hypothesis testing.
  

  • Michael Celentano, Andrea Montanari, Yuting Wei (alphabetical order) (slides)

  • The Annals of Statistics, 2023

• Uniform consistency of cross validation estimators for high-dimensional ridge regression.
  

  • Pratik Patil, Yuting Wei, Alessandro Rinaldo, Ryan Tibshirani

  • AISTATS, oral presentation, 2021

• Sharp statistical guarantees for adversarially robust Gaussian classification.
  

  • Chen Dan, Yuting Wei, Pradeep Ravikumar

  • ICML, 2020

• Early stopping for kernel boosting algorithms: A general analysis with localized complexities.
  

  • Yuting Wei∗, Fanny Yang∗ and Martin Wainwright

  • Short version accepted to NeurIPS, spotlight presentation, 2017

  • IEEE Transactions on Information Theory, 2019

Reinforcement Learning and Strategic Games

• Statistical inference under adaptive sampling with LinUCB.

  • Wei Fan∗, Kevin Tan∗, Yuting Wei

• Statistical and Algorithmic Foundations of Reinforcement Learning.

  • Yuejie Chi, Yuxin Chen, Yuting Wei

  • INFORMS TutORial, 2025

• Actor-critics can achieve optimal sample efficiency.

  • Kevin Tan∗, Wei Fan∗, Yuting Wei

  • ICML, 2025

• Uncertainty quantification for Markov chains with application to temporal difference learning.

  • Weichen Wu, Yuting Wei, Alessandro Rinaldo

• Statistical inference for temporal difference learning with linear function approximation.

  • Weichen Wu, Gen Li, Yuting Wei, Alessandro Rinaldo

• Hybrid reinforcement learning breaks sample size barriers in linear MDPs.

  • Kevin Tan, Wei Fan, and Yuting Wei

  • NeurIPS, 2024

• Federated Natural Policy Gradient and Actor Critic Methods for Multi-task Reinforcement Learning.

  • Tong Yang, Shicong Cen, Yuting Wei, Yuxin Chen, Yuejie Chi

  • NeurIPS, 2024

• Settling the sample complexity of model-based offline reinforcement learning.
  

  • Gen Li, Laixi Shi, Yuxin Chen, Yuejie Chi, Yuting Wei

  • The Annals of Statistics, 2024

• High-probability sample complexities for policy evaluation with linear function approximation.

  • Gen Li∗, Weichen Wu∗, Yuejie Chi, Cong Ma, Alessandro Rinaldo, Yuting Wei

  • IEEE Transactions on Information Theory, 2024

• The curious price of distributional robustness in reinforcement learning with a generative model.
  

  • Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Matthieu Geist, Yuejie Chi

  • Short version accepted to NeurIPS, 2023

• Is Q-Learning minimax optimal? A tight sample complexity analysis.
  

  • Gen Li, Changxiao Cai, Yuxin Chen, Yuting Wei, Yuejie Chi

  • Short version accepted to ICML, 2021

  • Operations Research, 2024

• Breaking the sample size barrier in model-based reinforcement learning with a generative model.
  

  • Gen Li, Yuting Wei, Yuejie Chi, Yuxin Chen

  • Short version accepted to NeurIPS, 2020 (slides)

  • Operations Research, 2024

• Minimax-optimal multi-agent RL in zero-sum Markov games with a generative model.
  

  • Gen Li, Yuejie Chi, Yuting Wei, Yuxin Chen

  • NeurIPS, oral presentation, 2023

• Fast policy extragradient methods for competitive games with entropy regularization.
  

  • Shicong Cen, Yuting Wei, Yuejie Chi

  • Short version accepted to NeurIPS, 2021

  • Journal of Machine Learning Research, 2024

• Pessimistic Q-Learning for offline reinforcement learning: Towards optimal sample complexity.
  

  • Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi

  • ICML, 2022

• Softmax policy gradient methods can take exponential time to converge.
  

  • Gen Li, Yuting Wei, Yuejie Chi, Yuxin Chen

  • Short version accepted to Conference on Learning Theory (COLT), 2021

  • Mathematical Programming, 2023

• Sample-efficient reinforcement learning is feasible for linearly realizable MDPs with limited revisiting.
  

  • Gen Li, Yuxin Chen, Yuejie Chi, Yuantao Gu, Yuting Wei

  • NeurIPS, 2021 (slides)

• Fast global convergence of natural policy gradient methods with entropy regularization.
  

  • Shicong Cen, Chen Cheng, Yuxin Chen, Yuting Wei, Yuejie Chi

  • Operations Research, 2022 (INFORMS George Nicholson award finalist)

• Sample complexity of asynchronous Q-learning: sharper analysis and variance reduction.
  

  • Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen

  • Short version accepted to NeurIPS, 2020 (slides)

  • IEEE Transactions on Information Theory, 2021

Valid and Efficient Statistical Inference

• Statistical inference under adaptive sampling with LinUCB.

  • Wei Fan∗, Kevin Tan∗, Yuting Wei

• Derandomizing Knockoffs.
  

  • Zhimei Ren, Yuting Wei, Emmanuel Candès

  • Journal of the American Statistical Association, 2023 (code)

• Tackling small eigen-gaps: Fine-grained eigenvector estimation and inference under heteroscedastic noise.
  

  • Chen Cheng, Yuting Wei, Yuxin Chen

  • IEEE Transactions on Information Theory, 2021

• Randomized tests for high-dimensional regression: A more efficient and powerful solution.
  

  • Yue Li, Ilmun Kim, Yuting Wei

  • NeurIPS, 2020

• The geometry of hypothesis testing over convex cones: Generalized likelihood tests and minimax radii.
  

  • Yuting Wei, Martin Wainwright and Adityanand Guntuboyina

  • The Annals of Statistics, 2019

• From Gauss to Kolmogorov: Localized measures of complexity for ellipses.
  

  • Yuting Wei, Billy Fang and Martin Wainwright

  • Electronic Journal of Statistics, 2020

• The local geometry of testing in ellipses: Tight control via localized Kolmogorov widths.
  

  • Yuting Wei and Martin Wainwright

  • IEEE Transactions on Information Theory, 2020

Shape Constrained Inference

• Debiasing evaluations that are biased by evaluations.
  

  • Jingyan Wang, Ivan Stelmakh, Yuting Wei, Nihar B. Shah

  • Short version accepted to AAAI Conference on Artificial Intelligence, 2021

  • Journal of Machine Learning Research, 2024

• The geometry of hypothesis testing over convex cones: Generalized likelihood tests and minimax radii.
  

  • Yuting Wei, Martin Wainwright and Adityanand Guntuboyina

  • The Annals of Statistics, 2019

• Adaptive estimation of planar convex sets.
  

  • Tony Cai, Adityanand Guntuboyina and Yuting Wei (alphabetical order).

  • The Annals of Statistics, 2018

• Sharp minimax bounds for testing discrete monotone distributions.
  

  • Yuting Wei and Martin Wainwright

  • International Symposium on Information Theory (ISIT), 2016

Applications in Bioinformatics and Genomics

• Predicting the unseen: a diffusion-based debiasing framework for transcriptional response prediction at single-cell resolution.

  • Ergan Shang, Yuting Wei, Kathryn Roeder

  • Accepted to Proceedings of the National Academy of Sciences (PNAS), 2025

• Integration and transfer learning of single-cell transcriptomes via cFIT.
  

  • Minshi Peng, Yue Li, Brie Wamsley, Yuting Wei, Kathryn Roeder

  • Proceedings of the National Academy of Sciences (PNAS), 2021 (code)

• Cell type hierarchy reconstruction via reconciliation of multi-resolution cluster tree.
  

  • Minshi Peng, Brie Wamsley, Andrew Elkins, Daniel M Geschwind, Yuting Wei, Kathryn Roeder

  • Nucleic Acids Research, 2021 (code)

• MHC binding prediction with KernelRLSpan and its variations.

  • Wen-Jun Shen∗, Yuting Wei∗, Xin Guo∗, Stephen Smale, Hau-San Wong and Shuai Cheng Li

  • Journal of Immunological Methods, 2014