SympGNNs: Symplectic Graph Neural Networks for identifying high-dimensional Hamiltonian systems and node classification.

Alan John Varghese, Zhen Zhang, George Em Karniadakis
Author Information
  1. Alan John Varghese: School of Engineering, Brown University, Providence, RI 02912, USA.
  2. Zhen Zhang: Division of Applied Mathematics, Brown University, Providence, RI 02912, USA.
  3. George Em Karniadakis: School of Engineering, Brown University, Providence, RI 02912, USA; Division of Applied Mathematics, Brown University, Providence, RI 02912, USA. Electronic address: george_karniadakis@brown.edu.

Abstract

Existing neural network models to learn Hamiltonian systems, such as SympNets, although accurate in low-dimensions, struggle to learn the correct dynamics for high-dimensional many-body systems. Herein, we introduce Symplectic Graph Neural Networks (SympGNNs) that can effectively handle system identification in high-dimensional Hamiltonian systems, as well as node classification. SympGNNs combine symplectic maps with permutation equivariance, a property of graph neural networks. Specifically, we propose two variants of SympGNNs: (i) G-SympGNN and (ii) LA-SympGNN, arising from different parameterizations of the kinetic and potential energy. We demonstrate the capabilities of SympGNN on two physical examples: a 40-particle coupled Harmonic oscillator, and a 2000-particle molecular dynamics simulation in a two-dimensional Lennard-Jones potential. Furthermore, we demonstrate the performance of SympGNN in the node classification task, achieving accuracy comparable to the state-of-the-art. We also empirically show that SympGNN can overcome the oversmoothing and heterophily problems, two key challenges in the field of graph neural networks.

Keywords

Word Cloud

Created with Highcharts 10.0.0systemsneuralHamiltonianclassificationhigh-dimensionalSymplecticGraphnodenetworkstwoSympGNNlearndynamicsNeuralNetworksSympGNNscanmapsgraphSympGNNs:potentialdemonstrateExistingnetworkmodelsSympNetsalthoughaccuratelow-dimensionsstrugglecorrectmany-bodyHereinintroduceeffectivelyhandlesystemidentificationwellcombinesymplecticpermutationequivariancepropertySpecificallyproposevariantsG-SympGNNiiLA-SympGNNarisingdifferentparameterizationskineticenergycapabilitiesphysicalexamples:40-particlecoupledHarmonicoscillator2000-particlemolecularsimulationtwo-dimensionalLennard-JonesFurthermoreperformancetaskachievingaccuracycomparablestate-of-the-artalsoempiricallyshowovercomeoversmoothingheterophilyproblemskeychallengesfieldidentifyingDeeplearningNodePhysics-informed

Similar Articles

Cited By