[論文レビュー] Multi-Marginal Gromov-Wasserstein Transport and Barycenters

Florian Beier, Robert Beinert+1 more|arXiv (Cornell University)|May 13, 2022

Topological and Geometric Data Analysis被引用数 5

ひとこと要約

本稿では、複数の測度付き距離空間を同時に比較できるようにGW距離を一般化した多準拠Gromov-Wasserstein（GW）輸送を導入する。これにより、形状、グラフ、構造的およびラベル情報を有するデータの共同整列が可能になる。本手法は、タイトな二重凸緩和と多準拠Sinkhornアルゴリズムを用い、条件付き負定値性の下で理論的保証が得られ、非平衡および融合GW重心への応用も可能である。

ABSTRACT

Gromov-Wasserstein (GW) distances are combinations of Gromov-Hausdorff and Wasserstein distances that allow the comparison of two different metric measure spaces (mm-spaces). Due to their invariance under measure- and distance-preserving transformations, they are well suited for many applications in graph and shape analysis. In this paper, we introduce the concept of multi-marginal GW transport between a set of mm-spaces as well as its regularized and unbalanced versions. As a special case, we discuss multi-marginal fused variants, which combine the structure information of an mm-space with label information from an additional label space. To tackle the new formulations numerically, we consider the bi-convex relaxation of the multi-marginal GW problem, which is tight in the balanced case if the cost function is conditionally negative definite. The relaxed model can be solved by an alternating minimization, where each step can be performed by a multi-marginal Sinkhorn scheme. We show relations of our multi-marginal GW problem to (unbalanced, fused) GW barycenters and present various numerical results, which indicate the potential of the concept.

研究の動機と目的

• GW距離を二重比較から拡張し、複数の入力測度付き距離空間を同時に処理できるようにすること。
• 二重凸緩和とエントロピー正則化を用いた、数値的に取り扱いやすい多準拠GW輸送の定式化を開発すること。
• 多準拠設定における融合GW変種を用いてラベル情報を統合すること。
• 強収縮性および弱下半連続性などの弱い条件下で、解の存在と収束の理論的保証を確立すること。
• 本フレームワークが、マルチソースデータ統合および形状解析におけるGW重心の計算に有効であることを実証すること。

提案手法

• 標準GWの一般化として、N個の測度付き距離空間間の対間距離構造を保持するコスト関数を最小化する多準拠GW輸送問題を提案する。
• 数値的安定性と柔軟性を確保するため、エントロピー項およびCsiszárダイバージェンス項を用いた正則化および非平衡バージョンを導入する。
• コスト関数が条件付き負定値である場合に、多準拠GW問題の二重凸緩和がタイトであることを示す。
• 緩和問題を交互最小化により解き、各部分問題は多準拠Sinkhornスキームを用いて解く。
• 構造的データとラベル空間を組み合わせることで、融合GW輸送にフレームワークを適用し、幾何学とラベルの両方を統合最適化可能にする。
• Banach–Alaogluの定理およびDiniの定理を用いて、正則化目的関数の弱下半連続性と強収縮性を用いて、最小化子の存在を確立する。

実験結果

リサーチクエスチョン

• RQ1 Gromov-Wasserstein輸送は、二つを超える入力測度付き距離空間に対して、整合的かつ数値的に解ける形で一般化可能か？
• RQ2 多準拠GW問題は、特定の条件下でタイトな二重凸形式に緩和可能か？
• RQ3 条件付き負定値性が二重凸緩和のタイトさを保証する役割を果たすのはどのような仕組みか？
• RQ4 ラベル情報は、融合変種を用いて多準拠GW輸送にどのように統合可能か？
• RQ5 提案されたフレームワークは、複数の入力に対して意味のあるGW重心を計算するために使用可能か？また、従来手法と比較してどのように異なるか？

主な発見

• 提案された多準拠GW輸送定式化により、複数の測度付き距離空間の同時整列が可能となり、GW距離の適用範囲が二重比較から拡張された。
• コスト関数が条件付き負定値である場合、多準拠GW問題の二重凸緩和はタイトであり、これらの条件下で元の問題の正確な回復が保証される。
• 正則化および非平衡バージョンの多準拠GW問題は強収縮性および弱下半連続性を満たし、最小化子の存在が保証される。
• 数値的結果により、形状クラスタリング、マルチグラフマッチング、融合データ整列などの応用分野におけるフレームワークの実現可能性と潜在的価値が示された。
• 多準拠Sinkhornスキームを用いることで、大規模なデータ統合タスクに対してもスケーラブルな解法が可能である。
• ラベル空間を統合することにより、フレームワークは自然に融合GW重心へと拡張され、幾何学的および意味的情報を併せ持つ構造的データの統一的アプローチを提供する。