$\lambda$-geodesic convexity of the squared sliced-Wasserstein distance

技術要約：二乗スライス・ワッサースタイン距離の $\lambda$-測地的凸性

問題設定
本論文は、最適輸送理論における根本的な限界に対処している。すなわち、固定されたターゲット測度 $\nu$ に対する二乗ワッサースタイン距離 $W_2^2(\cdot, \nu)$ は、次元 $d \geq 2$ のとき、確率測度の空間 $\mathcal{P}_2(\mathbb{R}^d)$ 上で一般に測地的凸性を備えていない。この関数は $d=1$ の場合には 2-測地的凸性を持つが、高次元における凸性の欠如は、勾配流（gradient flow）の解析や解の一意性を困難にしている。

高次元のワッサースタイン距離の計算コストを軽減するために、スライス・ワッサースタイン距離（$SW_2$）が導入された。これは、測度を直線へのすべての射影にわたって一次元のワッサースタイン距離を平均化することで定義される。本研究の核心的な問いは、密度が有界な測度の集合上で、二乗スライス・ワッサースタイン距離 $F_\nu(\rho) = \frac{1}{2}SW_2^2(\rho, \nu)$ が、一次元の場合と同様の $\lambda$-測地的凸性特性を持つかどうかである。

手法
著者らは、$\lambda$-測地的凸性を確立するために、最適輸送理論、幾何学的解析、および関数解析を組み合わせた手法を用いている。その手法は、以下の3つの主要な段階を経て進行する。

1. 勾配流の特定: 本論文はまず、スライス・ワッサースタイン流（SWF）――密度 $\rho$ の、ターゲット $\nu$ への等方的な連続時間進化――が、汎関数 $F_\nu$ のワッサースタイン勾配降下に対応することを確立する。この流れを駆動する速度場 $v[\rho]$ は、$F_\nu$ の第一変分（first variation）の負の勾配として明示的に導出される。
2. 速度場のリプシッツ境界: 技術的な中心戦略は、速度場 $v[\rho] + \frac{1}{d}\text{id}$ が定数 $L$ のリプシッツ連続であるならば、$F_\nu$ は $\lambda = \frac{1}{d} - L$ に対して $\lambda$-測地的凸性を持つという補題3.1に依拠している。したがって、問題は、有界な密度を持つ測度の集合上で速度場 $v[\rho]$ の一様なリプシッツ境界を導出することに帰着する。
3. 射影の解析: リプシッツ定数を抑えるために、著者らは一次元射影における最適輸送写像の微分を解析する。彼らは、輸送写像 $T_\vartheta$ の微分を、ソース密度の射影 $\rho_\vartheta$ とターゲット密度の射影 $\nu_\vartheta$ の比に関連付ける。
   • ケース1（強い仮定）: 当初、著者らはターゲットの射影が $\nu_\vartheta \geq c_0 > 0$ として一様に下から抑えられている場合を検討する。これにより直接的な推定が可能となるが、これは次元 $d \leq 3$ に実質的に限定される非常に制約の強い条件であることが示される。
   • ケース2（一般的な仮定）: 次に、ターゲット $\nu$ が特定の幾何学的条件（仮定4.6）を満たすことのみを要求する、より堅牢な結果（定理4.4）を導出する。この条件は、サポートの境界付近における $\nu$ の密度の下限を規定するものである。具体的には、$\nu_\vartheta(r) \geq \kappa (\min\{N_\vartheta(r), 1-N_\vartheta(r)\})^\alpha$ である（ここで $N_\vartheta$ は累積分布関数）。

主な貢献と結果

• 定理 4.4 (主要な結果): 本論文は、仮定 (4.6) を満たすターゲット測度 $\nu$ と、集合 $K = \{\rho \in \mathcal{P}_2(\mathbb{R}^d) : \|\rho\|_{L^\infty} \leq M\}$ に対して、汎関数 $F_\nu$ が $K$ 上で $\lambda$-測地的凸であることを証明している。凸性の係数（modulus）は以下のように与えられる：
  $$\lambda = \frac{1}{d} - \frac{c(d)}{(1 - \frac{\alpha d}{d-1})\kappa} M^{1/d}$$
  ここで $c(d)$ は次元のみに依存する定数であり、$\kappa$ と $\alpha$ は仮定 (4.6) におけるターゲット測度 $\nu$ を特徴付けるパラメータである。著者らは、$\lambda$ は一般に非正であり、密度境界 $M$ が増加するにつれて減少すること、すなわち、高密度な測度に対して凸性が退化することを指摘している。
• ターゲット測度に関する幾何学的条件: 論文は仮定 (4.6) の詳細な解析を提供しており、滑らかで一様に凸な領域上で密度がゼロから離れて下から抑えられている測度（例 4.10）、およびサポートの境界付近で特定の冪乗則挙動を持つ測度（例 4.8）によって、この仮定が満たされることを示している。
• スライス・ワッサースタイン流の一意性: $\lambda$-測地的凸性の結果から直接的な帰結として、著者らはスライス・ワッサースタイン流のウェルポーズドネス（well-posedness）を確立する。具体的には、系 1.1 は、$\nu$ が (4.6) を満たし、初期データ $\rho_0$ が $L^\infty$ に属する場合、フロー方程式 (1.3) に対する一意の解 $\{\rho_t\}_{t \geq 0}$ が存在し、$L^\infty$ ノルムが局的に有界に保たれることを述べている。
• 次元の制約: 「一様に有界な射影」という条件の幾何学的な限界を、注釈 4.2 を通じて明確にし、そのような測度は本質的に次元 $d \leq 3$ においてのみ存在することを示している。

意義と主張
本論文の主要な意義は、計算効率の高さから機械学習やデータサイエンスで広く用いられている指標であるスライス・ワッサースタイン距離に対して、測地的凸性の理論を拡張した点にあると著者らは主張している。現実的な密度境界の下で $\lambda$-測地的凸性を証明することにより、著者らは、スライス・ワッサースタイン流に対して、既存のロバストなワッサースタイン勾配流の理論（[AGS05] や [San17] 等の研究によるもの）を適用することを可能にしている。

具体的には、この結果が以下のことをもたらすと強調している：

1. SWF 解の一意性を保証すること（これは、これまでの一般的な条件下では確立されていなかった性質である）。
2. 離散的なジョルダン・キンダーライファー・オットー（JKO）スキームの、連続的なフロー解への収束率を意味すること。
3. 将来的な $SW_2$ に関する研究において、**フロー交換テクニック（flow interchange technique）**の使用を可能にすること。

著者らは、自身の凸性に関する結果の範囲について謙虚な姿勢を崩しておらず、$\lambda$ がしばしば負であること（標準的な意味での弱い凸性、あるいは凹性さえも示唆する）、および凸性の係数が密度境界 $M$ の増大に伴って劣化することを明示している。彼らは新しいアルゴリズムや実験的な検証を提案するのではなく、既存のスライス輸送法の挙動を厳密に解析するために必要な理論的基礎を提供することを目指している。