Accelerating Protein Molecular Dynamics Simulation with DeepJump
Costa, Jacobson et al. (MIT Center for Bits and Atoms) | arXiv: 2509.13294 | 2025年9月
🎯 ジャンプサイズ δ を条件入力とする等変 Flow Matching モデルで δ ns 後のコンフォメーションを直接予測し、単一モデルでマルチスケール MD を約 1000 倍高速化する。
① 背景と課題

タンパク質の機能的ダイナミクス（折りたたみ・アロステリクス等）は μs〜ms スケールで発現するが、古典 MD の数値積分は fs タイムステップで律速され、膨大な計算資源を要する。深層学習アンサンブルモデル（AlphaFlow 等）は時系列因果性を捉えず、既存の軌道生成モデルは大ジャンプによる加速が限定的だった。

アンサンブルモデル（AlphaFlow・BioEmu）は独立コンフォメーション生成—時系列遷移経路を再現しない
既存の大ステップジャンプ手法（EquiJump・ITO）は単一スケール固定—マルチスケール対応なし

→ δ を条件入力化し、単一モデルで δ=1〜100 ns のマルチスケール加速を実現

② 手法: DeepJumpアーキテクチャ
DeepJump 推論フロー 現在構造 X_t δ ns (条件) 温度 T Current-step Conditioner Euclidean 等変 NN → コンテキスト特徴（回転・並進不変） Generative Transport (Flow Matching) Huber ロス連続確率フロー目的 — 少ない NFE で高品質サンプル X_{t+δ} (δ ns後の構造)
② 手法: 訓練データ (mdCATH)
mdCATH データセット構成 5,398 CATH ドメイン × 500 ns × 5 レプリカ × 5 温度 (320/355/379/413/450 K) = 多様な折りたたみ状態を網羅 δ = 1 / 10 / 100 ns × 64/128 dim で評価 ハイパーパラメータ grid δ=1 ns δ=10 ns δ=100 ns 64 dim 128 dim
③ 本研究で示したこと（要点）
  • 単一モデルで δ=1〜100 ns のマルチスケール MD を学習・推論—初の条件付きジャンプモデル
  • fast-folding 12 種で約 1000 倍高速化を達成（δ=1 ns）
  • MSM で定常分布 JSD=0.06 bits、折りたたみ ΔG 誤差 1.24 kBT の定量精度
  • ab initio folding（伸展→ネイティブ）への適用を初めて実証
④ 主な結果 (a) 計算高速化率
計算高速化倍率（古典 MD 比） ×10 ×100 ×1000 δ=100ns δ=10ns δ=1ns★ ★ 最良モデル (64次元)
④ 主な結果 (b) MSM精度 (δ=1ns, 64dim)
0.06
定常分布 JSD (bits) — fast-folding 12種
1.24 kBT
折りたたみ ΔG 誤差
0.3 μs
MFPT 誤差（mean first passage time）
④ 主な結果 (c) 加速フロンティア (精度-速度トレードオフ)
加速率 vs 精度 (JSD) トレードオフ JSD (誤差↑) 加速率 (δ大→右) δ=1ns δ=10ns δ=100ns ● 128dim ● 64dim 128dim が高δ域での精度低下を補償
④ 主な結果 (d) ab initio Folding
ab initio Folding (伸展→ネイティブ) 伸展構造 X_0 (全長) ×N iter ネイティブ 折りたたみ状態 DeepJump 繰り返し適用 正しい折りたたみ状態への収束を実証 （CLN025・BBL・WW domain 等 fast-folder で確認） 古典 MD なしで folding シミュレーションが可能に
⑤ テイクホームメッセージ
δ 条件付けの革新性
加速率を推論時に連続可変できる初のモデル。精度と速度のトレードオフを「加速フロンティア」として系統的に初定量化。
1000 倍加速
δ=1 ns で MSM 検証済みの 1000 倍加速。fast-folding 12 種で MSM 定常分布・MFPT を定量的に再現。
ab initio folding
伸展構造からのデノボ折りたたみを実証。構造予測（AlphaFold 等）との相補的ワークフローが可能。
現状の限界
タンパク質骨格中心でリガンド系未対応。500 ns 訓練データでは遅い集団運動の学習が不十分な可能性。
先行手法比較
手法時系列可変δ高速化
AlphaFlow✗-高
MDGen✓✗中
EquiJump✓✗中
ITO✓✗中
DeepJump✓✓×1000
本研究のインパクト
  • lib/md の長時間コンフォメーションサンプリングに DeepJump を統合—μs スケール探索コストを削減
  • 加速率-精度のトレードオフ曲線が実用的な δ 設定の指針を提供
  • ab initio folding 能力により AlphaFold 構造検証・多コンフォマー生成への応用が期待
Costa, Jacobson et al. Accelerating Protein Molecular Dynamics Simulation with DeepJump. arXiv: 2509.13294 (2025). MIT Center for Bits and Atoms.