arXiv Machine Learning

One-Sided Quantile Coupling for Flow Matching

arXiv:2608. 00978v1 Announce Type: new Abstract: Flow Matching trains continuous-time generative models by regressing the velocity field of a probability path between a simple source distribution and a target data distribution.

arXiv Machine Learning
Sep 4

Beyond Straightness: Non-Crossing Flow Matching via Quantile AlignTree Coupling

The paper introduces Quantile AlignTree Flow Matching (QAT‑FM), a structured coupling method that builds a hierarchical, quantile‑aligned tree to connect a Gaussian prior with a target distribution. QAT‑FM achieves efficient coupling construction in ≠ Nd log N time and allows per‑pair source sampling in ≠ d time, enabling scalable training for high‑dimensional generative tasks. The authors prove that the coupling preserves marginal consistency, produces non‑crossing interpolation paths, and improves path separation compared to independent coupling, while also extending naturally to conditional generation.

By Junyi Lin, Mengyu Li, Jingxuan Hu, Kejun He, Cheng Meng
Hugging Face Trending Papers
Aug 6

Energy-Guided Flow Matching

Pixel-space generative models bypass lossy latent compression, yet necessitate joint learning of global structure and fine-grained details in a high-dimensional space. Standard flow matching interpolates noise toward a fixed clean-image endpoint, leaving the spectral evolution to be learned implicitly.

arXiv Machine Learning
Sep 17

Beyond Random Couplings: Contrastive Noise Alignment in Generative Flows

The paper introduces Contrastive Noise Alignment (CNA), a training-time method for generative flow models that dynamically aligns Gaussian noise with data samples using a cross-modal InfoNCE objective. By modeling noise as an interacting particle system and regularizing with angular entropy and radial norm penalties, CNA reduces arbitrary data-noise couplings and flow curvature. Empirical results show that CNA improves generation quality, lowering FID by over 50% for few-step pixel-space generation compared to standard rectified flow and outperforming optimal transport baselines by at least 24%.

By Lennart Wittke, Vinicius Azevedo