
傳統的計算藥物研發幾乎完全依賴於高度特定於任務的計算模型來識別命中率和優化潛在客戶。使這些專業模型適應新任務需要大量的時間、計算能力和專業知識,當研究人員同時跨多個目標或屬性開展工作時,這些挑戰就會增加。
雖然 專業模型 仍被廣泛使用,但多面手模型的興起點燃了人們的希望,即這些多功能框架可以獲得有用的 化學直覺 ,這意味著它們可以處理各種藥物研發任務,併發現專業模型往往忽視的解決方案和模式。
最近推出的 SAFE-GPT 模型通過引入與藥物化學家的分子設計方法一致的化學直觀框架,代表了 AI 驅動的分子生成的範式轉變。通過使用基於序列連接的片段嵌入 (Sequential Attachment-based Fragment Embedding, SAFE) 表徵 (稍後將在本文中介紹),SAFE-GPT 解決了早期分子生成模型中的關鍵限制,以充分捕獲分子結構的靈活性和模組化。這使得 SAFE-GPT 在各種藥物研發相關任務中的表現優於基於 SMILES 的生成模型、圖神經網路和早期的基於片段的模型。
儘管 SAFE-GPT 在當時具有變革性作用,但對於各種藥物研發任務,其效率、可擴展性和適應性存在明顯局限性。
在本文中,我們將 SAFE-GPT 與最近推出的模型 GenMol 進行了比較,介紹了每種模型的優缺點,並討論了其對藥物研發的重要性。
安全概述
分子表徵的選擇對於分子設計中計算模型的 準確性 、效率和通用性至關重要,並且必須與用戶的化學直覺保持一致才能得到廣泛採用。
SAFE 表徵由 SAFE-GPT 和 GenMol 共同使用,通過將分子分解為模組化、互聯的片段來重塑分子的描述方式。與 SMILES 等將分子編碼為線性字符串的傳統分子符號不同,SAFE 將分子視為片段的無序序列。這種方法保持了化學固有的靈活性和模組化,同時與現有的 SMILES 解析器保持相容。
SAFE 特別適用於支架裝飾、連接器設計和基序擴展任務,因為它將這些問題簡化為序列完成任務 (圖 1)。通過保持分子支架的完整性並適應複雜結構,SAFE 實現了基於片段的直觀分子設計,無需複雜的基於圖形的模型。

