Tail Annealing for Heavy-Tailed Flow Matching
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Standard generative models struggle with heavy-tailed data: Lipschitz architectures cannot produce power-law tails from Gaussian noise, and interpolating between heavy-tailed data and Gaussians is ill-posed. We propose a simple fix: apply the soft-log transform φ(x) = sign(x) · log(1 + |x|) coordinate-wise to data before training, then exponentiate samples after generation. A Hill diagnostic decides per-coordinate whether to transform, leaving light-tailed margins untouched at no added complexity. This compresses heavy tails into a range where standard flow matching succeeds, without heavy-tailed base distributions or architectural modifications. We provide theoretical intuition for why this works: the log-transform maps Pareto tails to exponentials, and the induced dynamics implement a form of tail annealing via power transformations. On a 144-configuration multivariate benchmark (3 copulas, d up to 100, 4 tail indices), Log-FM dominates specialized baselines on W1, CVaR99, and extreme-quantile metrics, and is the only method with zero severe divergences across 2,880 runs.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 11 of its 23 references that are in the map. First posted on arXiv 19 May 2026.
Builds on
11 of 23 references.
Closest ancestors · 1
Same direction, newest first
Other references · 10
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2016Density estimation using Real NVPcited by 2,042
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2024Heavy-Tailed Diffusion Modelscited by 40
- 2022Marginal Tail-Adaptive Normalizing Flowscited by 12
- 2024Flexible Tails for Normalizing Flowscited by 12
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Radial Tail Repair: Post-Training Correction for Diffusion and Flow Models
- 2026Heavy-Tailed Flow Matching via Random Clocks
- 2026Do Heavy Tails Help Diffusion? On the Subtle Trade-off Between Initialization and Training
- 2026Generalizing Score-based generative models for Heavy-tailed Distributions
- 2026Self-Regulating Annealing in Heavy-Tailed Diffusion Models
- 2026Diffusion Models with Heavy-Tailed Targets: Score Estimation and Sampling Guarantees
- 2025Reweighted Flow Matching via Unbalanced OT for Label-free Long-tailed Generation
- 2025Mirror Flow Matching with Heavy-Tailed Priors for Generative Modeling on Convex Domains