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 | 174 | |
| Molecular property prediction | QM9 | Cv0.033 | 70 | |
| Force Prediction | MD17 (test) | Aspirin Force Error1.35 | 24 | |
| Atomic force prediction | MD17 (test) | Force Error (Benzene)0.17 | 22 | |
| Formation energy prediction | Materials Project (test) | MAE (eV/atom)0.035 | 20 | |
| Initial Structure to Relaxed Energy | OC20 IS2RE Direct (test) | Energy MAE (ID)639 | 11 | |
| Molecular Dynamics Simulation | Aspirin (test) | Time per Step (ms)0.3 | 7 | |
| Total Energy Prediction | MD17 (test) | Energy Error: Benzene0.07 | 6 | |
| Energy and force prediction | QM7-X known molecules unknown conformations full | Energy MAE50.847 | 6 | |
| Energy and force prediction | QM7-X full (unknown molecules / unknown conformations) | Energy MAE51.275 | 6 |