用物理定律提升轨迹预测,UDE比NODE更省数据。
Forecasting N-Body Dynamics: A Comparative Study of Neural Ordinary Differential Equations and Universal Differential Equations
- 将物理规律嵌入神经网络,用微分方程建模天体运动。
- UDE仅需20%数据就可准确预测,NODE需90%。
- 适合关注数据效率与物理可解释性的研究者。
n体问题在天体物理学中至关重要,用于模拟n个物体在相互引力作用下的运动。传统机器学习模型依赖大量数据且为黑箱,忽视物理规律,缺乏可解释性。而科学机器学习(Scientific ML)将已知物理定律直接融入机器学习框架。本文基于Julia语言,采用神经常微分方程(NODEs)和通用微分方程(UDEs)对系统动态进行预测与预报。关键分析环节是确定预报失效点——即模型准确预测未来未见数据所需的最小训练数据量。通过合成的含噪数据模拟真实观测限制,结果表明:UDE模型数据效率极高,仅需20%数据即可正确预报;而神经ODE需90%数据才能达到相同效果。
原文摘要 · Abstract (English)
The n body problem, fundamental to astrophysics, simulates the motion of n bodies acting under the effect of their own mutual gravitational interactions. Traditional machine learning models that are used for predicting and forecasting trajectories are often data intensive black box models, which ignore the physical laws, thereby lacking interpretability. Whereas Scientific Machine Learning ( Scientific ML ) directly embeds the known physical laws into the machine learning framework. Through robust modelling in the Julia programming language, our method uses the Scientific ML frameworks: Neural ordinary differential equations (NODEs) and Universal differential equations (UDEs) to predict and forecast the system dynamics. In addition, an essential component of our analysis involves determining the forecasting breakdown point, which is the smallest possible amount of training data our models need to predict future, unseen data accurately. We employ synthetically created noisy data to simulate real-world observational limitations. Our findings indicate that the UDE model is much more data efficient, needing only 20% of data for a correct forecast, whereas the Neural ODE requires 90%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。