Node-Equivariant Message Passing for Efficient and Accurate Machine Learning Interatomic Potentials

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Yaolong, Guo, Hua
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912558584168448
author Zhang, Yaolong
Guo, Hua
author_facet Zhang, Yaolong
Guo, Hua
contents Machine learned interatomic potentials, particularly equivariant message-passing (MP) models, have demonstrated high fidelity in representing first-principles data, revolutionizing computational studies in materials science, biophysics, and catalysis. However, these equivariant MP models still incur substantial computational and memory needs due to their expensive tensor product operations over edge space, significantly limiting their applicability in large-scale or long-time simulations. In this work, we propose a node-equivariant MP (NEMP) framework that performs equivariant operations between the central node and a virtual summed node encoding structure information of its neighbors. Crucially, NEMP maintains comparable or even superior accuracy across diverse test systems-including molecules, extended systems, and universal potential benchmarks-while achieving 1-2 orders of magnitude reduction in memory and computational costs compared to edge equivariant MP models. In fact, NEMP reaches computational efficiency comparable to that of local descriptor-based models, and enabling previously inaccessible large-scale simulations.
format Preprint
id arxiv_https___arxiv_org_abs_2508_16086
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Node-Equivariant Message Passing for Efficient and Accurate Machine Learning Interatomic Potentials
Zhang, Yaolong
Guo, Hua
Chemical Physics
Machine learned interatomic potentials, particularly equivariant message-passing (MP) models, have demonstrated high fidelity in representing first-principles data, revolutionizing computational studies in materials science, biophysics, and catalysis. However, these equivariant MP models still incur substantial computational and memory needs due to their expensive tensor product operations over edge space, significantly limiting their applicability in large-scale or long-time simulations. In this work, we propose a node-equivariant MP (NEMP) framework that performs equivariant operations between the central node and a virtual summed node encoding structure information of its neighbors. Crucially, NEMP maintains comparable or even superior accuracy across diverse test systems-including molecules, extended systems, and universal potential benchmarks-while achieving 1-2 orders of magnitude reduction in memory and computational costs compared to edge equivariant MP models. In fact, NEMP reaches computational efficiency comparable to that of local descriptor-based models, and enabling previously inaccessible large-scale simulations.
title Node-Equivariant Message Passing for Efficient and Accurate Machine Learning Interatomic Potentials
topic Chemical Physics
url https://arxiv.org/abs/2508.16086