[论文解读] A JKO splitting scheme for Kantorovich-Fisher-Rao gradient flows

Thomas Gallouët, Léonard Monsaingeon|arXiv (Cornell University)|Feb 14, 2016

Geometry and complex manifolds参考文献 32被引用 23

一句话总结

本文提出了一种新型的JKO型分裂格式，用于Kantorovich-Fisher-Rao（KFR）度量下的梯度流，将流分解为顺序的Wasserstein（扩散）和Fisher-Rao（反应）步骤。该方法证明了对反应-对流-扩散PDE弱解的收敛性，并通过KFR度量在MK与FR分量上的正交分解，建立了能量耗散不等式。

ABSTRACT

In this article we set up a splitting variant of the JKO scheme in order to handle gradient flows with respect to the Kantorovich-Fisher-Rao metric, recently introduced and defined on the space of positive Radon measure with varying masses. We perform successively a time step for the quadratic Wasserstein/Monge-Kantorovich distance, and then for the Hellinger/Fisher-Rao distance. Exploiting some inf-convolution structure of the metric we show convergence of the whole process for the standard class of energy functionals under suitable compactness assumptions, and investigate in details the case of internal energies. The interest is double: On the one hand we prove existence of weak solutions for a certain class of reaction-advection-diffusion equations, and on the other hand this process is constructive and well adapted to available numerical solvers.

研究动机与目标

• 开发一种在Kantorovich-Fisher-Rao（KFR）度量下构造性且数值可处理的梯度流格式，允许测度的质量发生变化。
• 通过解耦Wasserstein（MK）与Fisher-Rao（FR）分量，建立在KFR度量下时间分裂型Jordan-Kinderlehrer-Otto（JKO）格式的收敛性。
• 在底层度量空间的几何假设最小的前提下，通过所提出的分裂格式证明一类反应-对流-扩散PDE弱解的存在性。
• 通过证明在MK与FR步骤中能量衰减被正确近似，从而恢复完整KFR流的严格能量耗散不等式（EDI）。
• 证明经典JKO估计——如能量单调性、质量控制与BV有界性——在分裂框架下依然成立，确保稳定性与收敛性。

提出的方法

• 提出一种时间分裂型JKO格式，通过在连续时间步长中交替最小化Wasserstein（MK）距离与Fisher-Rao（FR）距离，近似完整KFR梯度流。
• 利用KFR度量的正式Riemann结构，证明其无穷小范数可分解为正交和：$\|\operatorname{grad}_{\mathtt{KFR}}\mathcal{F}(\rho)\|^{2} = \|\operatorname{grad}_{\mathtt{MK}}\mathcal{F}(\rho)\|^{2} + \|\operatorname{grad}_{\mathtt{FR}}\mathcal{F}(\rho)\|^{2}$，从而为分裂提供理论依据。
• 利用KFR度量的下确界卷积结构，严格证明顺序最小化过程的合理性，并推导离散能量估计。
• 应用标准JKO格式（针对MK）的已知结果，并通过变量替换$s = \sqrt{\rho}$重述FR步骤，将其转化为凸Hilbert空间问题。
• 建立密度序列在$L^1 \cap L^∞$下的统一有界性，并证明密度的强$L^1$收敛性以及梯度项的弱收敛性。
• 结合两步估计，推导出离散能量耗散不等式（EDI），并证明当$\tau \to 0$时该不等式可取极限。

实验结果

研究问题

• RQ1 能否通过解耦Wasserstein与Fisher-Rao分量，构造出KFR梯度流的分裂格式，并保证其收敛到弱解？
• RQ2 在度量的正式正交分解成立的前提下，对MK与FR度量的顺序最小化是否能正确近似完整KFR梯度流？
• RQ3 经典JKO估计（如能量单调性、总平方距离有界性与质量控制）在分裂框架下是否仍保持有效？
• RQ4 在能量泛函$\mathcal{F}$满足何种条件下，分裂格式能产生收敛到连续版本的离散能量耗散不等式？
• RQ5 所提出的格式在数值上是否可行，并与现有求解器兼容，特别是MK步骤的Monge-Ampère求解器与FR步骤的凸优化方法？

主要发现

• 在适当的紧致性与凸性假设下，所提出的分裂格式收敛于反应-对流-扩散PDE $\partial_t\rho = \operatorname{div}(\rho \nabla(U'(\rho) + \Psi + K*\rho)) - \rho(U'(\rho) + \Psi + K*\rho)$ 的弱解。
• 能量耗散不等式 $\mathcal{F}(\rho(t_2)) + \int_{t_1}^{t_2} \left( \int_\Omega |\nabla U'(\rho)|^2 \, d\rho \, dt + \int_\Omega |U'(\rho)|^2 \, d\rho \, dt \right) \leq \mathcal{F}(\rho(t_1))$ 在$\tau \to 0$极限下被恢复，确认了度量梯度流结构。
• 在每个时间步长，离散能量估计 $\mathcal{F}(\rho^{n+1}) + \tau \left( \int_\Omega |\nabla U'(\rho^{n+1/2})|^2 \, d\rho^{n+1/2} + \int_\Omega |U'(\rho^{n+1})|^2 \, d\rho^{n+1} \right) \leq \mathcal{F}(\rho^n)$ 成立，确保了稳定性。
• 建立了$\rho^\tau(t) \to \rho(t)$的强$L^1$收敛性，以及$\tilde{\rho}^\tau \nabla U'(\tilde{\rho}^\tau) \rightharpoonup \rho \nabla U'(\rho)$的弱收敛性，从而在能量不等式中实现极限过渡。
• 该方法具有构造性，并与现有数值求解器兼容：MK步骤使用标准的Monge-Ampère求解器，FR步骤通过$s = \sqrt{\rho}$变换转化为凸优化问题。
• 证明了结构性条件$\rho U''(\rho) + U'(\rho)/2 \geq 0$等价于$U$关于FR度量的测地线凸性，且与MK-凸性结合，可保证完整的KFR-EDI。