Title
Stiffness-aware neural network for learning Hamiltonian systems
Abstract
We propose stiffness-aware neural network (SANN), a new method for learning Hamiltonian dynamical systems from data. SANN identifies and splits the training data into stiff and nonstiff portions based on a stiffness-aware index, a simple, yet effective metric we introduce to quantify the stiffness of the dynamical system. This classification along with a resampling technique allows us to apply different time integration strategies such as step size adaptation to better capture the dynamical characteristics of the Hamiltonian vector fields. We evaluate SANN on complex physical systems including a three-body problem and billiard model. We show that SANN is more stable and can better preserve energy when compared with the state-of-the-art methods, leading to significant improvement in accuracy.
Year
Venue
Keywords
2022
International Conference on Learning Representations (ICLR)
Hamiltonain systems,Neural network,Stiff dynamical systems,Data-driven method
DocType
Citations 
PageRank 
Conference
0
0.34
References 
Authors
0
3
Name
Order
Citations
PageRank
Senwei Liang102.37
Zhongzhan Huang200.34
Hong Zhang300.34