FORGE: Fragment-Oriented Ranking and Generation for Context-Aware Molecular Optimization
分子最適化を「どこを/どう編集するか」の文脈依存ローカル編集に再定式化 — Qwen3-0.6B 二段フレームワーク (arXiv:2602.10230, 2026)
🎯 高価な自然言語/CoT蒸留に頼らず、自動マイニングした検証済み編集ペアで小型LMを学習し、類似度制約下の分子最適化で大型モデルを上回る。
① 背景と課題

分子最適化は「出発化合物との類似性を保ちつつ小編集で性質改善する」ローカル編集問題。近年は chemistry LLM で「分子+性質記述プロンプト→改良SMILES」と解くが、局所編集設定に不適合。

プロンプトが弱い制御信号: 目的を別名/unknownに替えても PMO スコアは <3% しか変化(22中13タスクで±0.05以内)
フラグメント効果は局所文脈に強依存。context-free スカラーは構造情報を潰しラベルノイズ化
graph手法はフラグメント編集を固定演算子/外部語彙にハードコードし文脈で平均化

→ 最適化を where(どこ)と how(どう)に分解し、文脈条件付き編集空間で解く

② 手法の概要: FORGE (two-stage)
FORGE: where-to-edit + how-to-edit (Qwen3-0.6B)Input molecule(seed SMILES)atom-level tok.Stage 1: Rankfragments (where)Stage 2: Generatereplace (how)SME+ contextECFP r=2 binsChEMBL MMPlow->high pairsReplaybuffer ICLoracle score

Stage1: 全分子文脈下でフラグメント寄与をランキング(where)。Stage2: 明示的フラグメント置換を生成(how)、SME+とChEMBL MMPで監視。atom-levelトークン化でフラグメント同一性を安定化、推論はreplay-buffer ICLで勾配なし適応。

③ 本研究で示したこと
  • context-conditioned・rule-verified なフラグメント編集が高スケーラブルな学習データになる(自然言語/CoT蒸留不要)
  • 最適化を where+how に分解し、文脈条件付きランキング+明示置換+atom-levelトークン化で実現
  • 0.6B モデルが Prompt-MolOpt/PMO-1k/QED-DRD2/Lead-Opt/ChemCoTBench で大型ベースラインを凌駕
  • ECFP文脈グルーピングで寄与分散を 8-11% 低減、within-context監視が局所編集に最適
④ 主な結果 (a) Prompt-MolOpt
2.88FORGE2.86-S12.79-S1S20.82P-MolOptPrompt-MolOpt SUM (delta=0.6)
④ 主な結果 (b) 0.6B vs 8B/GPT-4o
913848DRD2862030JNK3943639GSK3bFORGE 0.6BQwen3-8B SFTGPT-4oChemCoTBench success rate (%)
④ 主な結果 (c) 利得の分解
Gain decomposition (Prompt-MolOpt delta=0.6)Prompt-MolOpt baselineSUM 0.824+ backbone upgrade+0.530 -> 1.354+ SME+ data+0.833 -> 2.187+ modif. format (S1+S2)+0.695 -> 2.882
④ 主な結果 (d) PMO-1k
PMO-1k method22-task SUM ↑
FORGE + ICL12.42
LICO11.71
MOLLEO11.65
Genetic GFN11.56
GP-BO11.27
Graph GA10.90
REINVENT10.68

1000-call budget; FORGE は replay-buffer ICL のみで適応(タスク特化FTなし)

⑤ テイクホームメッセージ
言語より編集
自然言語は分子最適化の制御信号として最適でない。検証済みフラグメント置換ペアの方がスケーラブルでハルシネーションが少ない。
文脈が価値を決める
フラグメントの価値は host 分子に依存。within-context監視(SME+)は出力形式変更より大きく効く(δ=0.6で+0.833)。
小型でも勝てる
0.6Bが8B推論モデル/GPT-4oをChemCoTBench全タスクで凌駕(DRD2/JNK3/GSK3-β SR 91/86/94%)。
勾配なし適応
replay-buffer ICLで未知black-boxオラクル(ドッキング等)にターゲット個別の再学習なしで適応。
従来手法との比較
観点従来 (prompt LLM / graph)FORGE
定式化prompt条件付き全SMILES生成where+howの局所編集
監視信号自然言語 / LLM蒸留CoTverified low-to-high編集ペア
frag寄与context-freeグローバル統計SME+ context binning
モデル規模8B推論 / GPT-4o0.6B で凌駕
black-box適応prompt rewritingreplay-buffer ICL(勾配なし)
本研究のインパクト
  • 局所編集最適化を lib/molgen に移植すれば類似度制約下のリード最適化を構造保持しつつ改善できる
  • SME+ の ECFP 文脈ビニングはフラグメント寄与の標準前処理として汎用性が高い
  • UniDock/MM-GBSA を black-box オラクルに直結し再学習コストなしの goal-directed 最適化が可能