arXiv cs.LG

Generalized Wasserstein Flow Matching: Nested Couplings for Point Cloud Generation

· 1小时前
AI 摘要

arXiv:2605.08424v3 Announce Type: replace Abstract: We formulate flow matching on probability measures over probability measures. Directly from the Wasserstein-on-Wasserstein (WoW) continuity equation, we show that admissible nested outer and inner couplings induce a valid metameasure flow and non-local velocity field. Validity requires neither optimal plans nor a smooth manifold. Practically, this opens up the design space of multisample flow matching: exact Wasserstein plans are distinguished by energy and straightness, while sliced and lazy linear plans yield scalable alternatives. Point-cloud experiments, including controlled Bures-Wasserstein and Euclidean minibatch baselines, demonstrate the resulting geometry-cost trade-off.

查看原文(arXiv cs.LG)