Probabilistic Representation and Convergence of Gromov-Wasserstein Gradient Flows

Venkatkrishna Karumanchi, Ziv Goldfeld, Kengo Kato, Zhengxin Zhang

Abstract

Wasserstein gradient flows are intimately connected with evolution partial differential equations and diffusion processes. We take the first step in developing such connections for inner product Gromov--Wasserstein (IGW) gradient flows by studying the IGW gradient flow of the relative entropy $\mathsf{H}(\cdot\|γ)$ with respect to the standard Gaussian measure $γ$. We first show that $\mathsf{H}(\cdot\|γ)$ fails to be $λ$-convex along generalized or modified generalized IGW geodesics for any $λ\in \mathbb{R}$, and therefore falls outside the scope of the existing IGW gradient flow theory from Zhang et al. (2026). We bridge this gap by establishing a suitable \emph{local} convexity estimate that enables the construction of the gradient flow and its extension to the infinite time horizon. We then obtain increasingly explicit representations of the resulting dynamics. Starting from a partial integro-differential equation, we derive a nonlinear Fokker--Planck equation and show that its second-moment dynamics decouple from the law as they satisfy an autonomous matrix ODE. This reduces the IGW dynamics to a linear, time-inhomogeneous Fokker--Planck equation, yielding a probabilistic representation as the time-marginal flow of a linear stochastic differential equation resembling the Ornstein--Uhlenbeck process. Finally, we study its asymptotic behavior by establishing exponential convergence of the flow to $γ$ in relative entropy.

Disclosure

“LYS+ 23, CLGL+ 20] . Finally, it remains an open question whether the FPE or SDE identified here arises naturally as a model for a physical or biological phenomenon. 8. Acknowledgements The authors used ChatGPT 5.6 Sol plus to calibrate numerical values used in the coun- terexample in Section B. They independently verified the argument therein.”

PDF page 23
Classification
Computational experiments or data processing
Multiplier
3
Verified

Structural counts

Pages 32 pdf
Theorems 6 source
Lemmas 7 source
Propositions 1 source
Corollaries 3 source
Definitions 6 source
Displayed equations 169 source
Bibliography entries 68 source
Appendix pages 9 estimated

Count notes

  • Source counts use the expanded primary TeX file main.tex.
  • Appendix pages include the first PDF page with an explicit Appendix heading through the final page.