精读笔记

Problem Setting

A Unified Variational Framework for Optimal Transport with Lagrangian Costs（arXiv preprint / 2026-07-20）

这篇论文解决的不是“如何定义一般 Lagrangian cost 的 OT”，这个在 Kantorovich + action minimization 里已经很自然；它真正要解决的是：当 cost 由 c(x,y)=inf_ω∫L(ω,ωdot)dt 给出时，Benamou-Brenier 型动态 formulation、Kantorovich formulation、Hamilton-Jacobi dual、以及 Wasserstein 上的 Hamiltonian flow 是否可以被看成同一个对象的不同坐标。

困难点在于二次 cost 的特殊结构不再存在。W2 中 |v|^2 带来的 Hilbert 型切空间、速度-势函数关系 v=∇φ、动量凸化以及 Hamilton-Jacobi 方程都非常干净；一般 L(x,v) 下，速度到动量的映射变成 Legendre transform，cost 可位置依赖，geodesic 未必有对称 metric 结构，HJ 方程也可能只在 viscosity 意义下成立。以前方法卡在各自领域内：OT theory 有静态 dual，BB 有二次动态凸优化，Tonelli theory 有 action minimizer，Wasserstein Hamiltonian flow 有形式 PDE，但缺少明确的等价链条。

Motivation

已有路线“不够”的地方不是表达能力，而是互相之间的接口不清楚。对于一般 Lagrangian cost，Kantorovich formulation 可以直接写，但它不暴露连续时间输运结构；Eulerian formulation 可以模仿 BB 写，但 ρL(x,v) 对 (ρ,v) 不凸；Hamilton-Jacobi 方程看起来是自然对偶，但它和 Kantorovich endpoint potentials 的关系需要证明；Hamiltonian flow 形式也容易写，但是否真来自同一个 variational principle 并不自动成立。

作者的核心观察是：一旦把速度变量换成 momentum m=ρv，Lagrangian action density 变成 perspective form ρL(x,m/ρ)，它正好是凸分析里能和 Hamiltonian H 做 Fenchel dual 的对象。也就是说，缺口不是“缺一个新模型”，而是缺一个把 action minimization、mass conservation 和 convex duality 对齐的变量选择。

Core Idea

论文的核心思想是把一般 Lagrangian OT 从“transport plan over endpoints”提升到“density-momentum path optimization”，再通过 Legendre duality 降回 Hamilton-Jacobi endpoint dual。这个循环让 Kantorovich plan、粒子轨道、Eulerian PDE 和 Hamiltonian system 成为同一最优性条件的不同投影。

本质区别在于：prior 中二次 cost 的 BB formulation 依赖 L(v)=|v|^2 的特殊几何，而这里把可迁移部分抽象成“L 对 v 凸 + perspective convexification + Hamiltonian Legendre dual”。因此它改变的是建模坐标：不再把 Wasserstein-2 的 Riemannian structure 当作模板，而是把 Lagrangian action 本身作为生成 transport geometry 的原语。这个 inductive bias 更一般，因为它允许位置依赖、各向异性、机械系统和控制系统 cost 进入同一套 Eulerian/dual 机制。

Method

第一，点对点 cost 由 Tonelli Lagrangian 的最小作用量定义。它解决的是 endpoint cost 的生成问题：不是手工指定 c(x,y)，而是通过 underlying dynamics 生成。这样 cost 内部已经编码了动力学可达性、势能、各向异性或控制代价。

第二，Eulerian formulation 用 continuity equation 约束密度和速度。它解决的是从静态 coupling 到时间演化的转化问题，使输运过程能被写成场变量优化。但直接使用 (ρ,v) 时目标一般非凸，因此只是自然表述，不是最适合证明和计算的表述。

第三，momentum formulation m=ρv 是关键机制。ρL(x,m/ρ) 是 L 的 perspective extension；只要 L(x,·) 凸，这个 integrand 对 (ρ,m) 凸。它把问题从流体力学形式转成凸优化形式，这也是后续 duality 成立的技术支点。

第四，Hamilton-Jacobi dual 来自 continuity equation 的 multiplier φ。Legendre transform 把局部约束写成 ∂tφ+H(x,∇φ)≤0，endpoint objective 则变成 ∫φ(1)ρ1-φ(0)ρ0。它解决的是 optimality certificate 问题：一个可行 φ 同时给出 Kantorovich dual lower bound 和动态 action 的对偶证书。

第五，Hamiltonian PDE system 是最优性条件，不是额外模型。饱和条件给出 ∂tφ+H(x,∇φ)=0，速度由 v=∇pH(x,∇φ) 给出，密度满足 ∂tρ+div(ρ∇pH)=0。这个系统把 OT geodesic 解释为 Wasserstein 空间上的 Hamiltonian flow。

Key Insight / Why It Works

最重要的 insight 是：一般 Lagrangian cost 的动态 OT 等价性不是靠 W2 的二次结构，而是靠 perspective convexity 和 Legendre duality。二次 Wasserstein 只是 L(v)=|v|^2 时的一个特例；真正可迁移的结构是 ρL(m/ρ) 与 H(∇φ) 的 Fenchel 配对。

方法有效的原因有三层。第一，continuity equation 是把 microscopic trajectories 聚合成 macroscopic density 的守恒律，因此 Eulerian 解和路径族之间存在自然 correspondence。第二，Tonelli 假设保证单粒子 action cost 良定，使 Kantorovich endpoint cost 能和 trajectory action 对齐。第三，convexity 允许无 gap duality，把“最小 action”转成“最大 HJ subsolution endpoint gap”。这三者同时成立时，多个 formulation 才是同一个 variational object。

