AI & ML
Coarse-Graining Optimal Transport for Rollout-Free Post-Training of Flow Matching
arXiv stat.ML0 views
arXiv:2610.41269v1 Announce Type: new Abstract: Flow matching depends critically on the coupling used to pair source and target samples. Optimal transport (OT) couplings can reduce velocity conflict and straighten trajectories; but, in practice, challenges remain: the source comes from a continuous distribution, only a finite number of target samples data are accessible, and the neural network has finite capacity. For these reasons, making the coupling more OT-like may not consistently improve sampling quality, in particular in the low number-of-function-evaluation (NFE) regime. Motivated by this, we introduce ReCouple, a controllable framework that constructs coarse-grained OT couplings, to better understand rollout-free OT-based post-training. We also consider extensions of this basic framework: ReCouple+ extends this toward the continuous-source setting by increasing source resolution, converging to semi-discrete OT in the appropriate limit; and ReCouple++ further adds spatial regularization, motivated by the observation that straight trajectories need not correspond to spatially simple velocity fields. Empirical validation demonstrates the effectiveness of these methods, and it reveals an NFE-dependent interplay between coupling optimality, neural realizability, and generation quality.
Source:https://arxiv.org/abs/2610.41269PreviousAdaptive Uncertainty Quantification on Simplicial Complexes with Conformalized Simplicial Neural NetworksNextConditional Neural SDEs for Financial Time-Series Generation
Related posts

Formally verifying agent-written full-stack web apps with PDD Kit
0Microsoft Research
I built an Android app with AI. Does that count as vibe coding?
0dev.to Discuss
Post-Construction Cleaning Before Opening Day
0dev.to Productivity
Podcastle: The AI Studio Revolution and the Enterprise Guide to Scalable Audio Production
0dev.to Productivity
The SaaS launch checklist I keep reusing (4 weeks out to a month after)
0dev.to Productivity
Gen AI Concepts for Bedrock
0dev.to Machine Learning