Entropy-Based Dimension-Free Convergence and Loss-Adaptive Schedules for Diffusion Models
Abstract
Diffusion generative models synthesize samples by discretizing reverse-time dynamics driven by a learned score (or denoiser). Existing convergence analyses of diffusion models typically scale at least linearly with the ambient dimension, and sharper rates often depend on intrinsic-dimension assumptions or other geometric restrictions on the target distribution. We develop an alternative, information-theoretic approach to dimension-free convergence that avoids any geometric assumptions. Under mild assumptions on the target distribution, we bound KL divergence between the target and generated distributions by (up to endpoint factors), where is the Shannon entropy and is the number of sampling steps. Moreover, using a reformulation of the KL divergence, we propose a Loss-Adaptive Schedule (LAS) for efficient discretization of reverse SDE which is lightweight and relies only on the training loss, requiring no post-training heavy computation. Empirically, LAS improves sampling quality over common heuristic schedules.
Cite
@article{arxiv.2601.21943,
title = {Entropy-Based Dimension-Free Convergence and Loss-Adaptive Schedules for Diffusion Models},
author = {Ahmad Aghapour and Erhan Bayraktar and Ziqing Zhang},
journal= {arXiv preprint arXiv:2601.21943},
year = {2026}
}