English

Tail Annealing for Heavy-Tailed Flow Matching

Machine Learning 2026-05-20 v1 Machine Learning

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 ϕ(x)=sign(x)log(1+x)\phi(x) = \mathrm{sign}(x) \cdot \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, dd up to 100, 4 tail indices), Log-FM dominates specialized baselines on W1W_1, CVaR99_{99}, and extreme-quantile metrics, and is the only method with zero severe divergences across 2{,}880 runs.

Cite

@article{arxiv.2605.20068,
  title  = {Tail Annealing for Heavy-Tailed Flow Matching},
  author = {Jean Pachebat},
  journal= {arXiv preprint arXiv:2605.20068},
  year   = {2026}
}

Comments

18 pages

R2 v1 2026-07-22T07:22:08.628Z