最核心贡献是组织性和结构性：它把已有的几个成熟对象摆成一个闭合 diagram，并指出 Hamilton-Jacobi/continuity system 是 general Lagrangian OT 的自然最优性系统。相对而言，例子部分和 Neural ODE/ML 连接更像应用动机扩展，技术增益不清；这些例子没有提供新的 theorem，也没有证明该框架在机器学习任务上带来新的能力。

这篇不是 scaling、retrieval、data coverage 或 benchmark engineering 的工作；它是理论重参数化和 convex duality 框架。若说“增益”，主要来自 better variational coordinates，而不是新的估计或算法。需要警惕的是，文中不少等价证明在 smooth regime 下比较形式化，尤其是 duality gap、HJ 正则性和 Hamiltonian formulation 的势函数表示，严格数学上可能需要更重的 weak solution machinery。

Relation To Prior Work

最接近的谱系有四条：Benamou-Brenier dynamic OT、Kantorovich duality for general costs、Tonelli/weak KAM/Mather action theory、Wasserstein Hamiltonian flows。论文的新意不是发明这些对象，而是把它们通过一般 Lagrangian L(x,v) 的 Legendre structure 统一起来。

和 Benamou-Brenier 的本质差异是：BB 的 convexity 来自 |m|^2/ρ，本文把它推广为 ρL(x,m/ρ)。这不是简单替换符号，因为 Hamiltonian dual constraint 从 ∂tφ+|∇φ|^2≤0 变成 ∂tφ+H(x,∇φ)≤0，速度-势关系也从 v=∇φ 变成 v=∇pH(x,∇φ)。

和一般 Kantorovich cost theory 的差异是：Kantorovich 只看 endpoint coupling，而本文强调 cost 的 action origin，因此能引入连续时间 PDE、Hamiltonian flow 和控制解释。和 weak KAM / optimal control 的差异是：那些理论主要处理单粒子 value/action；本文把一族粒子的分布演化和 OT coupling 接起来。

看似新的地方中，Hamilton-Jacobi dual、Legendre transform、momentum convexification 都是已有思想；实质新增在于把这些思想用于一般 Lagrangian cost 的 OT，并给出一套统一表述。它更像 framework consolidation，而不是突破某个单点技术瓶颈。

Dataset / Evaluation

这篇没有实验数据集，也没有传统 benchmark evaluation。第 4 节的 examples 只是验证框架覆盖范围：quadratic cost 回到 W2，mechanical Lagrangian 接 Hamiltonian mechanics，translation-invariant convex cost 接 h(x-y)，anisotropic quadratic cost 接 weighted Wasserstein，optimal control / Neural ODE 说明 action cost 的建模解释。

这些 examples 支持“框架覆盖多个已知对象”的 claim，但不支持任何算法效率、数值稳定性、真实任务性能或 scalability claim。论文结尾提到可启发 particle methods、PDE discretization、convex optimization、neural variational solvers，但没有实际验证。若把本文理解为理论 note，这是合理的；若把它理解为 computational OT 贡献，则 evidence 明显不足。

Limitation

第一，成立条件偏理想化。smooth strictly positive densities、Tonelli Lagrangian、强凸性、周期 domain、smooth flow 等假设让推导干净，但实际 OT 常遇到 singular measures、density vacuum、非光滑 cost、边界、不可达控制系统和弱解。文中未充分说明这些情形下哪些等价仍然成立。

第二，Hamilton-Jacobi 部分偏形式。一般 HJ 方程自然应在 viscosity solution 意义下处理，而文中主要用 C1 potential 做推导。对偶约束 ∂tφ+H≤0 与 Kantorovich dual 的严格闭包关系、正则化需求、以及 endpoint potential 的可达性都没有充分展开。

第三，I6 的 Hamiltonian formulation 可能是最脆弱的一环。把任意 admissible momentum m 转成 ρ∇pH(x,∇φ) 形式，本质上是在做非线性加权投影；在非二次 H 下，这个表示是否总能以论文所需强度成立并不显然。文中未充分说明 elliptic problem 的可解性、唯一性、规范条件和时间正则性。

第四，论文把很多已有结构统一起来，但新的数学增量有限。它没有给出新的 sharp existence theorem、regularity theorem、metric geometry theorem 或算法收敛结果。增益来源主要是 formulation unification，而不是解决一般 Lagrangian OT 中最难的分析问题。

第五，所谓“distance”需要谨慎。一般 L(x,v) 若非偶、非正、含 potential 项，诱导的 c(x,y) 未必对称，也未必满足 metric 的所有性质。文中有时使用 transport distance 的语言，但对何时真是 metric 没有系统区分。

Takeaway

• 1. 真正值得记住的是变量选择：一般 Lagrangian OT 的动态凸化核心是 m=ρv 与 perspective integrand ρL(x,m/ρ)，这比模仿 W2 的 Riemannian 形式更本质。
• 2. Hamilton-Jacobi dual 不是附属解释，而是 Kantorovich dual 在 action-generated cost 下的动态版本；HJ subsolution 给出的 endpoint gap 正是 transport cost 的对偶证书。
• 3. 这篇把 OT、optimal control、Hamiltonian mechanics、Wasserstein geometry 的接口整理得比较清楚，适合作为后续做 generalized dynamic OT 或 control-induced transport metric 的理论 scaffold。
• 4. 未来真正有价值的推进不在继续列更多 examples，而在弱解层面补齐等价、处理 vacuum / singular measures、明确 metric 条件，并从 convex formulation 导出可计算算法及收敛保证。

一句话总结

这篇论文的定位是一篇 general Lagrangian cost 下的 Benamou-Brenier / Kantorovich / Hamilton-Jacobi / Wasserstein-Hamiltonian 统一化理论笔记，真正贡献是用 perspective convexity 和 Legendre duality重组已有结构，而不是提出新的算法或新的实验能力。