MatRIS: Toward Reliable and Efficient Pretrained Machine Learning Interatomic Potentials
About
Foundation MLIPs demonstrate broad applicability across diverse material systems and have emerged as a powerful and transformative paradigm in chemical and computational materials science. Equivariant MLIPs achieve state-of-the-art accuracy in a wide range of benchmarks by incorporating equivariant inductive bias. However, the reliance on tensor products and high-degree representations makes them computationally costly. This raises a fundamental question: as quantum mechanical-based datasets continue to expand, can we develop a more compact model to thoroughly exploit high-dimensional atomic interactions? In this work, we present MatRIS (\textbf{Mat}erials \textbf{R}epresentation and \textbf{I}nteraction \textbf{S}imulation), an invariant MLIP that introduces attention-based modeling of three-body interactions. MatRIS leverages a novel separable attention mechanism with linear complexity $O(N)$, enabling both scalability and expressiveness. MatRIS delivers accuracy comparable to that of leading equivariant models on a wide range of popular benchmarks (Matbench-Discovery, MatPES, MDR phonon, Molecular dataset, etc). Taking Matbench-Discovery as an example, MatRIS achieves an F1 score of up to 0.847 and attains comparable accuracy at a lower training cost. The work indicates that our carefully designed invariant models can match or exceed the accuracy of equivariant models at a fraction of the cost, shedding light on the development of accurate and efficient MLIPs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Energy and force prediction | Zeolite (test) | Energy MAE (meV)3 | 64 | |
| Stability prediction | Matbench-Discovery unique structure prototypes | F1 Score84.7 | 26 | |
| Energy and force prediction | Alloy DPA2 (test) | Energy RMSE (meV/atom)1.2 | 12 | |
| Material Discovery | Matbench-Discovery non-compliant full (test) | F1 Score90.3 | 10 | |
| Material Discovery | Matbench-Discovery 10k most stable | F1 Score98.6 | 10 | |
| Energy and force prediction | Cathode-P DPA2 (test) | Energy RMSE (meV/atom)0.3 | 6 | |
| Energy and force prediction | Cluster-P DPA2 (test) | Energy RMSE (meV/atom)12.8 | 6 | |
| Energy and force prediction | FerroEle-P DPA2 (test) | Energy RMSE (meV/atom)0.2 | 6 | |
| Energy and force prediction | SemiCond DPA2 (test) | Energy RMSE (meV/atom)2.7 | 6 | |
| Energy and force prediction | Cu DPA2 (test) | Energy RMSE (meV/atom)0.5 | 6 |