Artificial Intelligence for Direct Prediction of Molecular Dynamics Across Chemical Space

TL;DR

This paper introduces MDtrajNet, a neural network model that predicts molecular dynamics trajectories directly, significantly speeding up simulations and maintaining accuracy across various chemical systems, surpassing traditional methods.

Contribution

The authors develop a novel equivariant transformer-based neural network architecture, MDtrajNet, capable of directly generating MD trajectories across chemical space without force calculations.

Findings

Accelerates MD simulations by up to 100 times.

Achieves better accuracy than existing machine-learning potentials.

Performs well on unseen molecular systems, close to ab initio MD errors.

Abstract

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.