SAFE 分子代可用於表示與 SAFE-GPT 和 GenMol 相關的各種任務的分子 (圖 1),包括:
先導優化 :使用掩碼標記動態替換分子片段,以迭代方式優化設計。
從頭生成 :從頭開始創建全新的分子結構,從而發現具有所需屬性的新型化合物。
鏈接器設計:優化連接功能片段的分子鏈接器,增強結構穩定性和活動性。
基序擴展: 擴展關鍵分子基序,以探索其他官能團或增強目標結合相互作用。
上部結構生成 :為先進的藥物研發和材料設計構建複雜的多片段分子架構。
支架裝飾 :通過添加不同的替代物來探索結構與活動之間的關係,從而修改核心分子支架。
GenMol 推理代碼示例
GenMol NIM 微服務及其配套 Notebook 可讓您輸入不同的 SAFE 或 SMILES 以及 mask 字符串,從而簡化推理請求,從 頭生成 僅需純掩膜和所需分子數量即可:
generator = GenMol_Generator(invoke_url='http://127.0.0.1:8000/generate)# provide a SMILES or a SAFE sequence stringmolecules = generator.inference(smiles='[*{15-25}]',num_molecules=20) |
圖 2 顯示了示例輸出。

對於鏈接器設計或基序擴展,您可以通過在片段中附加或插入遮罩來為一組片段提供遮罩:
# append a maskinput_text = 'c14ncnc2[nH]ccc12.C136CN5C1.S5(=O)(=O)CC.C6C#N.[*{15-35}]'# or insert the maskinput text = 'c14ncnc2[nH]ccc12.C136CN5C1.[*{5-15}].S5(=O)(=O)CC.C6C#N'# generate molecules # provide a SMILES or a SAFE sequence stringmolecules = genmol.inference(smiles=input_text,temperature=1.5, noise=1.0,num_molecules=1000) |
圖 3 顯示了示例輸出。

通過引入片段庫和 QED 評分器,您可以迭代地使用 GenMol 推理進行高級引導優化,從而根據輸入掩碼結構實現命中生成和線索優化。
library = Library(max_fragments = 1000)oracle = Oracle(score = Oracle.RDKitScore('QED'))optimizer = MolecularOptimizer(library = library, oracle = oracle, generator = generator)data = []def eval(): best = library.molecules.score[0] mean = library.molecules.head(50)['score'].mean() std = library.molecules.head(50)['score'].std() print("BEST", best, "MEAN", mean, "STD", std) data.append([best, mean, std])eval()for i in range(100): optimizer.run(iterations = 10, num_mutate = 50) eval() |
通過以迭代方式優化和更新具有高評分候選項的片段庫 (如此處的 QED 示例,但適用於任何屬性或屬性組合),您可以快速指導生成過程。
比較用於藥物研發任務的 SAFE-GPT 和 GenMol
GenMol 和 SAFE-GPT 代表了兩種不同的 AI 驅動分子生成方法,每種方法都具有獨特的優勢和局限性 (表 1)。
特徵 GenMol 安全 GPT 解碼 並行 (非自回歸) 順序 (自回歸) 任務通用性 寬 需要針對特定任務進行調整 效率 可擴展且高效 計算密集型 多樣性與品質之間的權衡 高平衡 中等
SAFE-GPT 基於自回歸 Transformer 架構構建,是一款功能強大的工具,可用於執行支架裝飾和連接器設計等片段受限的任務。它使用順序解碼,可確保在這些特定場景中的準確性和化學有效性。但是,它的順序性質和特定於任務的設計可能需要大量計算,並且無需重新訓練就無法適應新任務。
GenMol 憑借其基於擴散的離散架構和並行解碼,通過提高計算效率和任務通用性來解決許多限制。它擴展了分子生成的範圍,將更廣泛的挑戰包括在內,例如以目標為導向的潛在客戶優化,其性能甚至優於 f-RAG 和 REINVENT 等廣泛使用的模型。它的動態片段重掩蔽策略支援對化學空間進行可靠的探索,使其適用於更複雜的多目標藥物研發工作流程。
除了以目標為導向的潛在客戶優化,每個模型的解碼策略都會影響其在基於片段的任務中的性能,正如我們接下來將看到的那樣。
1. 分子生成與化學空間探索
SAFE-GPT 使用具有順序自回歸解碼功能的 GPT 架構,逐片生成分子片段。SAFE-GPT 與 SAFE 表徵的片段順序不敏感特性相結合,可應用於 從頭開始生成 和片段受限生成的分子。
GenMol 基於 BERT 架構構建,採用具有雙向注意力的並行非自回歸解碼,可同時處理分子片段。這使得 GenMol 能夠考慮不依賴於標記和片段任意順序的分子環境,並且在片段受限分子生成任務中 (根據品質分數衡量) 明顯優於 SAFE-GPT (表 2)。
任務 安全 GPT GenMol 基序擴展 18.6%*- 2.1 27.5%+- 0.8 支架裝飾 10.0+- 1.4 29.6%至 0.8 上層建築生成 14.3%– 3.7 33.3%*- 1.6
此外,離散擴散使 GenMol 能夠使用片段重新掩膜策略探索化學空間,該策略可以用掩膜標記動態替換片段,從而增強通過迭代細化發現新型優化分子的能力。這使 GenMol 能夠適用於命中生成和潛在客戶優化任務,而無需進行任何特定任務的微調。
計算效率
SAFE-GPT 的順序生成以及對強化學習目標的依賴使其成為計算密集型任務,尤其是在大規模或高吞吐量場景中。
GenMol 的離散擴散框架可提高生成效率,採樣速度最高可提升 35%,並降低計算開銷,使其在工業規模的藥物研發中更具可擴展性。
結束語
這些分子生成模型的重要性不僅限於如何完成分子生成。這也解釋了為甚麼需要重新設計。
在這個行業中,上市時間對患者來說意味著生命與死亡之間的差異,更廣泛使用的模型可以為研究人員提供通用、高效和精確的工具,以簡化發現過程、優化結果,並擴大化學可能性的視野。它們代表著從勞動密集型流程到 AI 驅動的創新的重大飛躍,而 AI 驅動的創新具有與其試圖解決的挑戰一樣的適應性。
這兩種模型均可根據研究專案的具體需求提供寶貴的工具。對於專注於具有嚴格片段限制的基序擴展和支架生成的專案而言,SAFE-GPT 是一個很好的選擇,而 GenMol 更適合需要更靈活、更統一的框架來處理各種藥物研發應用的研究人員。
WeChat
Profile