Cormorant: Covariant Molecular Neural Networks

by   Brandon Anderson, et al.

We propose Cormorant, a rotationally covariant neural network architecture for learning the behavior and properties of complex many-body physical systems. We apply these networks to molecular systems with two goals: learning atomic potential energy surfaces for use in Molecular Dynamics simulations, and learning ground state properties of molecules calculated by Density Functional Theory. Some of the key features of our network are that (a) each neuron explicitly corresponds to a subset of atoms; (b) the activation of each neuron is covariant to rotations, ensuring that overall the network is fully rotationally invariant. Furthermore, the non-linearity in our network is based upon tensor products and the Clebsch-Gordan decomposition, allowing the network to operate entirely in Fourier space. Cormorant significantly outperforms competing algorithms in learning molecular Potential Energy Surfaces from conformational geometries in the MD-17 dataset, and is competitive with other methods at learning geometric, energetic, electronic, and thermodynamic properties of molecules on the GDB-9 dataset.


page 1

page 2

page 3

page 4


N-body Networks: a Covariant Hierarchical Neural Network Architecture for Learning Atomic Potentials

We describe N-body networks, a neural network architecture for learning ...

Hierarchical modeling of molecular energies using a deep neural network

We introduce the Hierarchically Interacting Particle Neural Network (HIP...

Deep neural network for Wannier function centers

We introduce a deep neural network (DNN) model that assigns the position...

Multi-center decomposition of molecular densities: a mathematical perspective

The aim of this paper is to analyze from a mathematical perspective some...

Deep Density: circumventing the Kohn-Sham equations via symmetry preserving neural networks

The recently developed Deep Potential [Phys. Rev. Lett. 120, 143001, 201...

Multi-task learning for electronic structure to predict and explore molecular potential energy surfaces

We refine the OrbNet model to accurately predict energy, forces, and oth...

Please sign up or login with your details

Forgot password? Click here to reset