Coupled Optimal Transport with Landmark Constraints
Abstractの概要
本論文は、輸送コストの最小化のみでは分布間の幾何学的に意味のある変換を特定できない場合があるという、標準的な最適輸送の課題に対処しています。本研究では、変形と輸送計画の間の相互一貫性を強制しながら、少数のアノテーション付きランドマークを用いて変形を誘導し、輸送計画と連続的な変形場を共同で最適化する結合最適輸送フレームワークを提案します。著者らは一般的な変分設定における解の適切性と存在を確立し、極限パラメータ領域を分析した上で、理論的な収束保証を伴う有限要素法ベースの交互数値アルゴリズムを開発しました。人工的な魚の形状や実際の手書き数字マッチングを用いた実験により、特に疎なランドマーク教師データの下で、本手法が輸送のみの重心射影やランドマークのみのフィッティングよりも空間的に整合性の高い変形を復元することが示されています。
新規性
主な新規性は、輸送と幾何学的レジストレーションを個別のタスクとして扱うのではなく、相互一貫性ペナルティを介してコスト駆動型の分布マッチングとランドマーク誘導型の変形場を明示的に結合する統一された最適輸送の定式化にあります。本研究は、有限要素による計算の実現と厳密な収束理論に裏付けられた、ランドマークに基づくレジストレーションと最適輸送の間の変分的な関連性を確立しています。
成果
著者らは穏やかな条件下での最小化解の存在を証明し、結合重みが0および1に近づく際の極限挙動を明らかにし、エントロピー正則化モデルに対する全点収束を含む離散交互アルゴリズムの収束保証を確立しました。合成ベンチマークにおいて、結合アプローチはOT重心射影やランドマークのみのベースラインと比較して変形誤差、テスト用ランドマーク誤差、密度アライメント誤差を低減させ、ランドマークが疎な場合に最大の改善が観察されました。MNIST数字ペアに対する定性評価でも、輸送と疎なランドマークの結合が形状サポート全体で空間的に整合した変形場を生成することが確認されています。
論文の注目点
- 提案されたフレームワークは、疎なランドマーク観測と変形一貫性ペナルティによって制約された輸送計画と滑らかな変形場を共同で推定する。
- 理論的分析により、最小化解の存在、二乗ユークリッドコスト下での変分的解釈、極限挙動、および離散交互最適化スキームの収束性が確立されている。
- 合成形状およびMNIST数字マッチングの実証評価において、輸送のみおよびランドマークのみのベースラインと比較して優れた変形復元と空間的整合性が示されている。