通过自适应权重稳定神经微分方程求解,提升长期模拟精度与效率
Jacobian-Adaptive Weighting for Stability: Enhancing Long-term Rollout of Neural Partial Differential Equation Solvers via Spatially-Adaptive Regularization
- 根据局部物理复杂度动态调整正则化强度,平滑区抑制噪声,突变区保留梯度
- 在1D黏性伯格斯方程和2D雷诺数400流场上实现长时稳定,误差显著降低
- 内存消耗大幅减少,适合工程级大规模流动场的长期仿真
数据驱动的代理模型可显著加速连续动力系统仿真,但自回归时间步中误差累积常导致谱爆炸和非物理解。现有全局正则化方法虽能强制收缩动力学,却均匀抑制高频特征,造成过度平滑;而长时程轨迹优化受内存瓶颈严重制约。本文提出JAWS(Jacobian-Adaptive Weighting for Stability),将算子学习重构为具有空间异方差不确定性的最大后验估计问题,使正则化强度随局部物理复杂度自动调节:在平滑区域强制收缩以抑制噪声,在如激波等奇异特征附近放松约束以保留梯度信息。实验表明,JAWS作为有效的谱预条件器,使短时程、低内存训练达到长时程基线精度。在1D黏性伯格斯方程及2D绕圆柱流动($ ext{Re}=400$,分布外泛化)上的验证证实该方法在长期稳定性、物理守恒性保持与计算效率方面的优势。内存使用量显著降低,特别适用于实际工程中大规模流场的稳定高效长期模拟。
原文摘要 · Abstract (English)
Data-driven surrogate models can significantly accelerate the simulation of continuous dynamical systems, yet the step-wise accumulation of errors during autoregressive time-stepping often leads to spectral blow-up and unphysical divergence. Existing global regularization techniques can enforce contractive dynamics but uniformly damp high-frequency features, causing over-smoothing; meanwhile, long-horizon trajectory optimization methods are severely constrained by memory bottlenecks. This paper proposes Jacobian-Adaptive Weighting for Stability (JAWS), which reformulates operator learning as a Maximum A Posteriori (MAP) estimation problem with spatially heteroscedastic uncertainty, enabling the regularization strength to adapt automatically based on local physical complexity: enforcing contraction in smooth regions to suppress noise while relaxing constraints near singular features such as shocks to preserve gradient information. Experiments demonstrate that JAWS serves as an effective spectral pre-conditioner for trajectory optimization, allowing short-horizon, memory-efficient training to match the accuracy of long-horizon baselines. Validations on the 1D viscous Burgers' equation and 2D flow past a cylinder ($\text{Re}=400$ out-of-distribution generalization) confirm the method's advantages in long-term stability, preservation of physical conservation properties, and computational efficiency. This significant reduction in memory usage makes the method particularly well-suited for stable and efficient long-term simulation of large-scale flow fields in practical engineering applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。