Machine Learning of Accurate Energy-Conserving Molecular Force Fields
About
Using conservation of energy - a fundamental property of closed classical and quantum mechanical systems - we develop an efficient gradient-domain machine learning (GDML) approach to construct accurate molecular force fields using a restricted number of samples from ab initio molecular dynamics (AIMD) trajectories. The GDML implementation is able to reproduce global potential energy surfaces of intermediate-sized molecules with an accuracy of 0.3 kcal $\text{mol}^{-1}$ for energies and 1 kcal $\text{mol}^{-1}$ $\text{\AA}^{-1}$ for atomic forces using only 1000 conformational geometries for training. We demonstrate this accuracy for AIMD trajectories of molecules, including benzene, toluene, naphthalene, ethanol, uracil, and aspirin. The challenge of constructing conservative force fields is accomplished in our work by learning in a Hilbert space of vector-valued functions that obey the law of energy conservation. The GDML approach enables quantitative molecular dynamics simulations for molecules at a fraction of cost of explicit AIMD calculations, thereby allowing the construction of efficient force fields with the accuracy and transferability of high-level ab initio methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Force Prediction | MD17 (test) | Aspirin Force Error0.68 | 24 | |
| Energy and force prediction | MD17 Benzene (test) | Energy MAE (kcal/mol)0.07 | 12 | |
| Energy and force prediction | MD17 Toluene (test) | Energy MAE (kcal/mol)0.12 | 12 | |
| Energy and force prediction | MD17 Salicylic acid (test) | Energy MAE (kcal/mol)0.12 | 12 | |
| Energy and force prediction | MD17 Malonaldehyde (test) | Energy MAE0.16 | 12 | |
| Energy and force prediction | MD17 Ethanol (test) | Force MAE (kcal/mol/Å)0.79 | 9 | |
| Energy and force prediction | MD17 Aspirin (test) | Force MAE (kcal/mol/Å)0.99 | 8 | |
| Molecular Dynamics Force Prediction | MD17 50k samples | Error (Aspirin)0.02 | 6 | |
| Energy and force prediction | MD17 Uracil (test) | Energy MAE0.11 | 5 | |
| Energy and force prediction | MD17 Naphtalene (test) | Energy MAE (kcal/mol)0.12 | 5 |