用物理最小作用量原理设计更灵活的生成路径,提升模型动态表现。
Lagrangian Flow Matching: A Least-Action Framework for Principled Path Design

- 基于拉格朗日力学构建概率路径与速度场,通过最小化作用量确定演化轨迹。
- 理论证明等价于静态最优传输问题,支持无模拟训练,涵盖多种路径形式。
- 可灵活设计新路径,实验显示生成动态更合理,性能媲美现有最优方法。
流匹配通过回归目标速度场来训练神经速度场,该速度场将简单初始分布与数据分布连接起来。核心设计在于路径的选择。现有方法如修正路径和基于最优传输的路径,使样本沿端点间的直线运动,仅覆盖有限的动力学类型。我们观察到这对应经典力学中最简的最小作用量原理——动能拉格朗日量产生自由粒子的直线轨迹。基于此,提出拉格朗日流匹配:一种基于物理的作用量框架,通过在连续性方程和给定端点约束下最小化一般拉格朗日量,同时确定概率路径与速度场。我们证明该动力学问题存在等价的静态最优传输(OT)形式,从而导出一系列无需模拟的训练目标。其中,动能情形恢复了基于最优传输的流匹配,谐振子情形对应三角函数保持方差的扩散路径。更一般的拉格朗日量可生成新的概率路径与速度场,数值实验表明其显著改变学习动态,且性能与现有条件流匹配模型相当。
原文摘要 · Abstract (English)
Flow matching trains a neural velocity field by regression against a target velocity associated with a prescribed probability path connecting a simple initial distribution to the data distribution. A central design choice is the path itself. Existing constructions, including rectified and optimal-transport-based paths, transport samples along straight lines between coupled endpoints and thus cover only a narrow class of dynamics. We observe that this corresponds to the simplest case of the least-action principle in classical mechanics, in which the kinetic Lagrangian yields free-particle straight-line trajectories. Building on this observation, we propose Lagrangian flow matching, a physics-based framework in which the probability path and velocity field are determined by minimizing the action of a general Lagrangian subject to the continuity equation and the prescribed endpoints. We show that this dynamic problem admits an equivalent static optimal transport (OT) formulation, yielding a family of simulation-free training objectives that recover OT-based flow matching as the kinetic special case and the trigonometric variance-preserving diffusion path as the harmonic-oscillator case. More general Lagrangians give rise to new probability paths and velocity fields, and numerical experiments show that they induce meaningful changes in the learned dynamics while remaining competitive with existing conditional flow matching models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。