OMTRA: A Multi-Task Generative Model for Structure-Based Drug Design
de novo 設計・ドッキング・配座生成・ファルマコフォア条件付けを統一する多モダリティ・フローマッチング (arXiv:2512.05080)
🎯 SBDD の各タスクを「モダリティの生成/条件/欠落」分割の違いとして統一し、ひとつの SE(3)-等変 GNN を多タスク共有学習する。
① 背景と課題

SBDD の生成 SOTA は拡散・フローマッチング等の transport-based モデルだが、de novo 設計・ドッキング・配座生成はそれぞれ別モデルとして個別に設計・学習されてきた。

タスクごとに別モデルが必要で、データ・パラメータの共有が効かない
これらが本質的に同じ生成枠組みの異なる例だという視点が明示されていなかった
「ポケット+ファルマコフォア同時条件」など従来ワークフローに対応物のないタスクが扱えない

→ 全タスクを 1 つの多モダリティ・フローマッチングとして定式化し、転移学習の便益を検証できないか?

② 手法の概要: OMTRA
OMTRA: one model, many tasks via modality partitionHetero-graph8A pocket cropTask partitionMgen/Mcond/MabsPharmacophoreconditioningSE(3)-equiv GNN4 ConvBlocksODE: coordsCTMC: types

8Å ポケットを heterogeneous graph 化。タスクはモダリティを生成/条件/欠落に分割するだけで定義。FlowMol3 拡張の SE(3)-等変 GNN を 4 ConvBlock。連続=ODE, 離散=CTMC を同時サンプリング。

③ 本研究で示したこと
  • 多タスクを単一モデル・単一重みで実行する統一枠組みを定式化・実装
  • Crossdocked de novo 設計で SOTA (約90% PB-Valid, PB-Valid+IP 26%)
  • PoseBusters 再ドッキング top-1 91% PB-Valid で AlphaFold3 超え
  • 大規模事前学習・マルチタスクの転移効果は modest/非一貫と率直に報告
④ 主な結果 (a) de novo: 物理妥当+相互作用 parity
26OMTRA19Pkt2Mol17DrugFlow11Target Diff7DiffSBDDDe novo: %PB-Valid+IP
④ 主な結果 (b) 再ドッキング top-1 PB-Valid
91OMTRA84AF340SurfDock64Gnina58VinaRe-docking top-1 %PB-Valid
④ 主な結果 (c) ファルマコフォア条件の効果
Pharmacophore conditioning gainsPocket-only de novointeraction recov 60.7%+ Pharmacophoreinteraction recov 70.3% (+10pt)3 pharmacophoressampling -56%
④ 主な結果 (d) OMTRA vs ベースライン
Task / MetricOMTRABest baseline
De novo %PB-Valid+IP2619 (Pocket2Mol)
De novo %PB-Valid~9089.8 (Pocket2Mol)
Dock top-1 %PB-Valid9184 (AlphaFold3)
Dock top-1 %≤2A9293 (AlphaFold3)
Dock top-5 %PB-Valid9681 (Gnina)

Crossdocked (de novo) / PoseBusters Benchmark (docking)

⑤ テイクホームメッセージ
統一視点
SBDD タスクは modality partition の違い。1 モデルで多タスクを実行。
SOTA 性能
de novo 設計と再ドッキングの双方で既存手法を上回る/同等。
ファルマコフォア条件
interaction recovery +10pt、3 個でサンプリング -56%。
転移は未解決
大規模事前学習/多タスクの便益は modest。transfer の信頼性は open question。
従来手法との比較
観点従来 (タスク別モデル)OMTRA
対応タスク1モデル=1タスクde novo/docking/配座/pharm を1重み
定式化個別設計modality partition の違いで統一
同時条件付けpocket OR pharmpocket + pharm 同時 (新タスク)
事前学習データprotein-ligand のみ+ 500M 3D 配座 (Pharmit)
転移の便益-modest/非一貫と率直に報告
本研究のインパクト
  • SBDD 生成タスクを統一する設計思想と、その実証プラットフォームを提供
  • 500M 3D 配座 (Pharmit) を DL-ready で公開し化学多様性での事前学習を可能に
  • 「多タスク=高性能」という前提に疑問を投げ、転移学習研究の testbed を提示
Ian Dunn, Liv Toft, Tyler Katz, Juhi Gupta, Riya Shah, Ramith Hettiarachchi, David R. Koes. OMTRA: A Multi-Task Generative Model for Structure-Based Drug Design. arXiv preprint (cs.LG); ML for Structural Biology Workshop. 2025. https://arxiv.org/abs/2512.05080 · code: https://github.com/gnina/OMTRA