[論文レビュー] Quantum Optimal Transport for Tensor Field Processing

Gabriel Peyré, Lénaïc Chizat+2 more|arXiv (Cornell University)|Dec 20, 2016

Advanced Neuroimaging Techniques and Applications参考文献 45被引用数 5

ひとこと要約

本稿では、正定値（PSD）行列値をとるテンソル場（測度）に対する最適輸送の新しい定式化、すなわち量子最適輸送（Q-OT）を提案する。量子のボン・ネウマン量子エントロピーを適合性指標として用い、量子エントロピー正則化を適用することで、量子シンクホルンアルゴリズムを用いた効率的計算が可能となり、古典的シンクホルンアルゴリズムをPSD行列へ一般化する。本手法は、拡散テンソル画像法、手続き的ノイズ、異方的メッシュ生成、スペクトルテクスチャ合成への応用において、線形補間よりも優れた幾何的適合性を実現する重心計算を可能にする。

ABSTRACT

This article introduces a new notion of optimal transport (OT) between tensor fields, which are measures whose values are positive semidefinite (PSD) matrices. This "quantum" formulation of OT (Q-OT) corresponds to a relaxed version of the classical Kantorovich transport problem, where the fidelity between the input PSD-valued measures is captured using the geometry of the Von-Neumann quantum entropy. We propose a quantum-entropic regularization of the resulting convex optimization problem, which can be solved efficiently using an iterative scaling algorithm. This method is a generalization of the celebrated Sinkhorn algorithm to the quantum setting of PSD matrices. We extend this formulation and the quantum Sinkhorn algorithm to compute barycenters within a collection of input tensor fields. We illustrate the usefulness of the proposed approach on applications to procedural noise generation, anisotropic meshing, diffusion tensor imaging and spectral texture synthesis.

研究の動機と目的

• 空間上の各点が正定値（PSD）行列に紐付くテンソル場のための新しい最適輸送フレームワークを開発し、幾何的に意味のある補間と平均化を可能にする。
• 質量保存を緩和することで、質量の生成・消滅を許容する未均衡最適輸送をテンソル値測度へ拡張する。
• 量子エントロピー正則化を用いて、古典的シンクホルンアルゴリズムをPSD行列の量子的設定に一般化する。
• 医療画像およびコンピューターグラフィックス分野への応用を想定し、複数のテンソル場間の重心を効率的に計算する。
• 実世界の応用において、線形補間と比較して、テンソルの非等方性、方位、質量分布の保持においてQ-OTの優位性を示す。

提案手法

• PSD行列値測度間の静的で未均衡な最適輸送定式化を提案し、適合性項としてボン・ネウマン量子エントロピーを用いる。
• 輸送問題に量子エントロピー正則化を導入することで、反復スケーリングで解ける厳密に凸な最適化問題に変換する。
• 行列対数関数および指数関数を用いたマトリクス値輸送計画に一般化された量子シンクホルンアルゴリズムを構築する。
• 量子正則化を用いた凸最適化問題を解くことで、複数のテンソル場の重心を計算するQ-OTフレームワークを拡張する。
• 複素ヒルベルトPSD行列への量子シンクホルンアルゴリズムの適用により、フーリエスペクトルを用いたテクスチャ合成への応用を可能にする。
• 質量の生成・消滅を制御するための緩和パラメータρを導入し、DTIデータにおける高い被験者間変動下でも安定した補間を実現する。

実験結果

リサーチクエスチョン

• RQ1 各位置に正定値行列が対応するテンソル場に対して、スカラーまたはベクトル測度ではなく、最適輸送を意味的に拡張できるか？
• RQ2 幾何的構造を保ちつつ、テンソル値測度の文脈で質量の不均衡（未均衡輸送）を効果的にモデル化できるか？
• RQ3 量子ダイバージェンスを用いて、古典的シンクホルンアルゴリズムをマトリクス値輸送計画へ一般化できるか？
• RQ4 提案された量子エントロピー正則化は、テンソル場輸送のための安定的で効率的かつスケーラブルな数値解法をもたらすか？
• RQ5 得られたQ-OTフレームワークは、テンソル場補間および重心計算において、非等方性と方位の保持を線形補間よりも優れているか？

主な発見

• 量子シンクホルンアルゴリズムは、反復的スケーリングによるマトリクス値結合の操作を経て、効率的かつ安定的に収束し、テンソル場間の最適輸送計画の高速計算を可能にする。
• ρ = 0.05をρ = 1に比べて使用することで、質量の生成・消滅を許容するようになるため、特に高ばらつきのDTIデータにおいて補間アーチファクトが顕著に減少する。
• Q-OTは、楕円のサイズ（大きさ）と非等方性／方位を両方とも保持する幾何的に忠実なテンソル場補間を実現する。これは線形補間とは対照的である。
• 本手法は、拡散テンソル画像法における被験者間の一貫性ある平均化および手続き的ノイズ生成における複数のテンソル場の重心計算に成功している。
• スペクトルテクスチャ合成の文脈では、Q-OTによりパワースペクトルの視覚的に整合性のある補間が得られ、周波数帯域全体にわたり、大きさと構造の滑らかな遷移が実現される。
• 本フレームワークは、複素ヒルベルトPSD行列に対してもそのまま一般化可能であり、実世界の例から得たフーリエベースのテクスチャ合成への直接応用を可能にする。