核心概念
本文提出了一種新的神經常微分方程(NODE)訓練方法,可以有效地學習混沌動力系統。該方法通過在訓練數據軌跡中引入多個非重疊的時間窗口,並在優化損失函數中加入預測軌跡在時間窗口之間的不連續性,來解決混沌動力系統中非凸性和爆炸梯度的挑戰。
摘要
本文提出了一種新的神經常微分方程(NODE)訓練方法,名為多步驟懲罰神經常微分方程(MP-NODE),用於有效學習混沌動力系統。
主要內容如下:
-
混沌動力系統的建模和預測是一個基本的挑戰,因為這些系統對初始條件的微小變化非常敏感,導致預測結果隨時間的偏離。
-
傳統的NODE訓練方法在學習混沌動力系統時效果不佳,因為它們面臨非凸性和爆炸梯度的問題。
-
本文提出了MP-NODE方法,通過在訓練數據軌跡中引入多個非重疊的時間窗口,並在優化損失函數中加入預測軌跡在時間窗口之間的不連續性,來解決上述問題。
-
在洛倫茲方程、庫拉莫-西瓦辛斯基方程、二維科爾莫戈洛夫流和ERA5大氣再分析數據等混沌系統上的實驗結果表明,MP-NODE不僅可以準確預測短期軌跡,還能很好地捕捉系統的不變統計特性。
-
相比於傳統的NODE,MP-NODE在優化收斂和長期預測穩定性方面都有顯著改善。這為在各種科學領域構建穩定的混沌系統預測模型奠定了基礎。
统计
在洛倫茲方程中,使用多步驟懲罰優化可以將目標函數梯度的幅度從O(10^100)降低到O(10^3)。
在庫拉莫-西瓦辛斯基方程中,MP-NODE可以在30個Lyapunov時間尺度內保持系統的不變統計特性,而標準NODE在很短的時間內就會偏離。
在二維科爾莫戈洛夫流中,MP-NODE可以更好地捕捉渦動能量頻譜在慣性區間的特徵,並在長期預測中保持較高的相關性。
在ERA5大氣再分析數據中,MP-NODE不僅可以在14天內提供更準確的天氣預報,還能更好地捕捉長期氣候的季節性變化。
引用
"本文提出了一種新的神經常微分方程(NODE)訓練方法,可以有效地學習混沌動力系統。"
"該方法通過在訓練數據軌跡中引入多個非重疊的時間窗口,並在優化損失函數中加入預測軌跡在時間窗口之間的不連續性,來解決混沌動力系統中非凸性和爆炸梯度的挑戰。"