Calculation of solvation force in molecular dynamics simulation by deep-learning method

Electrostatic calculations are generally used in studying the thermodynamics and kinetics of biomolecules in solvent. Generally, this is performed by solving the Poisson-Boltzmann equation on a large grid system, a process known to be time consuming. In this study, we developed a deep neural network to predict the decomposed solvation free energies and forces of all atoms in a molecule. To train the network, the internal coordinates of the molecule were used as the input data, and the solvation free energies along with transformed atomic forces from the Poisson-Boltzmann equation were used as labels. Both the training and prediction tasks were accelerated on GPU. Formal tests demonstrated that our method can provide reasonable predictions for small molecules when the network is well-trained with its simulation data. This method is suitable for processing lots of snapshots of molecules in a long trajectory. Moreover, we applied this method in the molecular dynamics simulation with enhanced sampling. The calculated free energy landscape closely resembled that obtained from explicit solvent simulations.