arXiv:2506.18604cs.LGcs.AI2025-06被引 3

无需模拟即可训练连续时间扩散过程,适用于广泛问题。

Simulation-Free Differential Dynamics through Neural Conservation Laws

  • 联合建模概率路径与生成它的动力学,直接嵌入福克-普朗克方程
  • 可在无模拟情况下训练,支持生成建模、最优控制等多种目标
  • 适合需要精确密度函数的场景,如种群数据建模与时空事件分析

我们提出一种全新的无模拟框架,用于在非常通用的目标函数上训练连续时间扩散过程。现有方法通常需预设最优扩散过程(仅适用于受限问题),或依赖昂贵的数值模拟来获取时变密度并采样。相比之下,我们采用耦合参数化,联合建模时变密度函数(概率路径)与生成该路径的扩散动力学。通过扩展并大幅简化神经守恒律的构造,将福克-普朗克方程和密度函数要求作为硬约束直接嵌入模型中。这使得我们能在无需模拟的情况下,对多种问题形式进行训练,包括生成建模与动态最优传输中的数据驱动目标,以及随机最优控制中的最优性目标,并因可直接获得精确密度函数而轻松扩展至均场目标。我们在建模时空事件、从群体数据学习最优动力学等多样化应用场景中验证了方法的有效性。

原文摘要 · Abstract (English)

We present a novel simulation-free framework for training continuous-time diffusion processes over very general objective functions. Existing methods typically involve either prescribing the optimal diffusion process -- which only works for heavily restricted problem formulations -- or require expensive simulation to numerically obtain the time-dependent densities and sample from the diffusion process. In contrast, we propose a coupled parameterization which jointly models a time-dependent density function, or probability path, and the dynamics of a diffusion process that generates this probability path. To accomplish this, our approach directly bakes in the Fokker-Planck equation and density function requirements as hard constraints, by extending and greatly simplifying the construction of Neural Conservation Laws. This enables simulation-free training for a large variety of problem formulations, from data-driven objectives as in generative modeling and dynamical optimal transport, to optimality-based objectives as in stochastic optimal control, with straightforward extensions to mean-field objectives due to the ease of accessing exact density functions. We validate our method in a diverse range of application domains from modeling spatio-temporal events to learning optimal dynamics from population data.

扩散模型连续时间无模拟最优控制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。