arXiv AI By Yeonwoo Cha, Jaehoon Yoo, Semin Kim, Yunseo Park, Jinhyeon Kwon, Seunghoon Hong

Training-Free Refinement of Flow Matching with Divergence-based Sampling

Read the original on arXiv AI →

The paper introduces Flow Divergence Sampler (FDS), a training‑free method that refines intermediate states in flow‑matching models by using the divergence of the marginal velocity field to detect and correct misguidance toward low‑density regions. FDS operates during inference, requires no additional training, and can be applied as a plug‑and‑play module with standard solvers and existing flow backbones. Experiments show that FDS consistently improves fidelity in tasks such as text‑to‑image synthesis and inverse problems.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv Machine Learning
Jun 30

Momentum Guidance: Plug-and-Play Guidance for Flow Models

arXiv:2602. 20360v2 Announce Type: replace Abstract: Flow-based generative methods offer a simple and effective framework for high-fidelity generation, yet pretrained flow models are rarely used in their vanilla conditional form: in image generation, samples without guidance often appear diffuse and lack fine-grained detail.

By Runlong Liao, Jian Yu, Baiyu Su, Chi Zhang, Lizhang Chen, Qiang Liu
arXiv Machine Learning
Oct 2

A Flow Matching Algorithm for Many-Shot Adaptation to Unseen Distributions

The paper introduces Function Projection for Flow Matching (FP‑FM), an algorithm that conditions generative models directly on samples from a target distribution. FP‑FM learns basis functions spanning velocity fields for training distributions and adapts to new distributions via a simple least‑squares projection, enabling efficient sample generation without additional training at inference time. Variants of the method trade expressivity for computational cost by enriching coefficient calculations, and the approach shows markedly better precision and recall than baselines, especially on unseen distributions.

By Tyler Ingebrand, Ruihan Zhao, Kushagra Gupta, David Fridovich-Keil, Sandeep P. Chinchali, Ufuk Topcu
arXiv AI
Oct 5

AREX: Affine-Residual Exponential Integrator for Few-Step Sampling in Flow Matching

AREX is a training‑free sampler for pretrained flow matching models that leverages the target mean and covariance to analytically capture part of the sampling dynamics. By showing that the velocity field of a moment‑matched Gaussian target is the $L^2$‑optimal affine approximation to the marginal velocity field, AREX decomposes the learned dynamics into an affine component—determined by the first two target moments—and a neural residual term. The affine component is integrated explicitly with a matrix‑valued propagator, leaving only the residual term to be integrated numerically, which consistently improves sample fidelity in few‑step image and text‑to‑image generation tasks without retraining the underlying model.

By Shizheng Lin, Soon Hoe Lim, N. Benjamin Erichson