本發明公開了一種基于對比學習的分子圖表示學習方法,包括:獲取每個分子的分子指紋表示,計算每兩個分子指紋之間的相似度;收集全量的化學官能團信息,為分子中的每個原子匹配對應的官能團;用異構圖對分子圖建模;利用結構感知分子編碼器中的RGCN編碼分子中每個原子的表示及其所屬官能團的表示,通過聚合函數將分子映射到特征空間,得到具有結構感知的特征表示;根據分子之間的指紋相似度,選取正、負樣本,在特征空間中進行對比學習;在大樣本分子數據集上利用對比學習的方法進行訓練,得到具有結構感知的分子編碼器,應用于下游分子屬性的預測任務。本發明有助于捕捉更豐富的分子結構信息,解決分子屬性預測的問題。
聲明:
“基于對比學習的分子圖表示學習方法” 該技術專利(論文)所有權利歸屬于技術(論文)所有人。僅供學習研究,如用于商業用途,請聯系該技術所有人。
我是此專利(論文)的發明人(作者)