← Search

Ruofeng Yang

8 accepted papers

2026

Multi-Subspace Multi-Modal Modeling for Diffusion Models: Estimation, Convergence and Mixture of Experts

ICLR 2026poster

Recently, diffusion models have achieved a great performance with a small dataset of size $n$ and a fast optimization process. Despite the impressive performance, the estimation error suffers from the curse of dimensionality $n^{-1/D}$, where $D$ is the data dimension. Since images are usually a un…

Cited by 0SourceScholar
2025

Improved Discretization Complexity Analysis of Consistency Models: Variance Exploding Forward Process and Decay Discretization Scheme

ICML 2025poster

Consistency models, a new class of one-step generative models, have shown competitive performance with multi-step diffusion models. The most challenging part of consistency models is the training process, which discretizes the continuous diffusion process into $K$ steps and trains a one-step mapping…

Cited by 0SourcePDFScholar
2025

The Polynomial Iteration Complexity for Variance Exploding Diffusion Models: Elucidating SDE and ODE Samplers

AISTATS 2025poster

Recently, variance exploding (VE) diffusion models have achieved state-of-the-art (SOTA) performance in two implementations: (1) the SDE-based implementation and (2) the probability flow ODE (PFODE) implementation. However, only a few works analyze the iteration complexity of VE-based models, and mo…

Cited by 0SourceScholar
2024

Few-Shot Diffusion Models Escape the Curse of Dimensionality

NeurIPS 2024poster

While diffusion models have demonstrated impressive performance, there is a growing need for generating samples tailored to specific user-defined concepts. The customized requirements promote the development of few-shot diffusion models, which use limited $n_{ta}$ target samples to fine-tune a pre-t…

Cited by 1SourcePDFScholar
2024

Leveraging Drift to Improve Sample Complexity of Variance Exploding Diffusion Models

NeurIPS 2024poster

Variance exploding (VE) based diffusion models, an important class of diffusion models, have shown state-of-the-art (SOTA) performance. However, only a few theoretical works analyze VE-based models, and those works suffer from a worse forward convergence rate $1/\text{poly}(T)$ than the $\exp{(-T)}$…

Cited by 0SourcePDFScholar
2023

Learning Adversarial Linear Mixture Markov Decision Processes with Bandit Feedback and Unknown Transition

ICLR 2023poster

We study reinforcement learning (RL) with linear function approximation, unknown transition, and adversarial losses in the bandit feedback setting. Specifically, the unknown transition probability function is a linear mixture model \citep{AyoubJSWY20,ZhouGS21,HeZG22} with a given feature mapping, an…

Cited by 13SourcePDFScholar
2023

Learning Adversarial Low-rank Markov Decision Processes with Unknown Transition and Full-information Feedback

NeurIPS 2023poster

In this work, we study the low-rank MDPs with adversarially changed losses in the full-information feedback setting. In particular, the unknown transition probability kernel admits a low-rank matrix decomposition \citep{REPUCB22}, and the loss functions may change adversarially but are revealed to t…

Cited by 5SourcePDFScholar
2023

Understanding Representation Learnability of Nonlinear Self-Supervised Learning

AAAI 2023technical

Self-supervised learning (SSL) has empirically shown its data representation learnability in many downstream tasks. There are only a few theoretical works on data representation learnability, and many of those focus on final data representation, treating the nonlinear neural network as a ``black box…