Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Generalizing Score-based generative models for Heavy-tailed Distributions

About

Score-based generative models (SGMs) have achieved remarkable empirical success, motivating their application to a broad range of data distributions. However, extending them to heavy-tailed targets remains a largely open problem. Although dedicated models for heavy-tailed distributions have been proposed, their generative fidelity remains unclear and they lack solid theoretical foundations, leaving important questions open in this regime. In this paper, we address this gap through two theoretical contributions. First, we show that combining early stopping with a suitable initialization is sufficient to extend the diffusion framework to any target distribution; in particular, we establish the well-posedness of the backward process and prove convergence of the approximated diffusion in KL divergence. Second, we derive novel theoretical guarantees for generation with normalizing flows, obtaining convergence results that hold under mild conditions on the flow family and without any assumption on the tail behavior of the target distribution. Building on these results, we propose a unified generative framework for heavy-tailed distributions: a normalizing flow is first trained to capture the tail behavior and is then used as an initialization prior for an SGM, which refines the samples by recovering fine-grained structural details. This design leverages the complementary strengths of the two model classes within a theoretically principled pipeline, overcoming the limitations of existing approaches.

Tiziano Fassina, Gabriel Cardoso, Sylvan Le Corff, Thomas Romary• 2026

Related benchmarks

TaskDatasetResultRank
Image GenerationFFHQ 64x64 (test)
FID2.47
82
Generative ModelingGMM (d=40)
MaxSWD0.1
18
Generative ModelingGMM d=100
MaxSWD0.15
9
Generative ModelingHT d=100
MaxSWD0.529
9
Image GenerationImageNet dogs 512x512 (train)
FID5.53
3
Image GenerationImageNet-birds 50-class 512
FID3.38
3
Showing 6 of 6 rows

Other info

Follow for update