arXiv:2607.29158cs.LGcs.AI2026-07

用隐式方程替代深层神经网络,让分子模拟快2-5倍且不丢精度。

Implicit Machine Learning Force Fields Accelerate Molecular Dynamics Simulations

论文配图:Implicit Machine Learning Force Fields Accelerate Molecular Dynamics Simulations
图 1 · 摘自论文原文
  • 用固定点方程代替传统神经网络层,实现中间表示复用。
  • 在三种图神经网络上均提速2-5倍,内存占用大幅降低。
  • 适合需要高精度长时模拟的生物与材料系统研究者。

我们提出隐式机器学习力场(I-MLFF),将显式的神经网络层堆叠替换为自洽的固定点方程。在分子模拟中,该方法使中间表征可在连续时间步间复用,从而实现力计算的热启动。所得模型兼具浅层单层MLFF的计算开销与深层神经网络的表征能力与精度。本方法实现了架构无关的效率提升,突破了力预测与轨迹积分分离时的性能瓶颈。我们在三类主流图神经网络——不变性、等变笛卡尔张量、SO(3)等变球形张量架构上验证,均实现2至5倍的计算与内存开销降低。关键在于,这些加速在保持原子级分辨率和原始积分时间步的前提下达成,避免空间或时间粗粒化。本工作推进了量子力学精确分子模拟的可扩展边界,在固定GPU内存与算力预算下,支持更长轨迹与更大体系模拟,为生物与材料系统揭示新机理打开可能。

原文摘要 · Abstract (English)

We introduce implicit machine learning force fields (I-MLFFs), which replace explicit stacks of neural network layers with self-consistent fixed-point equations. In molecular simulations, this formulation enables intermediate representations to be reused across successive timesteps, thereby warm-starting force evaluation. The resulting models effectively combine the computational footprint of a shallow, single-layer MLFF with the representational capacity and accuracy of a deep neural network. Our approach unlocks architecture-agnostic efficiency gains that are inaccessible when force prediction and trajectory integration are considered separately. We demonstrate this across three major classes of graph neural networks: invariant, equivariant Cartesian tensor, and SO(3)-equivariant spherical-tensor architectures. Each yields a two- to five-fold reduction in compute and memory footprint. Crucially, these gains are achieved while retaining full atomistic resolution and the original integration timestep, avoiding spatial or temporal coarse graining. Our contribution therefore advances the scaling frontier of quantum-mechanically faithful molecular simulation, enabling longer trajectories and larger atomistic systems within fixed GPU memory and compute budgets, and thereby opening access to new insights across biomolecular and material systems.

分子模拟机器学习力场图神经网络加速计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。