Load balancing for adaptive-precision interatomic potentials in materials science
Abstract
For atomistic molecular dynamics simulations, we consider a recently developed hybrid coupling between the highly accurate machine learning (ML)-based atomic cluster expansion (ACE) interaction model and a less precise (but about 1-2 orders faster) EAM potential, in order to leverage the performance bottleneck of pure ML potentials in a parallel computing environment. This spatial-temporal adaptivity has the potential for a speedup of more than an order of magnitude, that would be lost without a dynamic load-balancing strategy that becomes critical due to fluctuating potential contributions. We compare the load-balancing strategies of LAMMPS with load-balancing methods provided by the library ALL in adaptive-precision EAM-ACE simulations of a copper nanoindentation. For a regular-grid domain decomposition, the modified tensor method of the ALL library, which conserves the simplicity and transferability of the tensor method, significantly reduces the imbalance factor of the force calculation from 3.0 to 1.7 compared to the native load-balancing method of LAMMPS. A nearly perfectly balanced system with an imbalance factor close to 1.0 can be reached by load balancing an irregular grid of domains with the recursive bisectioning method of LAMMPS and with ALL's histogram method for a staggered grid of domains. Load balancing a staggered grid with ALL is found to be significantly faster than load balancing an irregular grid with the bisectioning method of LAMMPS, but the communication costs during a regular timestep are lower for the irregular grid of domains, as a lower surface-by-volume ratio of the domains is reached. Furthermore, we suggest a strategy to find the ideal load-balancing frequency for adaptive-precision simulations.
Related
- BranchIP: Learning Adaptive Equivariant Computation for Interatomic Potentials
- Benchmarking machine-learning interatomic potentials using classical force fields
- Why Multi-Layer Message Passing Works: Completeness Theory for Graph Neural Network Interatomic Potentials
- MLIP Detective: Active Failure Mode Discovery Beyond Benchmark Scores for Machine-Learning Interatomic Potentials
- Cluster-based Structural Similarity for Dataset Visualization and Data Selection for Machine Learning Interatomic Potentials
BibTeX
@misc{immel2026load,
title = {{Load balancing for adaptive-precision interatomic potentials in materials science}},
author = {David Immel and Godehard Sutmann},
howpublished = {arXiv},
year = {2026},
month = sep,
doi = {10.48550/arxiv.2609.18604},
eprint = {2609.18604},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2609.18604},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Load balancing for adaptive-precision interatomic potentials in materials science
AU - David Immel
AU - Godehard Sutmann
PY - 2026
DA - 2026/09/16
PB - arXiv
DO - 10.48550/arxiv.2609.18604
UR - https://arxiv.org/abs/2609.18604
AB - For atomistic molecular dynamics simulations, we consider a recently developed hybrid coupling between the highly accurate machine learning (ML)-based atomic cluster expansion (ACE) interaction model and a less precise (but about 1-2 orders faster) EAM potential, in order to leverage the performance bottleneck of pure ML potentials in a parallel computing environment. This spatial-temporal adaptivity has the potential for a speedup of more than an order of magnitude, that would be lost without a dynamic load-balancing strategy that becomes critical due to fluctuating potential contributions. We compare the load-balancing strategies of LAMMPS with load-balancing methods provided by the library ALL in adaptive-precision EAM-ACE simulations of a copper nanoindentation. For a regular-grid domain decomposition, the modified tensor method of the ALL library, which conserves the simplicity and transferability of the tensor method, significantly reduces the imbalance factor of the force calculation from 3.0 to 1.7 compared to the native load-balancing method of LAMMPS. A nearly perfectly balanced system with an imbalance factor close to 1.0 can be reached by load balancing an irregular grid of domains with the recursive bisectioning method of LAMMPS and with ALL's histogram method for a staggered grid of domains. Load balancing a staggered grid with ALL is found to be significantly faster than load balancing an irregular grid with the bisectioning method of LAMMPS, but the communication costs during a regular timestep are lower for the irregular grid of domains, as a lower surface-by-volume ratio of the domains is reached. Furthermore, we suggest a strategy to find the ideal load-balancing frequency for adaptive-precision simulations.
ER -