Abstract:A rotationally invariant graph neural network (RotNet) is proposed for accurate and accelerated quantum mechanical calculations. Benchmark experiments demonstrate that RotNet outperforms prior state‐of‐the‐art methods and generalizes well to varying rotations. The high accuracy and efficiency suggest that RotNet has tremendous potential to significantly facilitate studies of protein dynamics simulation and materials engineering while maintaining physical plausibility. Deep learning has proven promising in biological and chemical applications, aiding in accurate predictions of properties such as atomic forces, energies, and material band gaps. Traditional methods with rotational invariance, one of the most crucial physical laws for predictions made by machine learning, have relied on Fourier transforms or specialized convolution filters, leading to complex model design and reduced accuracy and efficiency. However, models without rotational invariance exhibit poor generalization ability across datasets. Addressing this contradiction, this work proposes a rotationally invariant graph neural network, named RotNet, for accurate and accelerated quantum mechanical calculations that can overcome the generalization deficiency caused by rotations of molecules. RotNet ensures rotational invariance through an effective transformation and learns distance and angular information from atomic coordinates. Benchmark experiments on three datasets (protein fragments, electronic materials, and QM9) demonstrate that the proposed RotNet framework outperforms popular baselines and generalizes well to spatial data with varying rotations. The high accuracy, efficiency, and fast convergence of RotNet suggest that it has tremendous potential to significantly facilitate studies of protein dynamics simulation and materials engineering while maintaining physical plausibility.

Rotation-equivariant graph neural network for learning hadronic SMEFT effects

A rotation-equivariant graph neural network for learning hadronic SMEFT effects

Equivariant, Safe and Sensitive -- Graph Networks for New Physics

Equivariant Graph Neural Networks for Charged Particle Tracking

A Comparison Between Invariant and Equivariant Classical and Quantum Graph Neural Networks

Lie-Equivariant Quantum Graph Neural Networks

Lorentz-Equivariant Quantum Graph Neural Network for High-Energy Physics

Rotation invariant graph neural networks using spin convolutions

Improved particle-flow event reconstruction with scalable neural networks for current and future particle detectors

Optimal Equivariant Architectures from the Symmetries of Matrix-Element Likelihoods

RotNet: A Rotationally Invariant Graph Neural Network for Quantum Mechanical Calculations

An efficient Lorentz equivariant graph neural network for jet tagging

Unbinned multivariate observables for global SMEFT analyses from machine learning

NuGraph2: A Graph Neural Network for Neutrino Physics Event Reconstruction

Enhancing High-Energy Particle Physics Collision Analysis through Graph Data Attribution Techniques

Equivariant neural networks for robust observables

Predicting the Thermal Sunyaev-Zel'dovich Field using Modular and Equivariant Set-Based Neural Networks

Machine learning-based jet and event classification at the Electron-Ion Collider with applications to hadron structure and spin physics

Graph neural networks at the Large Hadron Collider