LSD 加速分子动力学:推测采样实现 3-9 倍提速

Speculative Sampling For Faster Molecular Dynamics

精选理由

分子动力学模拟的串行瓶颈终于被打破——LSD 用推测采样实现 3-9 倍加速,做计算化学或材料模拟的团队可以直接尝试,无需修改现有模型。

AI 摘要

分子动力学模拟因串行计算瓶颈难以提升单系统吞吐量。研究者提出 Langevin Speculative Dynamics (LSD),一种分布式、模型无关的推测采样方法,借鉴语言模型和扩散模型中的推测采样思路,用草稿模型快速生成模拟步骤,再通过目标模型并行验证,并引入传输映射保证分布一致性。LSD 将推测采样扩展到二阶 Langevin 动力学,理论上推导了加速比与物理参数的关系,实验显示在不同系统和模型组合下实现 3-9 倍加速,且采样轨迹与目标模型分布一致。该方法有望大幅提升分子模拟效率,对计算化学、材料科学等领域的研究者具有实用价值。

原文 · arXiv cs.LG

Speculative Sampling For Faster Molecular Dynamics

Molecular dynamics (MD) is a key tool for simulating the dynamical behavior of atomic systems. However, MD is inherently serial, which makes it difficult to increase single-system throughput with concurrent compute. To address this, we introduce Langevin Speculative Dynamics (LSD), a distributed and model-agnostic speculative sampler for accelerating MD without adding relative error. Inspired by speculative methods in language and diffusion modeling, LSD uses a draft model to propose fast simulation steps and verifies them in parallel with a slower target model, applying a transport map from the draft to the target distribution. We extend speculative sampling to second-order Langevin dynamics, derive the achievable speedup as a function of physical parameters, show that LSD generalizes across different systems and draft-target combinations with a 3-9x speedup, and confirm theoretically and empirically that LSD samples trajectories from its target model distribution.