A PDE approach to Benamou--Brenier formula for the Schrödinger problem

以下は、Garatti、Nenna、Rota Nodari、Tamanini による論文「A PDE approach to Benamou–Brenier formula for the Schrödinger problem」の詳細な技術的要約です。

1. 問題の定義と動機

シュレーディンガー問題:
本論文は、統計力学に由来する変分問題であるシュレーディンガー問題を取り扱います。$\mathbb{R}^n$ 上の 2 つの確率測度 $\mu_0, \mu_1$ と、パラメータ $\varepsilon > 0$（温度またはノイズを表す）が与えられたとき、これら 2 つの状態を結ぶブラウン粒子雲の最も確からしい進化を見つけることが目的です。数学的には、参照測度 $R_\varepsilon$（可逆ウィーナー過程の結合法則）に対する相対エントロピーを最小化することとして定式化されます：
$$\inf \{ H(\gamma | R_\varepsilon) : \gamma \in \Gamma(\mu_0, \mu_1) \}$$
ここで、$\Gamma(\mu_0, \mu_1)$ は周辺分布が $\mu_0$ と $\mu_1$ である輸送計画の集合です。

既存理論のギャップ:
シュレーディンガー問題は、二次の最適輸送（OT）問題のエントロピー正則化であることが知られています。OT において、ベナム＝ブレニエの公式は動的な再定式化を提供し、静的な輸送コストを、連続の方程式を満たす流体流れの最小運動エネルギーと等しくします。
シュレーディンガー問題においても、フォッカー＝プランク方程式を含む同様の動的定式化が存在します：
$$\partial_t \eta_t - \frac{\varepsilon}{2}\Delta \eta_t + \text{div}(v_t \eta_t) = 0$$
しかし、この動的公式の以前の厳密な証明（特に Gigli と Tamanini による [GT20]）は、コンパクト性の仮定に依存していました：周辺分布 $\mu_0$ と $\mu_1$ は有界な台を持つことが要求されていました。
問題点: 多くの実用的な応用（例えば、ガウスモデル、ガウス混合モデル、ベイズ推論）は、非有界な台を持つ測度を扱います。これらの場合、シュレーディンガーポテンシャル（$f_\varepsilon, g_\varepsilon$）およびその対数（$\varphi_t^\varepsilon, \psi_t^\varepsilon$）は無限遠で指数関数的な増大を示す可能性があり、慎重な制御なしには標準的な部分積分の議論や積分記号下の微分が有効ではなくなります。

2. 手法

著者らは、抽象理論と非コンパクトな応用の間のギャップを埋めるためのPDE ベースのアプローチを提案します。彼らの戦略は以下の 3 つの主要な柱から成り立ちます：

1. 周辺分布の仮定の緩和: 有界な台の代わりに、周辺分布が $\mu_i = e^{-V_i} \mathcal{L}^n$ の形であると仮定します。ここで、$V_i \in C^2(\mathbb{R}^n)$ は一様な両側ヘッセ行列の境界を満たします：
   $$C^{-1}|\xi|^2 \leq \xi \cdot D^2 V_i(x) \xi \leq C|\xi|^2$$
   このクラスには、ガウス測度および強対数凹分布（サブガウス型テール）が含まれます。

2. 正則性と増大の推定:

   • 彼らは最近の結果（[Chi+24] および [Con24] から）を活用し、シュレーディンガーポテンシャル $\psi_t^\varepsilon = \varepsilon \log g_t^\varepsilon$ が周辺分布からヘッセ行列の境界を継承することを確立します。
   • $\psi_t^\varepsilon$ は高々二次関数的に増大し、その勾配 $\nabla \psi_t^\varepsilon$ は高々線形的に増大することを証明します。
   • 決定的な点として、エントロピー補間 $\varrho_t^\varepsilon$（経路の密度）が、ポテンシャルの多項式/指数関数的な増大を相殺するのに十分な速さで減衰する強いガウスモーメントを持つことを示します。

