Machine learning builds full-QM precision protein force fields in seconds

Abstract

Abstract Full-quantum mechanics (QM) calculations are extraordinarily precise but difficult to apply to large systems, such as biomolecules. Motivated by the massive demand for efficient calculations for large systems at the full-QM level and by the significant advances in machine learning, we have designed a neural network-based two-body molecular fractionation with conjugate caps (NN-TMFCC) approach to accelerate the energy and atomic force calculations of proteins. The results show very high precision for the proposed NN potential energy surface models of residue-based fragments, with energy root-mean-squared errors (RMSEs) less than 1.0 kcal/mol and force RMSEs less than 1.3 kcal/mol/Å for both training and testing sets. The proposed NN-TMFCC method calculates the energies and atomic forces of 15 representative proteins with full-QM precision in 10–100 s, which is thousands of times faster than the full-QM calculations. The computational complexity of the NN-TMFCC method is independent of the protein size and only depends on the number of residue species, which makes this method particularly suitable for rapid prediction of large systems with tens of thousands or even hundreds of thousands of times acceleration. This highly precise and efficient NN-TMFCC approach exhibits considerable potential for performing energy and force calculations, structure predictions and molecular dynamics simulations of proteins with full-QM precision.