FLOWR.root: Flow-Matching Foundation Model for Joint 3D Ligand Generation & Affinity Prediction
SE(3)等変 flow matching でポケット条件生成と多エンドポイント親和性予測を同時学習する基盤モデル
🎯 ヒット同定〜リード最適化を一気通貫で支援。3D生成・親和性予測(pIC50/Ki/Kd/EC50)・信頼度を1モデルに統合し、親和性が生成軌道を導く。
① 背景と課題

拡散/flowベースのSBDD生成(DiffSBDD・TargetDiff)はポケット内3D生成を実現したが、生成リガンドの実効性は親和性に依存する。古典スコア(Vina/Glide/GOLD)は速いが不正確、FEP/ABFEは高精度だが大規模生成には高コスト。

de novo / フラグメント / 相互作用条件を行き来する実務的レジームを既存モデルは十分に扱えない
Boltz-2の親和性は構造モジュールと分離訓練で、毎回co-foldingを要し幾何のco-adaptationが弱い
公開複合体データは希少・ノイズ大で、プロジェクト固有SARへの汎化が本質的に困難

→ 構造と親和性をjoint学習し、高速・制御可能・ポケット認識生成を行い、PEFTで継続適応する基盤モデルを構築

② 手法の概要: FLOWR.root
FLOWR.root: joint generation + affinityProtein pocket(equivariantencoder)Ligand decoderequivariant +pocket x-attnStructure head: coords/atom/bond/chargeAffinity head: pIC50/pKi/pKd/pEC50Confidence head: pLDDTInference-time SMC steering(affinity / property guidance)

FLOWR.multi基盤の等変ポケットエンコーダ+リガンドデコーダに構造/親和性/信頼度の3ヘッドを並置。fidelity順の3段学習(大規模事前学習→高品質微調整→プロジェクトPEFT)。推論時はSMC重要度サンプリングで誘導。

③ 本研究で示したこと
  • 生成・親和性・信頼度を単一SE(3)等変flow matchingでjoint学習
  • GEOM-DRUGS / CrossDocked2020 / SPINDR で生成SOTA(高validity・低strain)
  • SPINDRでpIC50 Pearson 0.92、FEP+で近FEP精度を高速達成
  • 重要度サンプリングで高親和性へ誘導、PEFTで自社データに適応
④ 主な結果 (a) CrossDocked2020 strain
519DiffSBDD295TargetD147Pkt2Mol110PILOT87.8FLOWR67.1R.rootCrossDocked2020 strain (kcal/mol, lower=better)
④ 主な結果 (b) PoseBusters妥当性
0.38DiffSBDD0.83PILOT0.92FLOWR0.97R.root CD20200.97R.root SPINDRPoseBusters-validity (higher=better)
④ 主な結果 (c) SPINDR 親和性相関
Experimental pIC50Predicted pIC50SPINDR pIC50: Pearson 0.92, R2 0.84
④ 主な結果 (d) 推論時ステアリング
Importance-sampling steering -> higher pIC50Unguidedmean pIC50 5.60Steer 0.35.75Steer 0.45.84Steer 0.56.02
⑤ テイクホームメッセージ
joint学習が鍵
親和性を構造生成と同時学習し、生成軌道を能動的に形作る(分離型Boltz-2との差別化)
3段fidelity学習
~1.5B小分子+~2.5M複合体→高品質~30k→プロジェクトPEFT。事前学習なしは大幅劣化(Pearson0.63)
高速&高精度
~15.4 s/分子でPB-valid0.97・strain67・近FEP精度。FEP/ABFEの前段プレフィルタに最適
誘導生成
SMC重要度サンプリングで予測pIC50を5.60→6.02へ。CK2α/CLK3・ERα・TYK2でQMと高相関
比較: 従来手法 vs FLOWR.root
項目従来手法FLOWR.root
生成パラダイムSE(3)拡散 (DiffSBDD/TargetDiff)flow matching + 高速ODE
親和性予測分離型co-folding (Boltz-2)/FEP高コスト構造とjoint学習・即時ランク
PB-valid (CD2020)FLOWR 0.92 / PILOT 0.830.97 (参照0.95超)
strain (kcal/mol)FLOWR 87.8 / PILOT 11067.1
Vina (CD2020)FLOWR −6.29 / PILOT −5.73−7.76
推論時間/分子拡散系 数千秒~15.4 s
適応静的モデルPEFT/DPOで継続適応
本研究のインパクト
  • 生成器とlatentを共有する高速親和性ヘッドは、分子生成と FEP/MM-GBSA を結ぶ実用的トリアージ層になる
  • 「静的モデル」から「プロジェクトSARに継続適応する伴走者」への基盤モデル思想を提示
  • de novo/フラグメント/相互作用条件を1バックボーンで統合し、ヒット同定〜リード最適化を一気通貫支援