Advancing Ligand-based Virtual Screening and Molecular Generation with Pretrained Molecular Embedding Distance
Shiyun Wa, Yifei Wang, Simone Sciabola, Ye Wang (Biogen) — arXiv:2602.24474 (2026)
🎯 凍結済み事前学習分子モデルの埋め込み距離(PED)を、学習不要・3Dアライメント不要のスケーラブルな分子類似度指標として確立し、VSと分子生成の両方を加速する。
① 背景と課題

分子類似度はLBVSのランキング基準にもRL分子生成の報酬にもなる中核指標。だが従来手法には速度と精度のトレードオフがある。

2D fingerprint (ECFP/Tanimoto): 剛直・低次元で複雑な生物機序を捉えにくい
3D形状/カラー (ROCS, ROSHAMBO2): 業界標準だがコンフォマ生成+アライメントが計算集約的で大規模に不向き
DL類似度学習: 教師あり/対照学習も高品質アノテーションのコスト・既存ツール依存が残る

→ 大規模無ラベルデータで学んだ事前学習埋め込みを、タスク固有学習なしで直接類似度関数に使えないか?

② 手法の概要: Pretrained Embedding Distance (PED)
PED パイプライン: 凍結エンコーダ → 埋め込み距離Query +ReferenceSMILESGeoDiff(GIN 2D / SchNet 3D)MoLFormer(SMILES Transformer)mean-pool埋め込み z(frozen)PED 距離cosine /Euclidean

凍結 f(s)=z の埋め込みを mean-pool し cosine/Euclidean 距離で類似度を測る。GeoDiff(2D GIN/3D SchNet/Concat)とMoLFormer(SMILES)を使用。VSは昇順ランク(best-pooled)、生成は reverse-sigmoid で reward 化。

③ 本研究で示したこと
  • PEDはROCS 3D類似度と強く整列 (Concat r=-0.67)
  • LIT-PCBAでMoLFormer cosine PEDがECFP4を凌駕 (EF1% 6.15 vs 4.83)
  • RL生成報酬として有効、最大3.3×高速化
  • Boltz-2予測pIC50でPED誘導分子が高ポテンシー (REINVENT 10.27)
④ 主な結果 (a) PED パイプライン構造
PED パイプライン: 凍結エンコーダ → 埋め込み距離Query +ReferenceSMILESGeoDiff(GIN 2D / SchNet 3D)MoLFormer(SMILES Transformer)mean-pool埋め込み z(frozen)PED 距離cosine /Euclidean
④ 主な結果 (b) 3D類似度との相関
PED vs ROCS 3D 相関 (|r|, 負相関=整列)Concat vs Combinationr = -0.67MoLFormer vs Colorr = -0.64MoLFormer vs Combinationr = -0.63GeoDiff 2D vs Colorr = -0.60GeoDiff 3D vs Shaper = -0.60
④ 主な結果 (c) LIT-PCBA EF1%
6.15MoLFormer4.83ECFP4 bp4.53MoLF mean3.94ECFP4 meanLIT-PCBA EF1% (cosine)
④ 主な結果 (d) 生成分子の予測pIC50
10.27MoLFormer8.83GeoDiff3D7.7Concat7.4ROSHAMBO2REINVENT 予測 pIC50 (Boltz-2)
⑤ テイクホームメッセージ
学習不要の類似度
類似度教師・ファインチューニング無しで frozen 埋め込み距離を直接使える
3D並みの精度
ROCS 3Dと r≈-0.6〜-0.67 で整列し、LIT-PCBAで2D fingerprintを上回る
大幅な高速化
コンフォマ生成不要で生成サンプリング最大3.3×、VSは十億規模へスケール
VSと生成の統一指標
同一PEDがランキングとRL報酬を兼ね、モード選択で多様性/ポテンシーを調整
既存類似度指標との比較
類似度指標3D不要学習EF1% bp生成高速化
ECFP4 / Tanimoto不要4.83
ROCS / ROSHAMBO2 (3D)×不要基準基準 (低速)
対照学習 (DrugCLIP等)類似度教師要
PED (本研究/MoLFormer)不要(frozen)6.15最大3.3×
本研究のインパクト
  • 事前学習埋め込みが暗黙に3D/薬理学的情報を捉えることを実証
  • 3Dアライメントのボトルネックを回避し早期創薬を加速
  • VS・分子生成を貫く汎用スケーラブル類似度のレシピを提示