Advancing Ligand-based Virtual Screening and Molecular Generation with Pretrained Molecular Embedding Distance
追加学習なしの frozen 埋め込み距離(PED)を分子類似度尺度に ｜ arXiv:2604.24474 [cs.LG] (2026, Biogen)
🎯 コンフォマー生成・3D アラインメント不要で、frozen 事前学習モデル(GeoDiff/MoLFormer)の埋め込み距離を類似度関数に転用し、スケーラブルな LBVS と生成報酬を実現する。
① 背景と課題

分子類似度は VS・類似体探索・目標指向生成の計算エンジン。従来は二極化していた——指紋/2D 記述子(Tanimoto)は剛直で低次元、ROCS の 3D 形状・color アラインメントはゴールドスタンダードだがコンフォマー生成・空間アラインメントが計算集約的で大規模化が難しい。PED は追加学習なしで基盤モデル埋め込みを類似度関数に転用してこのトレードオフを回避する。

指紋/2D 記述子(Tanimoto)は剛直・低次元で複雑な生物学的機構を捉えきれない
ROCS 等 3D 形状アラインメントはゴールドスタンダードだが計算集約的で大規模ライブラリに不向き
教師あり/対照学習の類似度学習はターゲット特異データや既存ツール由来ペアに依存する

→ 類似度学習を一切せず、コンフォメーション生成(GeoDiff)/MLM(MoLFormer)で学んだ埋め込み距離をそのまま類似度に転用

② 手法: Pretrained Embedding Distance (PED)
Pretrained Embedding Distance frozen エンコーダ f(s)→z GeoDiff(2D GIN/3D SchNet,256) / MoLFormer(768) mean-pool → 固定長埋め込み D(s_i,s_j)=dist(z_i,z_j) cosine / Euclidean 多参照は best-pool min_j D → 昇順ランキング
D = dist( f(s_i), f(s_j) )
追加学習ゼロ。生成は reverse sigmoid で [0,1] 報酬化（Euclidean 推奨）

コンフォマー生成・空間アラインメント不要 → FAISS で billion-scale 近傍探索が現実的。

③ 本研究で示したこと（要点）
  • 類似度特化の学習を一切せず frozen 事前学習埋め込み距離(PED)を類似度尺度に
  • 埋め込み距離が ROCS 3D 類似度と高相関（明示的 3D アラインメントなしに空間情報を暗黙に捕捉）
  • LIT-PCBA で 3D 手法・ECFP4 を上回る LBVS 性能をコンフォマー不要で達成
  • 2D/3D/Concat の表現モードを分解し、どの構造情報が寄与するかを解析
④ 主な結果 (a) LBVS EF1%
4.53 MoLFormer PED 3.94 ECFP4 (2D) LIT-PCBA 平均 EF1%

best-pooled 6.15 vs 4.83。15 標的中 13 で EF1%>2 を達成。

④ 主な結果 (b) ROCS 相関
0.60 G-2D/color 0.60 G-3D/shape 0.67 Concat/comb 0.63 MoLF/comb ROCS との相関 |r|

明示的 3D アラインメントなしに空間/ファーマコフォア情報を暗黙に捕捉。

④ 主な結果 (c) 生成 高速化
3.3× REINVENT/MoLF 1.5× REINVENT/GeoDiff 2.0× SynFormer 生成 高速化倍率 (vs ROSHAMBO2)

コンフォマー生成不要で REINVENT 最大 3.3×・SynFormer 約 2× 高速化。

④ 主な結果 (d) 妥当性・多様性
pIC50 10.27 ± 1.34
REINVENT/MoLFormer top-500 の Boltz-2 予測（Cliff's Δ=0.92）
scaffold 多様性 5–46%
報酬モード依存（GeoDiff=高多様性 vs MoLFormer=収束的）
frozen = 追加学習ゼロ
既存パイプラインへ差し込み容易（学習コスト不要）
⑤ テイクホームメッセージ
🧊 学習ゼロ
frozen 基盤モデルの埋め込み距離をそのまま類似度に転用。
🚀 スケーラブル
コンフォマー/3Dアラインメント不要、FAISS で billion-scale 近傍探索。
🎚️ 多様性制御
報酬モード（GeoDiff=多様 vs MoLFormer=収束）で scaffold 多様性を調整。
🔀 モード選択課題
最良 PED モードが生成枠組（REINVENT vs SynFormer）で逆転する。
LIT-PCBA LBVS（EF1%）
尺度平均 EF1%best-pooled
ECFP4 (2D)3.944.83
MoLFormer cos PED4.536.15
GeoDiff 3D Euc PED—5.47
本研究のインパクト
  • lib/docking: frozen 埋め込み距離(PED)で大規模ライブラリを高速一次 LBVS、多参照は best-pool(min)で集約し上位のみ ProLIFCalculator/UniDockRunner へ。FAISS で billion-scale 近傍探索（EF1% 優位を保ち桁違いに高速）
  • lib/molgen: MolgenYaml/JobManager スコアラーに PED 報酬(reverse sigmoid・Euclidean・不良部分構造ペナルティ等重み)を追加し参照リガンド類似で生成誘導。報酬モードで scaffold 多様性を制御（3D 報酬比 REINVENT 3.3×/SynFormer 2× 高速化）
  • lib/fep: PED で参照アクティブと埋め込み距離が近い候補だけ MMGBSAEngine の高コスト評価に回すプレフィルタとし、計算予算を bioactive 領域へ集中配分
Advancing Ligand-based Virtual Screening and Molecular Generation with Pretrained Molecular Embedding Distance. arXiv:2604.24474 [cs.LG]. 2026 (Biogen). https://arxiv.org/abs/2604.24474