SchNet - a deep learning architecture for molecules and materials
About
Deep learning has led to a paradigm shift in artificial intelligence, including web, text and image search, speech recognition, as well as bioinformatics, with growing impact in chemical physics. Machine learning in general and deep learning in particular is ideally suited for representing quantum-mechanical interactions, enabling to model nonlinear potential-energy surfaces or enhancing the exploration of chemical compound space. Here we present the deep learning architecture SchNet that is specifically designed to model atomistic systems by making use of continuous-filter convolutional layers. We demonstrate the capabilities of SchNet by accurately predicting a range of properties across chemical space for \emph{molecules and materials} where our model learns chemically plausible embeddings of atom types across the periodic table. Finally, we employ SchNet to predict potential-energy surfaces and energy-conserving force fields for molecular dynamics simulations of small molecules and perform an exemplary study of the quantum-mechanical properties of C$_{20}$-fullerene that would have been infeasible with regular ab initio molecular dynamics.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Molecular property prediction | QM9 (test) | mu0.033 | 229 | |
| Molecular property prediction | QM9 | Cv0.033 | 80 | |
| Force Prediction | MD17 (test) | Aspirin Force Error1.35 | 30 | |
| Atomization energy prediction | QM7 (10-fold cross validation) | MAE74.2 | 27 | |
| Atomic force prediction | MD17 (test) | Force Error (Benzene)0.17 | 22 | |
| Initial Structure to Relaxed Energy | OC20 IS2RE (ID) | Energy MAE (eV)0.6372 | 22 | |
| Formation energy prediction | Materials Project (test) | MAE (eV/atom)0.035 | 20 | |
| Initial Structure to Relaxed Energy | OC20 IS2RE (OOD Cat) | Energy MAE (eV)0.6611 | 14 | |
| Molecular property prediction (EPPS) | PSB3 (test) | MAE0.05 | 12 | |
| Initial Structure to Relaxed Energy | OC20 IS2RE Direct (test) | Energy MAE (ID)639 | 11 |