5 citations · 9 across the 12 of their papers we have counts for
8 papers · 1 filter
Stochastic Runge-Kutta Methods: Provable Acceleration of Diffusion Models
Yuchen Wu, Yuxin Chen, Yuting Wei
Diffusion models play a pivotal role in contemporary generative modeling, claiming state-of-the-art performance across various domains. Despite their superior sample quality, mains…
Denoising diffusion probabilistic models are optimally adaptive to unknown low dimensionality
Zhihan Huang, Yuting Wei, Yuxin Chen
The denoising diffusion probabilistic model (DDPM) has emerged as a mainstream generative model in generative AI. While sharp convergence guarantees have been established for the D…
Statistical Inference for Policy Evaluation with Temporal Difference Learning
Weichen Wu, Gen Li, Yuting Wei +1
We investigate the statistical properties of Temporal Difference (TD) learning with Polyak-Ruppert averaging, arguably one of the most widely used algorithms in reinforcement learn…
A Sharp Convergence Theory for The Probability Flow ODEs of Diffusion Models
Gen Li, Yuting Wei, Yuejie Chi +1
Diffusion models, which convert noise into new data instances by learning to reverse a diffusion process, have become a cornerstone in contemporary generative modeling. In this wor…
Accelerating Convergence of Score-Based Diffusion Models, Provably
Gen Li, Yu Huang, Timofey Efimov +3
Score-based diffusion models, while achieving remarkable empirical performance, often suffer from low sampling speed, due to extensive function evaluations needed during the sampli…
Theoretical Insights for Diffusion Guidance: A Case Study for Gaussian Mixture Models
Yuchen Wu, Minshuo Chen, Zihao Li +2
Diffusion models benefit from instillation of task-specific information into the score function to steer the sample generation towards desired properties. Such information is coine…
Towards a mathematical theory for consistency training in diffusion models
Gen Li, Zhihan Huang, Yuting Wei
Consistency models, which were proposed to mitigate the high computational overhead during the sampling phase of diffusion models, facilitate single-step sampling while attaining s…
A non-asymptotic distributional theory of approximate message passing for sparse and robust regression
Gen Li, Yuting Wei
Characterizing the distribution of high-dimensional statistical estimators is a challenging task, due to the breakdown of classical asymptotic theory in high dimension. This paper…