本發(fā)明提供一種基于Transformer和增強交互型MPNN神經(jīng)網(wǎng)絡(luò)的小分子表示學(xué)習(xí)方法,包括以下步驟:S1:將藥物小分子轉(zhuǎn)化為包括有n個節(jié)點和m條邊的有向圖G=(ν,ε),從而將藥物小分子性質(zhì)預(yù)測歸類為有監(jiān)督的圖學(xué)習(xí)任務(wù);S2:構(gòu)建Transformer模型,所述的Transformer模型包括位置層、編碼層、生成器層;所述的位置層用于獲取原子與原子之間的位置信息;所述的編碼層用于學(xué)習(xí)小分子的空間結(jié)構(gòu)信息;所述的生成器層用于對藥物小分子性質(zhì)的預(yù)測;S3:將有向圖G=(ν,ε)輸入Transformer模型進行小分子性質(zhì)預(yù)測。本發(fā)明能夠在Transformer框架中很好的結(jié)合MPNN框架的點邊交互方式,既可以通過MPNN學(xué)習(xí)到局部的化學(xué)結(jié)構(gòu)信息,也能通過Transformer捕捉到遠程依賴信息。
聲明:
“基于Transformer和增強交互型MPNN神經(jīng)網(wǎng)絡(luò)的小分子表示學(xué)習(xí)方法” 該技術(shù)專利(論文)所有權(quán)利歸屬于技術(shù)(論文)所有人。僅供學(xué)習(xí)研究,如用于商業(yè)用途,請聯(lián)系該技術(shù)所有人。
我是此專利(論文)的發(fā)明人(作者)