Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
We introduce BranchIP, a novel framework for adaptive tensor product computation in equivariant MLIPs, addressing the computational bottleneck of foundation-scale models. By learning to dynamically adjust computational depth based on local chemical complexity, BranchIP achieves up to $2.4\times$ acceleration and $2.6\times$ memory reduction on heterogeneous catalysis and solid acid electrolyte systems, while preserving physical accuracy. This adaptive approach also offers interpretability by correlating computational depth with chemical dynamics.
Adaptive tensor product computation slashes MLIP runtime by $2.4\times$ and memory by $2.6\times$ without sacrificing accuracy, revealing computational depth's link to chemical complexity.
Equivariant machine learning interatomic potentials (MLIPs) have revolutionized atomistic modeling, but accurate treatment of complex materials and molecular systems demands expensive models. This limits simulation length- and time-scales, with tensor products a key computational bottleneck. The recent emergence of foundation-scale MLIPs further exacerbates this challenge. We present Branch Interatomic Potential (BranchIP), a single-model framework for learned adaptive tensor product computation, trained with a novel distillation loss. In our experiments on two systems of physical interest, a heterogeneous catalysis system and a proton-conducting solid acid electrolyte, BranchIP accelerates MLIPs across model sizes by up to $2.4\times$ while reducing memory usage by up to $2.6\times$. This is achieved while maintaining physical fidelity. Furthermore, the learned adaptive computation provides model interpretability by revealing which interactions demand deeper computation and showing how computational depth relates to chemical complexity and dynamics.