用AI直接生成分子动力学轨迹,速度提升100倍且精度更高。
Artificial Intelligence for Direct Prediction of Molecular Dynamics Across Chemical Space
- 用等变神经网络+Transformer架构,跳过力计算直接预测轨迹。
- 模拟速度比传统方法快100倍,小分子误差接近从头算精度。
- 适合需要快速模拟的分子系统,尤其适合未见体系的初值建模。
分子动力学(MD)是研究原子系统行为的强大工具,但其依赖于顺序数值积分,限制了模拟效率。本文提出一种新型神经网络架构MDtrajNet及预训练基础模型MDtrajNet-1,可直接跨化学空间生成MD轨迹,无需力计算与积分过程。MDtrajNet结合等变神经网络与基于Transformer的结构,在长时轨迹预测中实现高精度与强泛化能力。该方法将模拟速度提升达两个数量级,且在相同数据训练下,精度优于现有机器学习势能模型。值得注意的是,对于各类已见及未见的小分子体系,MDtrajNet-1生成轨迹的误差接近传统从头算分子动力学水平。当前局限源于训练数据化学空间相对有限;然而,即使对更大、未见体系,该模型仍可作为微调起点,构建特定系统模型。其灵活架构支持多种统计系综、边界条件与相互作用类型。通过突破传统MD的固有速度瓶颈,MDtrajNet为高效、可扩展的原子级模拟开辟新路径。
原文摘要 · Abstract (English)
Molecular dynamics (MD) is a powerful tool for exploring the behavior of atomistic systems, but its reliance on sequential numerical integration limits simulation efficiency. We present a novel neural network architecture, MDtrajNet, and a pre-trained foundational model, MDtrajNet-1, that directly generates MD trajectories across chemical space, bypassing force calculations and integration. MDtrajNet combines equivariant neural networks with a transformer-based architecture to achieve strong accuracy and transferability in predicting long-time trajectories. This approach accelerates simulations by up to two orders of magnitude and yields better accuracy than MD propagated with established machine-learning interatomic potentials trained on the same data. Remarkably, the errors of the trajectories generated by MDtrajNet-1 for various seen and even unseen small-sized molecular systems are close to those of the conventional ab initio MD. The current limitations of MDtrajNet-1 are attributed to the relatively small size of the chemical space in its training data; however, even for bigger, unseen systems, MDtrajNet-1 provides a good starting point for fine-tuning and obtaining system-specific models. The architecture's flexible design supports diverse application scenarios, including different statistical ensembles, boundary conditions, and interaction types. By overcoming the intrinsic speed barrier of conventional MD, MDtrajNet opens new frontiers in efficient and scalable atomistic simulations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。