Diffusion Models with Heavy-Tailed Targets: Score Estimation and Sampling Guarantees
Abstract
Score-based diffusion models have emerged as a powerful framework for generative modeling, with score estimation as a central statistical bottleneck. However, existing theoretical guarantees largely focus on light-tailed distributions or impose restrictive assumptions such as compact support, limiting their applicability to heavy-tailed data encountered in practice. In this work, we study standard score-based diffusion models for heavy-tailed distributions in a Sobolev class with smoothness , covering both exponential and polynomial tail decay characterized by a tail parameter . Based on kernel density estimation, we derive sharp minimax rates for score estimation, revealing a qualitative distinction between the two regimes: under exponential tails, the estimation rate matches the light-tailed benchmark up to polylogarithmic factors, whereas under polynomial tails it explicitly depends on the tail parameter . We further establish the convergence guarantees for the associated continuous-time reverse diffusion dynamics. For exponential tails and , the generated distribution converges in total variation at the rate up to logarithmic factors, matching the light-tailed benchmark. For polynomial tails, we obtain a -dependent total variation bound, while leaving its optimality open. These results characterize the statistical limits of score estimation and the resulting sampling accuracy for heavy-tailed targets, extending diffusion theory beyond the light-tailed setting.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.