MDtrajNet: AI for Direct Prediction of Molecular Dynamics

MDtrajNet: AI for Direct Prediction of Molecular Dynamics Across Chemical Space

Ge/Dral et al., Xiamen University | arXiv, 2026 | DOI: 10.5281/zenodo.15588975

🎯 力計算・時間積分を完全省略し、初期座標+時刻tから直接原子位置を予測する革新的MDアーキテクチャ。等変NN+Transformerでのベース基盤モデルがMLIP比最大100倍高速を達成。

① 背景と課題

MDシミュレーションの計算ボトルネックは力計算と数値積分という逐次プロセスに根ざす。MLFF（ANI・NequIP・MACE等）は力計算を高速化したが、femtosecond時間ステップの制約は解消されず、生物学的に重要なマイクロ秒〜ミリ秒スケールへのアクセスはなお困難だった。

MLFFでも femtosecond 積分ステップは残り、生物学的スケールへの到達が困難

TimeWarpやcoarse-grainingも依然として力計算を必要とする

→ 力計算・積分自体を省略して初期座標+時刻から直接軌道予測という根本的解法

② 手法の概要: MDtrajNet アーキテクチャ

② 手法の概要: MDtrajNet-1 基盤モデル

ANI-1xデータセットから173分子系（9原子以下）の多様な初期条件から生成したデータセットで訓練。アンサンブル4モデル（m1〜m4、各約300万パラメータ）で不確実性推定と汎化性向上。fine-tuning時は少量の系固有データで大型系に対応。

アラニン二ペプチド fine-tuning:
32 psの参照データ → 1 ns Ramachandran分布を良好に再現

③ 本研究で示したこと（要点）

④ 主な結果 (a) 速度比較（MLIP比）

④ 主な結果 (b) 精度比較（RMSE）

既知分子系

MDtrajNet ≈ ANI-1ccx MLIP — 同等以上の精度（RMSE）

未見分子系 (27種)

新規分子への汎化も確認。MLIPと同等精度を維持

テストセット: 3722本の1 ps軌道（200分子系：173既知+27未見）

④ 主な結果 (c) アラニン二ペプチド fine-tuning

④ 主な結果 (d) 先行MD加速手法との比較

現段階は9原子以下小分子特化。タンパク質規模への拡張は今後の課題。

⑤ テイクホームメッセージ

MDの根本的パラダイムシフト
力計算・積分を捨て「座標から座標への直接予測」。MDシミュレーションの概念を根本から変える可能性

100倍加速の実証
CCSD(T)/CBS精度の参照MLIPと同等精度で最大100倍高速。生物学的スケールへの直接アクセスの道

lib/md への統合展望
基盤モデル公開後にRMSDAnalyzer・HBondAnalyzerと組み合わせた超高速MDサロゲートとして統合候補

現状の限界と展望
現在9原子以下小分子に限定。大規模版（タンパク質）への拡張で創薬MDが劇的に変わる可能性

実装・データ公開情報

データセット公開

https://doi.org/10.5281/zenodo.15588975

コード/モデル

Aitomistic社（深圳）— 公開状況不明。問い合わせ要

訓練データ: ANI-1x (173分子系×多様初期条件). アンサンブル4モデル（各〜3M params）

本研究のインパクト