Tail Annealing for Heavy-Tailed Flow Matching
Jean Pachebat
Abstract
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 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, up to 100, 4 tail indices), Log-FM dominates specialized baselines on , CVaR, and extreme-quantile metrics, and is the only method with zero severe divergences across 2,880 runs.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 09316b97-b2dc-4685-abc7-23b8bfb90b29Builds on8
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Flow Matching for Generative ModelingYaron Lipman, Ricky T. Q. Chen, Heli Ben-Hamu, Maximilian Nickel et al.ICLR 2023 · 87 citations
- Tails of Lipschitz Triangular FlowsPriyank Jaini, Ivan Kobyzev, Yaoliang Yu, Marcus A. BrubakerICML 2020 · 60 citations
Related papers
- Mirror Flow Matching with Heavy-Tailed Priors for Generative Modeling on Convex DomainsYunrui Guan, Krishna Balasubramanian, Shiqian MaICLR 2026 · 10 citations
- Fat-Tailed Variational Inference with Anisotropic Tail Adaptive FlowsFeynman T. Liang, Michael W. Mahoney, Liam HodgkinsonICML 2022 · 16 citations
- Flexible Tails for Normalizing FlowsTennessee Hickling, Dennis PrangleICML 2025
- Adapting Noise to Data: Generative Flows from Learned 1D ProcessesJannis Chemseddine, Gregor Kornhardt, Richard Duong, Gabriele SteidlICML 2026 · 1 citation
- Marginal Tail-Adaptive Normalizing FlowsMike Laszkiewicz, Johannes Lederer, Asja FischerICML 2022 · 12 citations
