LEGOLAS: a Machine Learning method for rapid and accurate predictions of protein NMR chemical shifts.

This work introduces LEGOLAS, a fully open source TorchANI-based neural network model designed to predict NMR chemical shifts for protein backbone atoms. LEGOLAS has been designed to be fast, and without loss of accuracy, as our model is able to predict backbone chemical shifts with root-mean-square errors of 2.69 ppm for N, 0.95 ppm for Ca, 1.40 ppm for Cb, 1.06 ppm for C’, 0.52 ppm for amide protons, and 0.29 ppm for H. The program predicts chemical shifts at least one order of magnitude faster than the widely utilized SHIFTX2 model. This breakthrough allows us to predict NMR chemical shifts for a very large number of input structures, such as frames from a molecular dynamics trajectory. In our simulation of the protein BBL from E. coli, we observe that averaging the chemical shift predictions for a set of frames of an MD trajectory substantially improves the agreement with experiment with respect of using a single frame of the dynamics. We also show that LEGOLAS can be successfully applied to the problem of recognizing the native states of a protein among a set of decoys.