DeepAI
Log In Sign Up

ForceNet: A Graph Neural Network for Large-Scale Quantum Calculations

03/02/2021
by   Weihua Hu, et al.
54

With massive amounts of atomic simulation data available, there is a huge opportunity to develop fast and accurate machine learning models to approximate expensive physics-based calculations. The key quantity to estimate is atomic forces, where the state-of-the-art Graph Neural Networks (GNNs) explicitly enforce basic physical constraints such as rotation-covariance. However, to strictly satisfy the physical constraints, existing models have to make tradeoffs between computational efficiency and model expressiveness. Here we explore an alternative approach. By not imposing explicit physical constraints, we can flexibly design expressive models while maintaining their computational efficiency. Physical constraints are implicitly imposed by training the models using physics-based data augmentation. To evaluate the approach, we carefully design a scalable and expressive GNN model, ForceNet, and apply it to OC20 (Chanussot et al., 2020), an unprecedentedly-large dataset of quantum physics calculations. Our proposed ForceNet is able to predict atomic forces more accurately than state-of-the-art physics-based GNNs while being faster both in training and inference. Overall, our promising and counter-intuitive results open up an exciting avenue for future research.

READ FULL TEXT

page 1

page 2

page 3

page 4

03/18/2022

Towards Training Billion Parameter Graph Neural Networks for Atomic Simulations

Recent progress in Graph Neural Networks (GNNs) for modeling atomic simu...
06/29/2022

Spherical Channels for Modeling Atomic Interactions

Modeling the energy and forces of atomic systems is a fundamental proble...
05/16/2022

Scalable algorithms for physics-informed neural and graph networks

Physics-informed machine learning (PIML) has emerged as a promising new ...
10/05/2022

Bi-Stride Multi-Scale Graph Neural Network for Mesh-Based Physical Simulation

Learning physical systems on unstructured meshes by flat Graph neural ne...
05/07/2019

Fast Neural Network Approach for Direct Covariant Forces Prediction in Complex Multi-Element Extended Systems

Neural network force field (NNFF) is a method for performing regression ...
06/17/2021

Rotation Invariant Graph Neural Networks using Spin Convolutions

Progress towards the energy breakthroughs needed to combat climate chang...