核心概念
本文提出了一種名為 BiMoRec 的新型藥物推薦框架,該框架通過整合藥物的 3D 結構信息和多步驟分子增強機制,顯著提高了藥物推薦的準確性和安全性。
本研究旨在解決現有藥物推薦系統中忽略藥物 3D 結構信息,導致結構混淆和推薦準確性低的問題。
本研究提出了一種名為 BiMoRec 的雙分子模態藥物推薦框架,該框架主要包含以下三個部分:
1. 預訓練階段
利用圖神經網絡 (GIN) 從 2D 分子圖中提取藥物分子及其子結構的嵌入表示。
從藥物分子的 SDF 文件中獲取 3D 數據,並生成子結構的 3D 描述。
使用 3D 幾何向量機 (GVP) 獲得分子結構及其子結構的高維嵌入。
基於 2D 和 3D 模態的嵌入,設計了一個雙模態圖對比學習模塊,以比較兩種模態圖之間的互信息,從而實現 2D 和 3D 分子圖的融合。
2. 多步驟分子增強階段
使用預訓練階段開發的編碼器對分子及其子結構進行編碼,生成嵌入矩陣。
通過多步驟增強機制,在訪問級別對分子表示進行三重增強:
訪問內增強:使用子結構自注意力 (SSA) 模塊和全局選擇增強 (GSE) 模塊增強分子和子結構的表示。
分子逐次訪問增強:使用注意力聚合模塊將分子級表示和子結構交互表示進行基於注意力的聚合。
最新訪問增強:根據患者的最新健康狀況校準分子嵌入。
3. 時間信息聚合階段
將患者每次訪問生成的藥物表示嵌入到門控循環單元 (GRU) 中,以模擬時間關係。
將 GRU 的輸出與患者的最新表示連接起來,生成最終的藥物概率。
如果藥物的預測概率超過預定閾值,則接受該藥物推薦;否則,拒絕該推薦。