arXiv:2509.00183cs.LG2025-09被引 1

用加速率直接监督训练神经微分方程,提升约束多体系统模拟精度与效率。

FNODE: Flow-Matching for data-driven simulation of constrained multibody systems

  • 直接学习加速度映射,避免反向传播求解微分方程的计算瓶颈。
  • 在多个力学系统上预测误差更低,训练和推理速度更快。
  • 适合需要高精度物理模拟的机器人、车辆建模等应用。

数据驱动的约束多体系统建模仍面临两大挑战:(i) 神经微分方程(Neural ODE)训练成本高,通常需对微分方程求解器进行反向传播;(ii) 滚动预测中误差累积严重。本文提出流匹配神经微分方程(FNODE)框架,通过直接监督加速度而非积分状态,将训练转化为监督回归问题,消除对ODE伴随/求解器反向传播的依赖。加速度目标通过结合快速傅里叶变换(FFT)与有限差分(FD)的混合方法高效获取。运动学约束通过坐标分解实现:FNODE仅学习独立广义坐标的加速度,依赖坐标由位置级约束方程求解恢复。我们在单/三质量-弹簧-阻尼系统、双摆、带/不带摩擦的滑块曲柄机构、车辆模型及小车-摆杆系统上评估了FNODE,对比MBD-NODE、LSTM及全连接基线。实验表明,FNODE在各项基准任务中均实现了更高预测精度与更优训练/运行效率,且通过坐标分解保持约束满足。代码与脚本已开源,支持复现与后续研究。

原文摘要 · Abstract (English)

Data-driven modeling of constrained multibody dynamics remains challenged by (i) the training cost of Neural ODEs, which typically require backpropagation through an ODE solver, and (ii) error accumulation in rollout predictions. We introduce a Flow-Matching Neural ODE (FNODE) framework that learns the acceleration mapping directly from trajectory data by supervising accelerations rather than integrated states, turning training into a supervised regression problem and eliminating the ODE-adjoint/solver backpropagation bottleneck. Acceleration targets are obtained efficiently via numerical differentiation using a hybrid fast Fourier transform (FFT) and finite-difference (FD) scheme. Kinematic constraints are enforced through coordinate partitioning: FNODE learns accelerations only for the independent generalized coordinates, while the dependent coordinates are recovered by solving the position-level constraint equations. We evaluate FNODE on single and triple mass-spring-damper systems, a double pendulum, a slider crank with and without friction, a vehicle model, and a cart-pole, and compare against MBD-NODE, LSTM, and fully connected baselines. Across these benchmarks, FNODE achieves improved prediction accuracy and training/runtime efficiency, while maintaining constraint satisfaction through the partitioning procedure. Our code and scripts are released as open source to support reproducibility and follow-on research.

神经微分方程物理模拟约束系统流匹配

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