[论文解读] Discretization of functionals involving the Monge-Ampère operator

Jean‐David Benamou, Guillaume Carlier+2 more|arXiv (Cornell University)|Aug 20, 2014

Geometric Analysis and Curvature Flows参考文献 2被引用 11

一句话总结

该论文通过利用内部能量屏障的凸性约束，提出了一种一致的、针对涉及Monge-Ampère算子泛函的有限差分离散化方法，实现了在≥2维空间中Wasserstein梯度流的稳定数值求解。该方法通过凸优化实现了非线性扩散和人群运动模型的有效模拟。

ABSTRACT

Gradient flows in the Wasserstein space have become a powerful tool in the analysis of diffusion equations, following the seminal work of Jordan, Kinderlehrer and Otto (JKO). The numerical applications of this formulation have been limited by the difficulty to compute the Wasserstein distance in dimension >= 2. One step of the JKO scheme is equivalent to a variational problem on the space of convex functions, which involves the Monge-Ampère operator. Convexity constraints are notably difficult to handle numerically, but in our setting the internal energy plays the role of a barrier for these constraints. This enables us to introduce a consistent discretization, which inherits convexity properties of the continuous variational problem. We show the effectiveness of our approach on nonlinear diffusion and crowd-motion models.

研究动机与目标

• 解决在≥2维空间中计算Wasserstein距离的数值挑战，该挑战限制了Jordan-Kinderlehrer-Otto（JKO）格式的实际应用。
• 通过将内部能量作为自然屏障，克服在最优传输数值格式中强制实施凸性约束的困难。
• 为涉及Monge-Ampère算子的变分问题开发一种一致且保持凸性的离散化方法，以用于梯度流模拟。
• 通过时间离散化的Wasserstein梯度流，实现对非线性PDE（如多孔介质方程和人群运动模型）的数值模拟。
• 为人群运动模型中密度界限的约束提供一种平滑且凸的近似，确保收敛至原始约束。

提出的方法

• 利用Brenier定理通过凸函数的梯度参数化概率测度，将Wasserstein距离计算转化为在凸势函数上的凸优化问题。
• 提出一种一致的有限差分离散化方法来处理Monge-Ampère算子，其凸性继承自连续问题，从而保证数值稳定性。
• 用平滑且凸的近似 $\mathcal{U}_\alpha$ 替代非光滑的密度界限指示泛函，该近似在 $\alpha \to \infty$ 时 $\Gamma$-收敛于原始约束。
• 通过求解一系列包含Wasserstein距离和凸能量泛函的凸最小化问题来应用JKO格式，使用有限差分方法计算传输势函数的梯度。
• 采用固定欧拉网格演化密度，每时间步通过最小化器的前推实现质量重分配。
• 通过依赖McCann的位移凸性定理以及内部能量作为屏障的结构，确保离散泛函的凸性。

实验结果

研究问题

• RQ1 如何以保持凸性的方式对Monge-Ampère算子进行一致离散化，以实现Wasserstein梯度流的数值求解？
• RQ2 在位移凸泛函中，内部能量能否作为自然屏障，用于在数值格式中强制实施凸性约束？
• RQ3 在≥2维空间中，Wasserstein距离是否存在一种稳定、一致且凸的有限差分离散化方法？
• RQ4 如何通过一种平滑且凸的泛函来近似人群运动模型中密度有界的约束，以确保收敛至原始问题？
• RQ5 JKO格式能否在2D中有效实现，以模拟如多孔介质方程和宏观人群运动模型等非线性PDE？

主要发现

• 所提出的离散化方法具有一致性，并继承了连续变分问题的凸性，从而实现了Wasserstein梯度流的稳定数值求解。
• 内部能量作为凸性约束的自然屏障，使得能够构建一致且凸的离散泛函。
• 平滑近似 $\mathcal{U}_\alpha$ 在广义位移下保持凸性，且当 $\alpha \to \infty$ 时 $\Gamma$-收敛于强制密度界限的指示泛函。
• 人群运动中拥堵情况的数值模拟显示，质量会避开过度拥挤区域，且随着 $\alpha$ 增大，密度边界更加清晰。
• 该方法成功利用JKO格式在2D中模拟了非线性扩散和人群运动，结果在图4中对不同拥堵参数进行了可视化。
• 该格式在时间离散化的Wasserstein空间梯度流中表现出鲁棒性和有效性，即使能量泛函包含非光滑约束也适用。