[論文レビュー] Efficient algorithms computing distances between Radon measures on R

Jędrzej Jabłoński, Anna Marciniak‐Czochra|arXiv (Cornell University)|Apr 11, 2013

Medical Imaging Techniques and Applications参考文献 3被引用数 11

ひとこと要約

本稿では、ℝ 上のRadon測度間のWasserstein型距離を計算するための効率的なアルゴリズムを提示している。特に、有界リプシッツ距離（フラット距離）に注目し、平衡二分探索木を用いてO(n²)手法よりも性能を最適化することで、O(n log n)のアルゴリズムを導入した。これは、輸送方程式におけるエスカレータボックスカー列車（EBT）法の検証に不可欠である。

ABSTRACT

In this paper numerical methods of computing distances between two Radon measures on R are discussed. Efficient algorithms for Wasserstein-type metrics are provided. In particular, we propose a novel algorithm to compute the flat metric (bounded Lipschitz distance) with a computational cost O(nlogn). The flat distance has recently proven to be adequate for the Escalator Boxcar Train (EBT) method for solving transport equations with growth terms. Therefore, finding efficient numerical algorithms to compute the flat distance between two measures is important for finding the residual error and validating empirical convergence of different methods.

研究の動機と目的

• 質量増加を伴う問題において、標準的なWasserstein距離が失敗する状況でも、Radon測度間の距離を測定するための計算的に効率的な数値的手法の開発。
• 距離計算における線形計画法の計算ボトル neck を解消するため、Diracデルタから構成される離散測度に特化した最適化アルゴリズムの提案。
• フラット距離の計算に対して実用的でスケーラブルな解決策を提供。これは、エスカレータボックスカー列車（EBT）法における収束解析において不可欠であることが示されている。
• 非保存的過程に対応するための正規化されたWasserstein距離の導入と分析。質量保存系に限らない応用範囲の拡張。
• 平衡二分探索木（BST）を用いて、正しくかつ数値的に安定したまま、アルゴリズムの計算量をO(n²)からO(n log n)に最適化。

提案手法

• 1-Lipschitz関数上の双対最適化問題としてフラット距離を定式化し、最小費用流問題に再定式化可能にした。
• 測度の差を符号付き重みを持つ点質量の列として表現。累積フラックスと位置を追跡するためのデータ構造（funcDescription）を用いた。
• 反復処理における質量点の処理中にfuncDescription構造を維持・更新するため、配列ベースの操作を用いたO(n²)アルゴリズムを実装した。
• 配列を平衡二分探索木（BST）に置き換えることで、挿入・削除・探索の効率を向上させ、O(n log n)のアルゴリズムに最適化した。
• 関数空間内の累積シフトを追跡するグローバル変数p_modifierを導入し、反復毎の計算コストを削減した。
• [-1, 1] 上にランダムに分布したDiracデルタを用いた数値ベンチマークにより、O(n²)とO(n log n)バージョンの性能を比較検証した。

実験結果

リサーチクエスチョン

• RQ1 一般の線形計画法を用いるより、2つのRadon測度間のフラット距離をより効率的に計算できるか？
• RQ2 Diracデルタから構成される離散測度のフラット距離の計算における計算量は何か？ また、O(n²)未満に低減可能か？
• RQ3 データ構造の選択（配列 vs. 衝撃的平衡BST）が、フラット距離計算の実行時間性能に与える影響は？
• RQ4 入力データの構造（例：凝集した点群 vs. 均等に配置された点）が、提案されたアルゴリズムの性能に与える影響はどの程度か？
• RQ5 正規化されたWasserstein距離は、非保存的輸送モデルにおいて、標準的なWasserstein距離の代わりに安定的かつ計算可能である代替手段を提供できるか？

主な発見

• 提案されたフラット距離用O(n log n)アルゴリズムは、O(n²)ベースラインと比較して、特にnが大きい場合に顕著な漸近的計算量の改善を達成した。
• 数値実験の結果、密度が高く凝集した入力では、BST構造がバランスを保ち効率的であるため、O(n log n)アルゴリズムがO(n²)バージョンを上回る性能を示した。
• 点間隔が2より大きい疎な入力では、両アルゴリズムともほぼ線形に動作するが、O(n log n)バージョンはBST操作に起因する高いオーバーヘッドを伴う。
• 平衡二分探索木を用いてfuncDescriptionデータ構造を維持することで、フラット距離の計算コストをO(n²)からO(n log n)に削減した。
• 正規化されたWasserstein距離が有効な距離関数であることが証明され、非保存的系に対しても実用的な代替手段を提供した。
• EBT法における残差誤差推定においてフラット距離が不可欠であることが示され、その効率的計算ツールの開発が正当化された。