用改进的拉格朗日神经网络学习复杂物理系统的相对论测地线运动。
Learning Relativistic Geodesics and Chaotic Dynamics via Stabilized Lagrangian Neural Networks
- 引入海森正则化和物理感知坐标缩放,解决拉格朗日神经网络训练不稳问题。
- 在双摆系统上验证损失降低96.6%,稳定性提升90.68%,成功训练三重摆系统。
- 首次从轨迹数据直接学习反德西特时空下的相对论测地线拉格朗日量。
拉格朗日神经网络(LNNs)可从轨迹数据中学习任意拉格朗日量,但其特殊优化目标导致显著训练不稳定性,限制了在复杂系统中的应用。本文提出多项改进:基于速度二阶导数的海森正则化,抑制拉格朗日量中非物理解;设计更适合学习拉格朗日量的激活函数;引入物理感知坐标缩放以提升稳定性。系统评估表明,改进架构在双摆系统中实现验证损失降低96.6%、稳定性提升90.68%,并成功训练三重摆等复杂系统。进一步扩展正则化以惩罚洛伦兹性违反,在AdS₄时空度规下直接从轨迹数据学习到相对论测地线拉格朗日量,据我们所知为文献首例。这为物理中几何结构的自动发现开辟新路径,包括从测地轨迹提取时空度量张量分量。尽管仍需可逆海森矩阵,但显著拓展了LNN在科学发现中的实用性。
原文摘要 · Abstract (English)
Lagrangian Neural Networks (LNNs) can learn arbitrary Lagrangians from trajectory data, but their unusual optimization objective leads to significant training instabilities that limit their application to complex systems. We propose several improvements that address these fundamental challenges, namely, a Hessian regularization scheme that penalizes unphysical signatures in the Lagrangian's second derivatives with respect to velocities, preventing the network from learning unstable dynamics, activation functions that are better suited to the problem of learning Lagrangians, and a physics-aware coordinate scaling that improves stability. We systematically evaluate these techniques alongside previously proposed methods for improving stability. Our improved architecture successfully trains on systems of unprecedented complexity, including triple pendulums, and achieved 96.6\% lower validation loss value and 90.68\% better stability than baseline LNNs in double pendulum systems. With the improved framework, we show that our LNNs can learn Lagrangians representing geodesic motion in both non-relativistic and general relativistic settings. To deal with the relativistic setting, we extended our regularization to penalize violations of Lorentzian signatures, which allowed us to predict a geodesic Lagrangian under AdS\textsubscript{4} spacetime metric directly from trajectory data, which to our knowledge has not been done in the literature before. This opens new possibilities for automated discovery of geometric structures in physics, including extraction of spacetime metric tensor components from geodesic trajectories. While our approach inherits some limitations of the original LNN framework, particularly the requirement for invertible Hessians, it significantly expands the practical applicability of LNNs for scientific discovery tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。