arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.02128cs.LGcs.AIphysics.bio-phq-bio.BMq-bio.QM
收藏

可扩展的时空SE(3)扩散用于长时间尺度蛋白质动力学

Scalable Spatio-Temporal SE(3) Diffusion for Long-Horizon Protein Dynamics

Georgia Institute of Technology(佐治亚理工学院)

Nima Shoghi, Yuxuan Liu, Yuning Shen, Rob Brekelmans, Pan Li, Quanquan Gu

AI总结:

STAR-MD通过联合时空建模实现高效蛋白质动态模拟,显著提升构象覆盖和动态保真度,适用于微秒级长时间尺度生成。

摘要:

分子动力学(MD)模拟仍然是研究蛋白质动力学的黄金标准,但其计算成本限制了对生物相关时间尺度的访问。最近的生成模型在加速模拟方面表现出潜力,但因架构限制、误差累积和对时空动态建模不足而难以处理长时间尺度生成。我们提出了STAR-MD(Spatio-Temporal Autoregressive Rollout for Molecular Dynamics),一种可扩展的SE(3)-等变扩散模型,能够生成在微秒时间尺度上物理合理的蛋白质轨迹。我们的关键创新是一个具有联合时空注意力的因果扩散变压器,能够高效捕捉复杂的时空依赖关系,同时避免现有方法的记忆瓶颈。在标准的ATLAS基准上,STAR-MD在所有指标上均取得最佳性能——与先前方法相比,显著提高了构象覆盖、结构有效性以及动态保真度。STAR-MD能够外推生成稳定的微秒级轨迹,其中基线方法会灾难性地失败,整个扩展滚动过程中保持高质量的结构特性。我们的全面评估揭示了当前模型在长时间尺度生成方面的严重局限性,同时展示了STAR-MD的联合时空建模能够实现生物相关时间尺度的稳健动态模拟,为加速探索蛋白质功能铺平了道路。

英文摘要:

Molecular dynamics (MD) simulations remain the gold standard for studying protein dynamics, but their computational cost limits access to biologically relevant timescales. Recent generative models have shown promise in accelerating simulations, yet they struggle with long-horizon generation due to architectural constraints, error accumulation, and inadequate modeling of spatio-temporal dynamics. We present STAR-MD (Spatio-Temporal Autoregressive Rollout for Molecular Dynamics), a scalable SE(3)-equivariant diffusion model that generates physically plausible protein trajectories over microsecond timescales. Our key innovation is a causal diffusion transformer with joint spatio-temporal attention that efficiently captures complex space-time dependencies while avoiding the memory bottlenecks of existing methods. On the standard ATLAS benchmark, STAR-MD achieves state-of-the-art performance across all metrics--substantially improving conformational coverage, structural validity, and dynamic fidelity compared to previous methods. STAR-MD successfully extrapolates to generate stable microsecond-scale trajectories where baseline methods fail catastrophically, maintaining high structural quality throughout the extended rollout. Our comprehensive evaluation reveals severe limitations in current models for long-horizon generation, while demonstrating that STAR-MD's joint spatio-temporal modeling enables robust dynamics simulation at biologically relevant timescales, paving the way for accelerated exploration of protein function.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