Lightweight equivariant model for efficient machine learning interatomic potentials
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2023
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866913532689252352 |
|---|---|
| author | Yang, Ziduo Wang, Xian Li, Yifan Lv, Qiujie Chen, Calvin Yu-Chian Shen, Lei |
| author_facet | Yang, Ziduo Wang, Xian Li, Yifan Lv, Qiujie Chen, Calvin Yu-Chian Shen, Lei |
| contents | In modern computational materials science, deep learning has shown the capability to predict interatomic potentials, thereby supporting and accelerating conventional simulations. However, existing models typically sacrifice either accuracy or efficiency. Moreover, lightweight models are highly demanded for offering simulating systems on a considerably larger scale at reduced computational costs. Here, we introduce a lightweight equivariant interaction graph neural network (LEIGNN) that can enable accurate and efficient interatomic potential and force predictions for molecules and crystals. Rather than relying on higher-order representations, LEIGNN employs a scalar-vector dual representation to encode equivariant features. By learning geometric symmetry information, our model remains lightweight while ensuring prediction accuracy and robustness through the equivariance. Our results show that LEIGNN consistently outperforms the prediction performance of the representative baselines and achieves significant efficiency across diverse datasets, which include catalysts, molecules, and organic isomers. Furthermore, we conduct molecular dynamics (MD) simulations using the LEIGNN force field across solid, liquid, and gas systems. It is found that LEIGNN can achieve the accuracy of \textit{ab initio} MD across all examined systems. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2311_02869 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | Lightweight equivariant model for efficient machine learning interatomic potentials Yang, Ziduo Wang, Xian Li, Yifan Lv, Qiujie Chen, Calvin Yu-Chian Shen, Lei Computational Engineering, Finance, and Science In modern computational materials science, deep learning has shown the capability to predict interatomic potentials, thereby supporting and accelerating conventional simulations. However, existing models typically sacrifice either accuracy or efficiency. Moreover, lightweight models are highly demanded for offering simulating systems on a considerably larger scale at reduced computational costs. Here, we introduce a lightweight equivariant interaction graph neural network (LEIGNN) that can enable accurate and efficient interatomic potential and force predictions for molecules and crystals. Rather than relying on higher-order representations, LEIGNN employs a scalar-vector dual representation to encode equivariant features. By learning geometric symmetry information, our model remains lightweight while ensuring prediction accuracy and robustness through the equivariance. Our results show that LEIGNN consistently outperforms the prediction performance of the representative baselines and achieves significant efficiency across diverse datasets, which include catalysts, molecules, and organic isomers. Furthermore, we conduct molecular dynamics (MD) simulations using the LEIGNN force field across solid, liquid, and gas systems. It is found that LEIGNN can achieve the accuracy of \textit{ab initio} MD across all examined systems. |
| title | Lightweight equivariant model for efficient machine learning interatomic potentials |
| topic | Computational Engineering, Finance, and Science |
| url | https://arxiv.org/abs/2311.02869 |