The Geometry Behind Diffusion and Flow Matching: Gradient Flows and Geodesics in Wasserstein Space

Yian Yao, Weiwei Zhang

The space ${\mathcal{P}}_2({\mathbb{R}}^d$) of probability measures with finite second moment carries a natural geometry: the quadratic Wasserstein distance W_2 makes it a complete met…