English

Sharpen Your Flow: Sharpness-Aware Sampling for Flow Matching

Machine Learning 2026-05-13 v1 Artificial Intelligence

Abstract

Flow matching models generate samples by numerically integrating a learned velocity field, with each integration step requiring a neural network evaluation. Fast generation therefore requires using a small fixed evaluation budget effectively: the key question is not only how to integrate the flow, but where the sampler should spend its steps. We propose SharpEuler, a training-free sampler that profiles a pretrained model offline by estimating where the learned velocity field changes most rapidly along calibration trajectories. This finite-difference estimate defines a solver-aware sharpness profile, which is smoothed and converted by a quantile transform into a timestep grid for any desired inference budget. At test time, sampling remains ordinary Euler integration with the same number of model evaluations as a uniform schedule. We justify SharpEuler using three principles: a numerical principle identifying trajectory acceleration as the leading source of Euler discretization error, a variational principle deriving sharpness-based power-law timestep densities, and a statistical guarantee showing that the finite-sample calibrated sampler is stable at the terminal distribution level. Our experiments show that SharpEuler improves sample quality at fixed budgets, reducing inter-mode leakage and increasing mode coverage.

Keywords

Cite

@article{arxiv.2605.11547,
  title  = {Sharpen Your Flow: Sharpness-Aware Sampling for Flow Matching},
  author = {Aditi Gupta and Soon Hoe Lim and Annan Yu and N. Benjamin Erichson},
  journal= {arXiv preprint arXiv:2605.11547},
  year   = {2026}
}