[2604.04646] Training-Free Refinement of Flow Matching with Divergence-based Sampling

Computer Science > Computer Vision and Pattern Recognition arXiv:2604.04646 (cs) [Submitted on 6 Apr 2026] Title:Training-Free Refinement of Flow Matching with Divergence-based Sampling Authors:Yeonwoo Cha, Jaehoon Yoo, Semin Kim, Yunseo Park, Jinhyeon Kwon, Seunghoon Hong View a PDF of the paper titled Training-Free Refinement of Flow Matching with Divergence-based Sampling, by Yeonwoo Cha and 5 other authors View PDF HTML (experimental) Abstract:Flow-based models learn a target distribution by modeling a marginal velocity field, defined as the average of sample-wise velocities connecting each sample from a simple prior to the target data. When sample-wise velocities conflict at the same intermediate state, however, this averaged velocity can misguide samples toward low-density regions, degrading generation quality. To address this issue, we propose the Flow Divergence Sampler (FDS), a training-free framework that refines intermediate states before each solver step. Our key finding reveals that the severity of this misguidance is quantified by the divergence of the marginal velocity field that is readily computable during inference with a well-optimized model. FDS exploits this signal to steer states toward less ambiguous regions. As a plug-and-play framework compatible with standard solvers and off-the-shelf flow backbones, FDS consistently improves fidelity across various generation tasks including text-to-image synthesis, and inverse problems. Comments: Subjects: Compu...