DMol: A Highly Efficient and Chemical Motif-Preserving Molecule Generation Platform
決定論的サブグラフ・ノイズスケジュール + motif圧縮による高速・motif保存型の離散グラフ拡散分子生成 (NeurIPS 2025)
🎯 拡散ステップを一桁以上削減しつつ、カーボンリング等の化学motifを保存して常に妥当な小分子を de novo 生成すること。
① 背景と課題

小分子をグラフとして扱う離散グラフ拡散(DiGress 等)は de novo 分子生成で有効だが、各ステップで全ノード・全エッジを独立に摂動するため非常に多くのステップを要し、学習・サンプリングが遅い。さらに薬物分子の重要な building block である motif(特にカーボンリング)の保存が苦手で、SMILES validity という限定的指標しか重視されてこなかった。

DiGress は 500〜1000 拡散ステップ → 学習・生成が極めて遅い
全要素を毎ステップ摂動 → motif(環構造)が壊れやすい
SMILES validity は化学合成・物性の実用的重要性が低い

→ 「決定論的個数のサブグラフだけ摂動」+「motif圧縮」で速度と化学的妥当性を同時に改善する

② 手法の概要: DMol
DMol パイプライン: 圧縮 → 決定論的サブグラフ拡散 → denoise分子グラフ G0X,E (one-hot)motif圧縮頻出環→supernode決定論サブグラフN(t)/M(t)摂動denoiser φθgraph transf.count-MSE損失 λ2,λ3

各ステップ t で決定論的個数 N(t)=⌊(1-α)n⌋ ノード・M(t) エッジ(選択ノードが誘導する部分グラフ内に限定)のみ摂動。T=k·n(k=1〜2, n≈40)。摂動数の一致を強制する MSE ペナルティ(λ2,λ3)を損失に追加。頻出 single-bond motif 3〜15 個を supernode に圧縮。

③ 本研究で示したこと
  • 拡散ステップを T=k·n まで一桁以上削減(DiGress 500 → DMol 約38)
  • SMILES validity を DiGress 比で平均 約3% 向上
  • ChEMBL likeness・QED・shingle distance で SOTA の DeFoG を 約4% 上回る
  • motif圧縮が分布バイアスを導入しないことを定量検証(supernode出現率 train 0.119 / gen 0.114)
④ 主な結果 (a) QM9 novelty (V.U.N.)
31.8DiGress57.5DisCo62.8DeFoG75.1DMolQM9 V.U.N. (Novelty) 上昇
④ 主な結果 (b) 化学的妥当性 (ChEMBL likeness)
4.4965DiGress4.1373DisCo4.0502DeFoG4.5033DMolMOSES ChEMBL likeness 上昇
④ 主な結果 (c) 効率: 平均拡散ステップ
500DiGress500DisCo50DeFoG38DMolMOSES 平均拡散ステップ ANS 減少
④ 主な結果 (d) サンプル時間
MOSES サンプル時間 (64分子) 減少DiGressST 68.24 sDisCoST 65.36 sDeFoGST 5.81 sDMol (ours)ST 1.52 s
⑤ テイクホームメッセージ
決定論的サブグラフ拡散
摂動するノード/エッジ数を時刻に直結させ、ステップ数を一桁削減。理論的にも δt(DiGress)/δt(DMol)>1 が成立。
count-mismatch MSE 損失
変化ノード/エッジ数を N(t)/M(t) に一致させる正則化で forced-motif 保存。置換不変性も証明。
motif compression
単結合接続の頻出環を supernode 化。JT-VAE と違い decode が単純で常に valid。
化学指標で SOTA 超え
QM9/MOSES/GUACAMOL で CL・QED・shingle distance が DeFoG 比 +4%、速度も大幅短縮。
比較: 拡散系分子生成手法 (MOSES)
手法ANS↓ST(s)↓CL↑QED↑motif保存
DiGress50068.244.500.802弱(ring freeze不調)
DisCo50065.364.140.752
DeFoG505.814.050.748
DMol (ours)381.524.500.806有(supernode圧縮)
MOSES, Table 4 より。ANS=平均拡散ステップ, ST=64分子サンプル時間, CL=ChEMBL likeness, QED=drug-likeness。
本研究のインパクト
  • 拡散ステップ数をグラフサイズ n に適応させる新しい設計指針を提示し、生成コストを桁違いに削減
  • SMILES validity 偏重から motif・化学的妥当性(ChEMBL likeness/QED/shingle distance)重視への評価軸転換を後押し
  • scaffold/docking クラスタリングと組合せた scaffold-informed 条件付き生成パイプラインへ拡張可能