arXiv:2506.10168stat.MLcs.LG2025-06NeurIPS被引 14

用多点约束建模复杂系统轨迹,提升长期依赖捕捉能力。

Momentum Multi-Marginal Schrödinger Bridge Matching

  • 将动态系统升维到相空间,构建多边缘条件的随机桥接模型。
  • 在真实数据上实现更平滑、更连贯的轨迹推断,收敛性显著提升。
  • 适合需要长时序建模的生物、气象等领域的轨迹推演任务。

从稀疏样本快照中推断复杂系统的轨迹是单细胞生物学、气象学和经济学等多个领域中的基础挑战。尽管桥梁匹配与流匹配框架已有进展,现有方法仍依赖相邻快照间的成对插值,难以捕捉长程时间依赖性,影响推断轨迹的一致性。为此,我们提出一种新框架——动量多边缘薛定谔桥匹配(3MSBM),通过将动力学升维至相空间,并将随机桥推广为基于多个位置约束的条件形式,形成多边缘条件随机最优控制问题。通过最小化变分目标学习底层动力学,固定多边缘条件桥生成的路径。作为匹配方法,3MSBM在训练过程中保持中间边缘分布不变,显著提升收敛速度与可扩展性。大量实验证明,3MSBM在捕捉具有时间依赖性的复杂动态方面优于现有方法,为多边缘设置下的匹配框架训练开辟了新路径。

原文摘要 · Abstract (English)

Understanding complex systems by inferring trajectories from sparse sample snapshots is a fundamental challenge in a wide range of domains, e.g., single-cell biology, meteorology, and economics. Despite advancements in Bridge and Flow matching frameworks, current methodologies rely on pairwise interpolation between adjacent snapshots. This hinders their ability to capture long-range temporal dependencies and potentially affects the coherence of the inferred trajectories. To address these issues, we introduce \textbf{Momentum Multi-Marginal Schrödinger Bridge Matching (3MSBM)}, a novel matching framework that learns smooth measure-valued splines for stochastic systems that satisfy multiple positional constraints. This is achieved by lifting the dynamics to phase space and generalizing stochastic bridges to be conditioned on several points, forming a multi-marginal conditional stochastic optimal control problem. The underlying dynamics are then learned by minimizing a variational objective, having fixed the path induced by the multi-marginal conditional bridge. As a matching approach, 3MSBM learns transport maps that preserve intermediate marginals throughout training, significantly improving convergence and scalability. Extensive experimentation in a series of real-world applications validates the superior performance of 3MSBM compared to existing methods in capturing complex dynamics with temporal dependencies, opening new avenues for training matching frameworks in multi-marginal settings.

轨迹推断随机桥接多边缘相空间

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