摘要
高维动力系统预测是地球科学和工程等众多领域中的一个基本挑战。神经常微分方程 (NODE) 结合了神经网络和数值求解器的优势,成为预测复杂非线性动力系统的一种很有前景的算法。然而,用于 NODE 训练的经典技术对于学习混沌动力系统是无效的。在这项工作中,我们提出了一种新颖的 NODE 训练方法,它允许对混沌动力系统进行稳健的学习。我们的方法解决了与潜在的混沌动力学相关的非凸性和梯度爆炸的挑战。来自此类系统的训练数据轨迹被分成多个不重叠的时间窗口。除了与训练数据的偏差外,优化损失项还进一步惩罚了时间窗口之间预测轨迹的不连续性。窗口大小根据系统的最快 Lyapunov 时间尺度选择。多步惩罚 (MP) 方法首先在 Lorenz 方程上进行演示,以说明它如何改善损失景观并因此加速优化收敛。MP 方法可以以类似于最小二乘阴影的方式优化混沌系统,但计算成本显着降低。我们提出的算法,称为多步惩罚 NODE,被应用于混沌系统,例如 Kuramoto-Sivashinsky 方程、二维 Kolmogorov 流和 ERA5 大气再分析数据。观察到 MP-NODE 为此类混沌系统提供了可行的性能,不仅用于短期轨迹预测,而且用于作为这些动力学混沌性质标志的不变统计数据。