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

以下是论文《Schrödinger 问题 Benamou–Brenier 公式的偏微分方程方法》（作者：Garatti, Nenna, Rota Nodari 和 Tamanini）的详细技术总结。

1. 问题陈述与动机

Schrödinger 问题：
本文探讨的是起源于统计力学的变分问题——Schrödinger 问题。给定 $\mathbb{R}^n$ 上的两个概率测度 $\mu_0, \mu_1$ 和一个参数 $\varepsilon > 0$（代表温度或噪声），目标是找到连接这两个状态的最可能的布朗粒子云演化。在数学上，这被表述为相对于参考测度 $R_\varepsilon$（可逆 Wiener 过程的联合律）最小化相对熵：
$$\inf \{ H(\gamma | R_\varepsilon) : \gamma \in \Gamma(\mu_0, \mu_1) \}$$
其中 $\Gamma(\mu_0, \mu_1)$ 是边缘分布为 $\mu_0$ 和 $\mu_1$ 的传输计划集合。

现有理论的缺口：
已知 Schrödinger 问题是二次最优传输（OT）问题的熵正则化。在 OT 中，Benamou–Brenier 公式提供了一种动态重构，将静态传输成本等同于满足连续性方程的流体流动的最小动能。
对于 Schrödinger 问题，存在一个类似的动态公式，涉及Fokker–Planck 方程：
$$\partial_t \eta_t - \frac{\varepsilon}{2}\Delta \eta_t + \text{div}(v_t \eta_t) = 0$$
然而，此前对该动态公式的严格证明（特别是 Gigli 和 Tamanini 的工作 [GT20]）依赖于紧性假设：要求边缘分布 $\mu_0$ 和 $\mu_1$ 具有有界支撑。
问题所在： 许多实际应用（例如高斯模型、高斯混合模型、贝叶斯推断）涉及具有无界支撑的测度。在这些情况下，Schrödinger 势 ($f_\varepsilon, g_\varepsilon$) 及其对数 ($\varphi_t^\varepsilon, \psi_t^\varepsilon$) 可能在无穷远处呈现指数增长，使得在不进行仔细控制的情况下，标准的分部积分论证和积分号下求导无效。

2. 方法论

作者提出了一种基于偏微分方程（PDE）的方法，以弥合抽象理论与非紧性应用之间的差距。他们的策略包含三个主要支柱：

1. 放宽边缘分布假设： 他们不再假设紧支撑，而是假设边缘分布具有形式 $\mu_i = e^{-V_i} \mathcal{L}^n$，其中 $V_i \in C^2(\mathbb{R}^n)$ 满足一致的双侧 Hessian 界：
   $$C^{-1}|\xi|^2 \leq \xi \cdot D^2 V_i(x) \xi \leq C|\xi|^2$$
   此类包括高斯测度和强对数凹分布（次高斯尾部）。

2. 正则性与增长估计：

   • 他们利用近期成果（来自 [Chi+24] 和 [Con24]）证明 Schrödinger 势 $\psi_t^\varepsilon = \varepsilon \log g_t^\varepsilon$ 继承了边缘分布的 Hessian 界。
   • 他们证明 $\psi_t^\varepsilon$ 的增长至多为二次，其梯度 $\nabla \psi_t^\varepsilon$ 的增长至多为线性。
   • 关键在于，他们表明熵插值 $\varrho_t^\varepsilon$（路径的密度）具有强高斯矩，其衰减速度足以补偿势函数的多项式/指数增长。

3. 严格的 PDE 分析：

   • 积分号下求导： 他们利用推导出的矩界来证明对 $\int \psi_t^\varepsilon \varrho_t^\varepsilon dx$ 关于时间求导的合理性。
   • 分部积分（Gauss–Green）： 他们构造了特定的截断函数，并利用抛物型 Kato 不等式证明边界项在无穷远处消失，从而允许在无界域上应用 Gauss–Green 公式。
   • 通过凸性证明唯一性： 他们利用动能泛函的严格凸性和特征线法，证明了动态公式中速度场的唯一性。

3. 主要贡献

• 推广至次高斯测度： 本文将 Schrödinger 问题的 Benamou–Brenier 公式的有效性从紧支撑测度扩展到了具有次高斯尾部的广泛测度类（包括高斯分布和混合分布）。
• 几乎自包含的 PDE 证明： 与依赖概率论证或紧性的先前证明不同，这项工作提供了一个严格的、几乎自包含的证明，仅使用 PDE 工具（热核估计、Hessian 界和矩控制）。
• 显式增长控制： 作者提供了显式估计，表明即使势函数在无穷远处增长，熵插值密度的衰减也能确保动态公式成立所需的可积性。
• 唯一性结果： 他们证明了在 Fokker–Planck 约束下，动态动能作用量的最小化子是唯一的。

4. 主要结果

定理 1.1（简化版 Benamou–Brenier 公式）：
在假设 $\mu_0, \mu_1$ 为次高斯测度（由具有有界 Hessian 的 $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\}$$
其中最小值是在满足分布意义下 Fokker–Planck 方程的 $(\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)$ 给出。

推论（对偶公式）：
本文还导出了涉及 Hamilton-Jacobi-Bellman 型不等式的 Schrödinger 问题的对偶公式：
$$\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\}$$
其中最大化子是 Schrödinger 势 $\psi_t^\varepsilon$。

5. 意义与影响

• 连接理论与实践： 该结果证明了动态公式（流体动力学解释）在数据天然无界的场景中的合理性，例如高斯混合模型、机器学习中的扩散模型以及随机控制。
• 算法启示： Schrödinger 问题是Sinkhorn 算法的理论基础，而 Sinkhorn 算法是现代最优传输和机器学习的基石。验证无界支撑下的动态公式，加强了将这些算法应用于现实世界非紧数据集的理论基础。
• 分析严谨性： 通过提供通过矩估计处理无界域的以 PDE 为中心的证明，本文为分析超越限制性紧性设置的熵插值提供了一个稳健的框架，为非紧最优传输和随机分析领域的进一步研究开辟了途径。

总之，本文解决了 Schrödinger 问题理论中的一个关键性技术局限，证明了熵最小化与动能之间优雅的动态联系，对于现代应用中至关重要的广泛次高斯分布类同样成立。