3. 厳密な PDE 解析:

   • 積分記号下の微分: 導出されたモーメントの境界を用いて、$\int \psi_t^\varepsilon \varrho_t^\varepsilon dx$ を時間に関して微分することを正当化します。
   • 部分積分（ガウス＝グリーン）: 特定の切断関数を構成し、放熱型カトの不等式を用いて、無限遠での境界項が消滅することを証明し、非有界領域に対するガウス＝グリーンの公式の適用を可能にします。
   • 凸性による一意性: 運動エネルギー汎関数の狭義凸性と特性曲線法を用いて、動的定式化における速度場の一意性を確立します。

3. 主要な貢献

• サブガウス測度への拡張: 本論文は、シュレーディンガー問題に対するベナム＝ブレニエの公式の有効性を、コンパクトな台を持つ測度から、サブガウス型テール（ガウスおよび混合を含む）を持つ広範なクラスの測度へと拡張します。
• ほぼ完結した PDE 証明: 確率的な議論やコンパクト性に依存した以前の証明とは異なり、この仕事は PDE ツール（熱核推定、ヘッセ行列の境界、モーメント制御）のみを用いた、厳密でほぼ完結した証明を提供します。
• 明示的な増大制御: ポテンシャルが無限遠で増大する場合でも、エントロピー補間密度の減衰が、動的定式化が成り立つために必要な可積分性を保証することを示す明示的な推定を提供します。
• 一意性の結果: フォッカー＝プランク制約の下で、動的運動作用の最小化子が一意であることを証明します。

4. 主要な結果

定理 1.1（簡略化されたベナム＝ブレニエの公式）:
$\mu_0, \mu_1$ が有界なヘッセ行列を持つ $C^2$ ポテンシャルによって定義されるサブガウス測度であるという仮定の下、すべての $\varepsilon > 0$ に対して：
$$\varepsilon \min_{\gamma \in \Gamma(\mu_0, \mu_1)} H(\gamma | R_\varepsilon) = \varepsilon H(\mu_0 | \mathcal{L}^n) + \min \left\{ \int_0^1 \int_{\mathbb{R}^n} \frac{|v_t|^2}{2} \eta_t \, dx dt \right\}$$
ここで、最小値は分布の意味でフォッカー＝プランク方程式を満たすペア $(\eta_t, v_t)$ に対して取られます：
$$\partial_t \eta_t - \frac{\varepsilon}{2}\Delta \eta_t + \text{div}(v_t \eta_t) = 0, \quad \eta_0 = \mu_0, \quad \eta_1 = \mu_1$$
運動エネルギーの一意な最小化子は、エントロピー補間経路 $(\mu_t^\varepsilon, \nabla \psi_t^\varepsilon)$ によって与えられます。

系（双対定式化）:
本論文はまた、ハミルトン＝ヤコビ＝ベルマン型の不等式を含むシュレーディンガー問題の双対定式化を導出します：
$$\max \left\{ \int f_1 d\mu_1 - \int f_0 d\mu_0 : \partial_t f + \frac{\varepsilon}{2}\Delta f + \frac{1}{2}|\nabla f|^2 \leq 0 \right\}$$
ここで、最大化子はシュレーディンガーポテンシャル $\psi_t^\varepsilon$ です。

5. 意義と影響

• 理論と実践の架け橋: この結果は、ガウス混合モデル、機械学習における拡散モデル、確率制御など、データが本質的に非有界であるような設定において、動的定式化（流体力学的解釈）の使用を正当化します。
• アルゴリズムへの含意: シュレーディンガー問題は、現代の最適輸送と機械学習の要であるシンクホルンアルゴリズムの理論的基盤です。非有界な台に対する動的定式化の妥当性を検証することは、これらのアルゴリズムを実世界の非コンパクトなデータセットに適用するための理論的基盤を強化します。
• 解析的な厳密性: モーメント推定を通じて非有界領域を処理する PDE 中心の証明を提供することで、本論文は、制限的なコンパクトな設定を超えてエントロピー補間を解析するための堅牢な枠組みを提供し、非コンパクトな最適輸送および確率解析におけるさらなる研究への道を開きます。

要約すると、本論文はシュレーディンガー問題の理論における重要な技術的制限を解決し、エントロピー最小化と運動エネルギーの間のエレガントな動的なつながりが、現代の応用に不可欠な広範なサブガウス分布のクラスに対して成り立つことを証明しています。