[論文レビュー] 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)スキームの実用的利用を制限する要因を克服する。
• 最適輸送の数値スキームにおける凸性制約の強制の難しさを、内部エネルギーを自然なバリアとして用いることで克服する。
• Wasserstein勾配流のシミュレーションに用いる、Monge-Ampère作用素を含む変分問題の、一貫性があり凸性を保つ離散化を構築する。
• 時間離散化されたWasserstein勾配流を用いて、多孔質媒体方程式や群衆移動モデルなどの非線形PDEの数値的シミュレーションを可能にする。
• 群衆移動モデルにおける密度の上限制約を、元の制約に収束する滑らかで凸な近似関数として提示する。

提案手法

• Brenierの定理を用いて、凸関数の勾配による確率測度のパラメータ化を行い、Wasserstein距離の計算を凸ポテンシャル上の凸最適化問題に変換する。
• 連続問題が持つ凸性を継承する一貫性のある有限差分離散化を導入し、数値的安定性を確保する。
• 密度の上限制約を表す非滑らかな指示関数関数を、$\alpha \to \infty$ のとき元の制約に$\Gamma$-収束する滑らかで凸な近似$\mathcal{U}_\alpha$で置き換える。
• JKOスキームを適用し、Wasserstein距離と凸エネルギー関数を含む一連の凸最小化問題を解く。輸送ポテンシャルの勾配は有限差分法で計算する。
• 質量の再分配を各タイムステップでの最小化子のプッシュフォワードにより行うために、固定されたEulerグリッドを用いて密度を進める。
• McCannの移動凸性定理と内部エネルギーのバリアとしての構造に依拠することで、離散的関数の凸性を保証する。

実験結果

リサーチクエスチョン

• RQ1 Wasserstein勾配流の数値的解法に適した、Monge-Ampère作用素の一貫性があり凸性を保つ離散化はどのように実現できるか？
• RQ2 移動凸性関数における内部エネルギーは、数値スキームにおける凸性制約を強制する自然なバリアとして機能できるか？
• RQ3 次元≥2におけるWasserstein距離の一貫性があり安定で凸な有限差分離散化は何か？
• RQ4 群衆移動モデルにおける密度の上限制約は、元の問題に収束する滑らかで凸な関数でどのように近似できるか？
• RQ5 JKOスキームは、多孔質媒体方程式やマクロスコピックな群衆移動モデルのような非線形PDEに対して、2次元で効果的に実装可能か？

主な発見

• 提案された離散化は一貫性があり、連続的な変分問題の凸性を継承しており、Wasserstein勾配流の安定な数値的解法を可能にする。
• 内部エネルギーは凸性制約の自然なバリアとして機能し、一貫性があり凸な離散的関数の構築を可能にする。
• 滑らかな近似$\mathcal{U}_\alpha$は一般化された移動凸性のもとで凸であり、$\alpha \to \infty$ のとき、密度の上限を強制する指示関数関数に$\Gamma$-収束する。
• 混雑状態下での群衆移動の数値的シミュレーションでは、質量が過密を避ける期待される挙動に収束し、$\alpha$が大きいほど密度制限が鋭くなる。
• 本手法により、非線形拡散および2次元における群衆移動のシミュレーションがJKOスキームを用いて成功裏に実行され、図4に varying congestion parameters に対する結果が可視化されている。
• エネルギー関数が非滑らかな制約を含んでも、時間離散化されたWasserstein空間における勾配流スキームは、本手法により頑健かつ効果的に実装可能である。