[論文レビュー] Minimax rates of estimation for smooth optimal transport maps

Jan-Christian Hütter, Philippe Rigollet|arXiv (Cornell University)|May 14, 2019

Statistical Methods and Inference被引用数 11

ひとこと要約

本稿では、一般次元における滑らかな最適輸送写像のための最初のミニマックス推定レートを確立する。これは、経験的半双対最適輸送問題を最小化するウェーブレットに基づく推定器を提案することで達成される。この手法は、新たな安定性の議論と補完的な下界を用いて、近似的にミニマックス最適性を達成する。合成実験により、滑らかさ正則化の利点が裏付けられている。

ABSTRACT

Brenier's theorem is a cornerstone of optimal transport that guarantees the existence of an optimal transport map $T$ between two probability distributions $P$ and $Q$ over $\mathbb{R}^d$ under certain regularity conditions. The main goal of this work is to establish the minimax estimation rates for such a transport map from data sampled from $P$ and $Q$ under additional smoothness assumptions on $T$. To achieve this goal, we develop an estimator based on the minimization of an empirical version of the semi-dual optimal transport problem, restricted to truncated wavelet expansions. This estimator is shown to achieve near minimax optimality using new stability arguments for the semi-dual and a complementary minimax lower bound. Furthermore, we provide numerical experiments on synthetic data supporting our theoretical findings and highlighting the practical benefits of smoothness regularization. These are the first minimax estimation rates for transport maps in general dimension.

研究の動機と目的

• 一般次元における滑らかさ仮定の下で、確率分布間の最適輸送写像のミニマックス推定レートを確立すること。
• 有限標本からの輸送写像のための実用的で理論的根拠を持つ推定器を開発すること。
• 高次元設定における滑らかな輸送写像の推定誤差に関する理論的理解のギャップを埋めること。

提案手法

• 経験的半双対最適輸送問題の最小化に基づく推定器を提案する。
• 滑らかさの強制と計算可能性を実現するため、探索空間を切断されたウェーブレット展開に制限する。
• 基礎となる測度の摂動に対して、半双対定式化のための新しい安定性バインドを確立する。
• 推定精度の根本的限界を特定するため、ミニマックス下界を導出する。
• 安定性結果と近似理論を組み合わせることで、提案された推定器の近似的にミニマックス最適性を示す。
• 合成データ上の数値実験により理論的発見を検証し、滑らかさ正則化の影響を示す。

実験結果

リサーチクエスチョン

• RQ1 一般次元における滑らかな最適輸送写像のミニマックス推定レートは何か？
• RQ2 ウェーブレットに基づく推定器は、滑らかな輸送写像の推定において近似的にミニマックス最適性を達成できるか？
• RQ3 半双対最適輸送問題の安定性特性は、推定誤差バインドにどのように寄与するか？
• RQ4 滑らかな輸送写像の推定精度に対する根本的限界（下界）は何か？
• RQ5 滑らかさ正則化は、有限標本設定における輸送写像推定器の性能にどのように影響するか？

主な発見

• 提案されたウェーブレットに基づく推定器は、$\rdd$ における滑らかな最適輸送写像に対して近似的にミニマックス最適性を達成する。
• 半双対最適輸送問題のための新しい安定性バインドが導出され、推定誤差の制御に用いられる。
• 補完的なミニマックス下界が確立され、推定器のレートが対数要因を除いて最適であることが示された。
• 数値実験により理論的発見が確認され、滑らかさ正則化の実用的利点が浮き彫りにされた。
• 本研究は、一般次元における滑らかな輸送写像のための最初のミニマックス推定レートを提供し、重要な理論的ギャップを埋めた。