[論文レビュー] Fast and Smooth Interpolation on Wasserstein Space

Sinho Chewi, Julien Clancy+4 more|arXiv (Cornell University)|Oct 22, 2020

Advanced Numerical Analysis Techniques参考文献 37被引用数 4

ひとこと要約

本論文は、Monge写像を介してユークリッドスプライン補間へと問題を還元することにより、Wasserstein空間上での確率測度の滑らかで粒子流れに基づく補間のための新規手法を提案する。本手法はWasserstein空間における測度補間のための最初の近似保証を達成し、単一細胞ゲノム解析などの応用分野において効率的で解釈可能な軌道を可能にする。主な貢献は、測度値表面の補間にスプライス・プレートスプラインを用いることである。

ABSTRACT

We propose a new method for smoothly interpolating probability measures using the geometry of optimal transport. To that end, we reduce this problem to the classical Euclidean setting, allowing us to directly leverage the extensive toolbox of spline interpolation. Unlike previous approaches to measure-valued splines, our interpolated curves (i) have a clear interpretation as governing particle flows, which is natural for applications, and (ii) come with the first approximation guarantees on Wasserstein space. Finally, we demonstrate the broad applicability of our interpolation methodology by fitting surfaces of measures using thin-plate splines.

研究の動機と目的

• Wasserstein空間上での滑らかで軌道に配慮した確率測度の補間法を構築し、粒子流れの解釈を維持すること。
• 従来のPスプライン手法の限界を克服すること。これらは一意な粒子軌道を提供せず、計算コストが高かった。
• Wasserstein空間上での測度値補間に対する最初の近似保証を提供すること。
• 単一細胞ゲノム解析における個々の細胞軌道の効率的で解釈可能な再構築を可能にすること。
• 多変数測度補間のため、薄板スプラインを用いた表面フィッティングへの拡張を実現すること。

提案手法

• 連続する測度間のMonge写像を構築することで、Wasserstein空間上での補間を標準的なユークリッド3次スプライン補間へ還元する。
• 累積分布関数（CDF）とその逆関数の合成を用いて測度間の輸送写像を定義し、単調性とMonge解を保証する。
• ガウス測度の場合、各々の平均と標準偏差にスプライス・プレートスプラインを別々に適用して、補間された分位数を計算する。
• 粒子軌道は輸送写像のスプライン補間から導出され、初期位置から補間位置へ滑らかで決定論的な経路を保証する。
• 既存のユークリッドスプラインツールボックスを活用することで、N回のMonge写像評価と標準スプライン補間のみで高速計算が可能になる。
• 非ガウス測度の場合、絶対連続性を仮定して、経験的CDFと逆CDFを用いて輸送写像を定義する。

実験結果

リサーチクエスチョン

• RQ1 Wasserstein空間上での与えられた確率測度を滑らかで決定論的な粒子軌道で補間することは可能か？
• RQ2 最適輸送の幾何を反映しつつ、計算効率を維持するにはどうすればよいか？
• RQ3 Wasserstein空間上での測度値補間における近似保証は何か？
• RQ4 薄板スプラインを用いて測度の表面補間を拡張することは可能か？
• RQ5 既存のPスプライン手法はなぜ一意な粒子軌道を生成できないのか？そして、その問題をどのように克服できるか？

主な発見

• 提案手法はMonge解を生成し、初期粒子ごとに一意で滑らかな軌道が保証される。
• 本手法はWasserstein空間上での補間に対する最初の既知の近似保証を達成し、収束に関する理論的境界を提供する。
• ガウス測度の場合、任意の点での補間分布は、平均と標準偏差にスプライス・プレートスプラインを別々に適用することで完全に決定され、分位数は $ m_x + \Phi^{-1}(\alpha) \cdot |s_x| $ で与えられる。
• 従来のPスプライン手法と比較して、計算コストが顕著に低減され、Monge写像評価N回と標準スプライン補間のみで済む。
• 単一細胞データおよび気象台の気温分布に対する実験結果から、滑らかで解釈可能な補間が得られ、明確な粒子流れの解釈が可能である。
• 数学的に整合性があり、実用的にも適用可能な形で、滑らかで一意な代替手法に比べて優れた性能を示した。