[論文レビュー] SiMGen example molecules

Rokas Elijošius|arXiv (Cornell University)|Jan 1, 2024

Spectroscopy Techniques in Biomedical and Chemical Research被引用数 5

ひとこと要約

SiMGenは、時間依存の類似性カーネルと事前学習済みML力場ディスクリプタを用いたゼロショット分子生成法を導入し、追加のトレーニングなしに大規模で条件付けされた分子を生成する。

ABSTRACT

Examples of structures generated using SiMGen.Macrocycles, including their generation trajectories.Small molecules.Baseline comparison of molecules generated via the linear interpolation mentioned in the paper.Note, some of these used an older version of the MACE-OFF models so a slight adjustment of the restorative force parameters might be necessary.The interactive version at https://zndraw.icp.uni-stuttgart.de/ will always have the latest stable version.

研究の動機と目的

• 広大な化学空間を効率的に探索する課題に対処し、 extensive なモデル訓練を行わずに望ましい性質を持つ分子を生成できるようにする。
• エネルギー基盤の拡散の知見を活用し、学習済みスコアが分子の組み立てと断片化ペナルティをどのように指導するかを理解する。
• 時刻依存の類似カーネルと事前学習済み力場ディスクリプタを用いて、任意のサイズの分子を構築する locality駆動のゼロショット生成フレームワーク（SiMGen）を開発する。
• 先験と点群ベースの制約を通じて、形状制御可能な生成と条件付き生成を提供する。

提案手法

• エネルギーベースの拡散モデルを訓練し、スコアを時間依存エネルギー E(x;t)として研究し、その風景を分析する。
• 局所的な原子環境 χ_i と参照集合 D_ref に基づく時間依存類似エネルギー E_sim を定義し、E_sim(x;t)=∑_i -log f(χ_i;t) とし、f はカーネル k(χ_i,χ_j;t) を含む。
• 先行力、類似力、QM風の力を統合した生成力 F(x;t) を F=k_prior(t)F_prior + k_sim(t)F_sim + k_QM(t)F_QM の形で結合する。
• アルケミカルスコアを訓練せずに z を最適化するために修正された Particle Swarm Optimization (PSO) を用いて元素置換を処理し、その後水素付加と最終 QM 弛緩を行う。
• 局所的なゼロショット生成ループ（SiMGen）を用い、時間変化するカーネル幅 σ(t) を用いて探索から化学的精製への移行を実現し、大規模なマクロ環とリンク構造を可能にする。
• 先行（異方性ガウス分布および点群先行）による形状制御を実証し、ZnDraw を用いた対話的制約付き生成をサポートする。

実験結果

リサーチクエスチョン

• RQ1 専用の生成モデルを訓練せずに、ゼロショット生成フレームワークは、QM9様の参照分布と比較して化学的に有効で多様な分子を生成できるか？
• RQ2 時間依存の類似カーネルは、生成過程における原子レベルの組み立てと断片化の抑制にどう影響するか？
• RQ3 局所性と先行が、マクロ環とリンカーを含む生成分子の形状とサイズをどの程度制御できるか？
• RQ4 類似ベースの力とPSO駆動のアルケミカル最適化を組み合わせると、頑健で制約を意識した分子生成が得られるか？
• RQ5 ZnDraw を通じた対話的制約付き生成が、断片結合と形状制御を導く上でどの程度有効か？

主な発見

• 時間依存カーネルを用いた類似ベースの生成は、エネルギー分布と類似性分布を QM9 の参照構造に近い分子を生み出す。
• 拡散モデルにおける学習済みエネルギー地形は、分子へと滑らかに下降することを示し、QMエネルギーの基準と比較して断片化をペナルティする。
• AirSS風のQMのみの弛緩は断片化した生成物を生み出しがちだが、SiMGenのアルケミカル力と類似力はより大きく、断片化の少ない分子へ偏らせる。
• 点群先行と異方性先行を用いると、形状制御された生成が可能となり、伸長する脂肪鎖や平面共役構造、さらには111個の重原子を持つマクロ環を含む。
• ゼロショット生成はスケーラブルかつ局所的であり、カーネルの局所性によりQM9より大きな分子の生成を可能にする。
• ZnDrawによる対話的制約付き生成は、再訓練なしに事前配置された断片や形状で条件付けを可能にする。