[論文レビュー] Equivariant Energy-Guided SDE for Inverse Molecular Design

Fan Bao, Min Zhao+4 more|arXiv (Cornell University)|Sep 30, 2022

Spectroscopy and Quantum Chemical Studies被引用数 15

ひとこと要約

本稿では、直交変換に対して不変なエネルギー関数を活用して幾何的対称性を活用する、制御可能な3次元分子設計を目的とした拡散型生成モデルである等長エネルギー誘導SDE（EEGSDE）を提案する。EEGSDEは、QM9およびGEOM-Drugデータセットにおいて、双極子モーメントなどの量子性質の平均絶対誤差を30%以上改善し、エネルギー関数の線形結合により複数の性質を同時に満たす生成を可能にし、ベースラインの条件付き拡散モデルを上回る性能を発揮する。

ABSTRACT

Inverse molecular design is critical in material science and drug discovery, where the generated molecules should satisfy certain desirable properties. In this paper, we propose equivariant energy-guided stochastic differential equations (EEGSDE), a flexible framework for controllable 3D molecule generation under the guidance of an energy function in diffusion models. Formally, we show that EEGSDE naturally exploits the geometric symmetry in 3D molecular conformation, as long as the energy function is invariant to orthogonal transformations. Empirically, under the guidance of designed energy functions, EEGSDE significantly improves the baseline on QM9, in inverse molecular design targeted to quantum properties and molecular structures. Furthermore, EEGSDE is able to generate molecules with multiple target properties by combining the corresponding energy functions linearly.

研究の動機と目的

• 条件付き拡散モデルが特定の目標性質を持つ分子を正確に生成するという限界を克服すること。
• 3次元分子コンformationにおける幾何的対称性を直交変換に対する不変性によって組み込むこと。
• エネルギー関数をガイドとして用いることで、柔軟で制御可能な逆分子設計フレームワークを構築すること。
• 複数のエネルギー関数を線形結合することで、複数の目標性質を同時に満たす分子生成を可能にすること。
• 分子設計における量子的および構造的性質の生成品質と安定性を向上させること。

提案手法

• 3次元幾何的対称性を尊重する等長確率微分方程式（SDE）として分子生成を定式化する。
• 回転および反転に対して不変なエネルギー関数を導入し、等長的ガイドを実現する。
• 逆SDEプロセスにおいてエネルギー関数の勾配をガイドとして用い、望ましい性質を持つ生成を誘導する。
• スコアベースのノイズ除去フレームワークを採用し、エネルギー関数がドリフト項を変更することで逆プロセスをガイドする。
• スケーリング係数を用いてエネルギー誘導の強度を制御し、忠実度と多様性のトレードオフを可能にする。
• 複数のエネルギー関数を線形結合することで、複数の目標性質を同時に満たす分子の生成を実現する。

実験結果

リサーチクエスチョン

• RQ1 幾何的対称性を活用することで、エネルギー誘導拡散モデルは3次元分子生成における制御性を向上させられるか？
• RQ2 標準的な条件付き拡散モデルと比較して、エネルギー誘導SDEは特定の量子的性質を持つ分子の生成において優れているか？
• RQ3 エネルギー関数の組み合わせによってEEGSDEは複数の目標性質を持つ分子を効果的に生成できるか？
• RQ4 エネルギー関数が直交変換に対して不変であることにより、より安定的かつ対称的な分子コンformationが得られるか？
• RQ5 エネルギー誘導のスケーリング係数は、生成された分子の品質および構造的一致性にどの程度影響を与えるか？

主な発見

• EEGSDEは、QM9における条件付きEDMと比較して、双極子モーメントの平均絶対誤差を30%以上低減した。
• 構造を標的とする生成において、GEOM-Drugにおけるベースラインモデルと比較して、構造類似度が10%以上向上した。
• エネルギー関数の線形結合により、$\alpha$、$\mu$、および$\Delta\varepsilon$といった複数の目標性質を満たす分子の生成が有効に可能となった。
• 複数性質のターゲットにおいて、生成された分子は平均で85.6%の新規性（novelty）、97.7%の原子安定性、74.6%の分子安定性を維持した。
• エネルギー誘導のスケーリング係数は構造的一致性を顕著に向上させ、高い値をとるとターゲット構造との一致度が向上した。
• 生成された分子性質の分布はトレーニングセットの分布とよく一致しており、正確な性質モデリングが実現していることが示された。