Potential Matching Optimal Transport: Continuous Normalizing Flows for Exact $p$-Wasserstein Dynamics

arXiv:2608.05666 2026 Geometry 1 ideas extracted · analyzed Sep 1, 2026

What the math gives to ML

This paper provides a principled way to replace the quadratic geometry built into common OT-regularized CNFs with an explicitly chosen p-Wasserstein geometry. The transferable asset is the generalized Benamou–Brenier potential flow: a scalar potential produces a velocity field whose direction and magnitude depend on the dual exponent q=p/(p-1), while a bridge-matching objective can train it without solving a full OT problem at every update. The most promising neural-network use is a p-controlled flow-matching CNF, with p selected according to whether robustness to outliers, sparse displacement, or smooth quadratic transport is desired.

Ideas from this paper

Mechanism confirmed, baseline not beaten 2026

Generalized-p Potential Flow Matching

Parameterize a continuous normalizing flow by a scalar potential and convert its gradient into the generalized p-optimal velocity field rather than using the usual quadratic-flow velocity. Train the field by matching velocities along straight source-target bridges, while retaining a terminal distribution loss so the flow remains useful when exact pointwise pairings are unavailable.

Useful7/10
Difficulty5/10
Novelty7/10
Paper: Potential Matching Optimal Transport: Continuous Normalizing Flows for Exact $p$-Wasserstein Dynamics arXiv:2608.05666