RosENet: Improving Binding Affinity Prediction by Leveraging Molecular Mechanics Energies with an Ensemble of 3D Convolutional Neural Networks
Hussein Hassan-Harrirou, Ce Zhang, Thomas Lemmin
Abstract
works), an ensemble of three-dimensional (3D) Convolutional Neural Networks (CNNs), which combines voxelized molecular mechanics energies and molecular descriptors for predicting the absolute binding affinity of protein-ligand complexes. By leveraging the physicochemical properties captured by the molecular force field, our ensemble model achieved a Root Mean Square Error (RMSE) of 1.24 on the PDBBind v2016 core set. We also explored some limitations and the robustness of the PDBBind data set and our approach on nearly 500 structures, including structures determined by Nuclear Magnetic Resonance and virtual screening experiments. Our study demonstrated that molecular mechanics energies can be voxelized and used to help improve the predictive power of the CNNs. In the future, our framework can be extended to features extracted from other biophysical and biochemical models, such as molecular dynamics simulations.