Learning Smooth and Expressive Interatomic Potentials for Physical Property Prediction
About
Machine learning interatomic potentials (MLIPs) have become increasingly effective at approximating quantum mechanical calculations at a fraction of the computational cost. However, lower errors on held out test sets do not always translate to improved results on downstream physical property prediction tasks. In this paper, we propose testing MLIPs on their practical ability to conserve energy during molecular dynamic simulations. If passed, improved correlations are found between test errors and their performance on physical property prediction tasks. We identify choices which may lead to models failing this test, and use these observations to improve upon highly-expressive models. The resulting model, eSEN, provides state-of-the-art results on a range of physical property prediction tasks, including materials stability prediction, thermal conductivity prediction, and phonon calculations.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Stability prediction | Matbench-Discovery unique structure prototypes | F1 Score83.1 | 26 | |
| Simulation Speed Estimation | Standard periodic atomic system ≈ 50 neighbors per atom, 6Å cutoff | Steps per Second8 | 23 | |
| Energy, Force, and Stress Prediction | OMat24 (val) | Energy per Atom10.7 | 21 | |
| Molecular energy prediction | OMol25 (test) | Average Rank4.14 | 17 | |
| Material Discovery | Matbench Discovery MPtrj | F1 Score83.1 | 12 | |
| Material Discovery | Matbench-Discovery non-compliant full (test) | F1 Score90.2 | 10 | |
| Material Discovery | Matbench-Discovery 10k most stable | F1 Score97.1 | 10 | |
| Energy Prediction | Diels-Alder reaction | MAE (mHa)1.15 | 8 | |
| Energy Prediction | QM7 (test) | MAE (mHa)0.13 | 8 | |
| Materials Stability Prediction | Matbench-Discovery | F1 Score92.5 | 8 |