arXiv Machine Learning By Yansen Han, Hongxin Sun, Tao Lin

When Can Conditional Flow Matching Replace Pointwise Negative Log-Likelihood?

Read the original on arXiv Machine Learning →

The paper investigates when conditional flow matching (CFM) can replace pointwise negative log-likelihood (NLL) calculations. It shows that for linear Gaussian paths, the endpoint NLL can be exactly decomposed into entropy, a weighted CFM objective, and residual terms, meaning CFM-only estimates are exact only when these residuals cancel. The study finds that ordinary CFM is generally not a pointwise NLL estimator, and even weighted variants may not fully eliminate bias, especially in training or on‑policy settings, with experiments confirming these theoretical insights.

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 Machine Learning.

arXiv Machine Learning
5d ago

Cumulative Marginal Matching for One-Step Diffusion

The paper introduces Cumulative Marginal Matching (CMM), a new objective for training few‑step diffusion and flow‑matching models. CMM treats flow‑matching as a sequence of intermediate marginal distributions and directly optimizes the cumulative Wasserstein‑2 discrepancy between the base and few‑step models across the entire trajectory. It derives a tractable surrogate for this objective, shows that Shortcut‑CSL is a first‑order approximation of CMM, and reports improved generation quality on standard benchmarks under the same training budget.

By Paribesh Regmi, Rui Li
arXiv AI
5d ago

Preconditioned Flow Matching

Preconditioned Flow Matching introduces a precondition-then-match framework that transforms target distributions into more isotropic representations before training flow matching models. By reshaping intermediate distributions, the method improves conditioning, leading to better low-eigenvalue recovery and enhanced performance metrics such as FID, MMD, precision, and recall across Gaussian, Gaussian-mixture, and high-resolution image datasets. The approach is theoretically justified and empirically validated, with ablations showing gains stem from improved geometry rather than extra parameters.

By Shadab Ahamed, Eshed Gal, Md Shahriar Rahim Siddiqui, Simon Ghyselincks, Moshe Eliasof, Eldad Haber
arXiv Statistics ML
Oct 26

Compatible Likelihoods for Flow Matching on Manifolds

The paper investigates flow matching on manifolds, focusing on the 1D flat torus. It identifies a failure mode caused by a property called compatibility, where a likelihood-based flow matching objective can be globally optimized by a velocity field that actually pushes samples away from the data distribution. The authors propose a new compatible objective built from an exponential family over endpoints, which restricts the velocity field to feasible values on compact manifolds and improves performance on high-dimensional product tori, geospatial benchmarks on the sphere, and a large human pose dataset.

By Lucas Ng, Georgios Batzolis, Mark Girolami