Symplectic Learning for Hamiltonian Neural Networks
This work addresses the challenge of making machine learning models more interpretable and accurate for physical systems, representing an incremental improvement over existing HNNs.
The paper tackles the problem of improving Hamiltonian Neural Networks (HNNs) by introducing a symplectic learning method with a new loss function, which mathematically guarantees the existence of an exact Hamiltonian and enables error analysis and post-training correction for explainability.
Machine learning methods are widely used in the natural sciences to model and predict physical systems from observation data. Yet, they are often used as poorly understood "black boxes," disregarding existing mathematical structure and invariants of the problem. Recently, the proposal of Hamiltonian Neural Networks (HNNs) took a first step towards a unified "gray box" approach, using physical insight to improve performance for Hamiltonian systems. In this paper, we explore a significantly improved training method for HNNs, exploiting the symplectic structure of Hamiltonian systems with a different loss function. This frees the loss from an artificial lower bound. We mathematically guarantee the existence of an exact Hamiltonian function which the HNN can learn. This allows us to prove and numerically analyze the errors made by HNNs which, in turn, renders them fully explainable. Finally, we present a novel post-training correction to obtain the true Hamiltonian only from discretized observation data, up to an arbitrary order.