Multiscale Neural Network Potential with Anisotropic Message Passing for the Fast and Accurate Simulation of Protein...

We present the next generation of AMP, a neural network potential (NNP) with anisotropic message passing designed to study large biomolecular systems at DFT accuracy in the condensed phase using a multiscale approach similar to quantum-mechanics/molecular-mechanics (QM/MM) with electrostatic embedding. We trained AMPv3 on our recently published biomolecular multiscale simulation (BMS25) data set and demonstrated the model's high efficiency, which enabled us to simulate proteins involving thousands of atoms at DFT accuracy in addition to explicit MM solvent for up to 100 ns, which presents a major leap for contemporary NNPs. We observe excellent scaling to large systems on a single GPU. AMPv3-BMS25 (or AMP-BMS for short) shows promising performance on benchmarks, and we demonstrate that the model can be used to accurately estimate experimental properties, including solvation free energies of small molecules and structural features of proteins. Finally, AMP-BMS/MM was employed to predict the free-energy profiles of reactions catalyzed by the enzymes chorismate mutase and fluoroacetate dehalogenase. In total, AMP-BMS/MM was used to simulate proteins in the condensed phase for a cumulative 23 μs simulation time or 48 billion integration steps. This work establishes AMP-BMS as a highly efficient and accurate model for multiscale simulations of biomolecules.