Matcha: Multi-stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking
拡散ではなく SO(3)/SO(2) 多様体上の Riemannian flow matching でブラインドドッキング ｜ arXiv:2510.14586 [cs.LG] (2025)
🎯 並進 R³・回転 SO(3)・ねじれ SO(2) 上で flow matching を行い、3段 coarse-to-fine 精密化＋教師なし物理妥当性フィルタで、物理的に妥当なポーズを co-folding 比 25倍速で生成。
① 背景と課題

近年のブラインドドッキングは DiffDock を嚆矢に拡散生成として定式化されてきた。MATCHA は拡散でなく flow matching を採用し、それを非ユークリッド多様体（並進 R³・全体回転 SO(3)・各回転可能結合のねじれ SO(2)）上で直接実行する初のパイプライン。リガンドは半柔軟（ねじれのみ可動）、タンパクは剛体として扱う。

拡散ベース（DiffDock 等）は物理妥当性が低く（PB-valid 50.7%）、衝突・破綻ポーズを生みやすい
co-folding 系（AlphaFold3/Chai-1/Boltz-2）はタンパク座標まで予測するため大規模学習＋長い推論（CHAI-1 学習 3,840 GPU日）
回帰一発予測（EquiBind/TankBind/Uni-Mol）は単一ポーズのみで多様性も物理選別もない

→ 剛体ドッキング枠で計算を軽くしつつ、生成的に多様ポーズをサンプル→教師なし物理フィルタ＋学習スコアラで選別

② 手法: MATCHA（Riemannian Flow Matching）
3段 coarse-to-fine + 後処理 Stage1: 粗い並進 大分散・角度一様 (40サンプル) Stage2: 並進精密化 中分散 Stage3: 並進+角度 小分散でシャープ化 教師なし物理フィルタ PoseBusters 4基準 スコアラで最良ポーズ選択 RMSD ペアワイズ・ランキング
R³ × SO(3) × SO(2)m
10ステップ Riemannian Euler ソルバ・損失重み w_tr=1, w_rot=1, w_tor=3

DiT + Uni-Mol 風 距離・方向アテンションバイアス。学習 38 GPU日（DiffDock 120・CHAI-1 3,840 比で桁違いに小）。

③ 本研究で示したこと（要点）
  • SO(3)/SO(2) リーマン多様体上で flow matching を行う初のドッキングパイプライン
  • 3段独立フローモデルの coarse-to-fine 精密化（粗並進→並進精密→並進+角度）
  • 推論時に教師なし PoseBusters 物理妥当性フィルタを組込み破綻ポーズを棄却
  • co-folding 比 25倍高速・学習 38 GPU日と桁違いに高効率
④ 主な結果 (a) ASTEX
65.9 MATCHA 55.3 AlphaFold3 ASTEX RMSD≤2Å & PB-valid (%)

物理妥当ポーズで次点 AlphaFold3 を +10.6pt 上回る（RMSD≤2Å は 68.2%）。

④ 主な結果 (b) 物理妥当性
MATCHA 85.9 AlphaFold3 68.9 DiffDock 50.7 PB-valid率 (RMSD≤2Å構造中,%)

PDBBind の正解ポーズのうち 85.9% が物理妥当（AF3 68.9% / DiffDock 50.7%）。

④ 主な結果 (c) アブレーション
27 スコアラのみ 44 +物理フィルタ PDBBind PB-valid成功率 (%)

アブレーション: 後段物理フィルタ併用で 0.27→0.44。サンプル 40 付近で頭打ち。

④ 主な結果 (d) 学習・速度
MATCHA 38 DiffDock 120 CHAI-1 3840 学習コスト (GPU日, ↓良)

学習 38 GPU日。推論も co-folding 系（AF3/Chai-1/Boltz-2）比 約25倍速。

⑤ テイクホームメッセージ
🌐 多様体上FM
SO(3)/SO(2) で直接 flow matching し半柔軟アラインメント問題を回避。
🧪 物理現実性
教師なし PoseBusters フィルタで PB-valid を底上げ（PDBBind 85.9%）。
⚡ 高速・低コスト
co-folding 比 25倍速、学習わずか 38 GPU日。
⚠️ 分布外課題
DockGen/PoseBusters V2 では未知ポケットへの汎化に限界。
手法比較（妥当性・コスト）
手法PB-valid率学習コスト推論速度
DiffDock50.7%120 GPU日速い
AlphaFold368.9%大規模遅い
CHAI-1—3,840 GPU日遅い
MATCHA（本研究）85.9%38 GPU日co-folding比25倍速
本研究のインパクト
  • lib/docking: 教師なし PoseBusters 物理フィルタを UniDockRunner 生成ポーズの後段に組込み、衝突・体積オーバーラップ・分子内クラッシュを純幾何計算で棄却し VS 偽陽性を削減
  • lib/docking: Riemannian flow matching 生成器を ProLIFCalculator 前段のポーズ生成器に導入し、co-folding 比 25倍速の高精度ドッキングを大規模ライブラリへ適用
  • lib/molgen: RMSD ペアワイズ・ランキング・スコアラの設計を MolgenYaml のスコアラに転用し、生成分子の結合ポーズ妥当性を生成ループ制約に組込み
Matcha: Multi-stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking. arXiv:2510.14586 [cs.LG]. 2025. https://github.com/LigandPro/Matcha