[發明專利]分子生成方法、裝置、電子設備及存儲介質

在審

• 申請號: 202010884581.5
• 申請日: 2020-08-28
• 公開（公告）號: CN112086144A
• 公開（公告）日: 2020-12-15
• 發明（設計）人: 鄭奕嘉;吳紅艷;蔡云鵬;紀超杰
• 申請（專利權）人: 深圳先進技術研究院
• 主分類號: G16C20/50
• 分類號: G16C20/50;G16C20/70
• 代理公司: 深圳中一聯合知識產權代理有限公司 44414
• 代理人: 張瑞志

本申請適用于計算機技術領域，提供了分子生成方法、裝置、電子設備及存儲介質，分子生成方法包括：獲取源分子數據，在d個時刻中，在每一個時刻分別將源分子數據和采樣向量輸入預設的分子生成模型中，輸出與源分子數據對應的第一分子數據，其中，第r個時刻輸入的采樣向量是根據第r?1個時刻輸出的第一分子數據確定的，d≥r＞1，r和d為整數，從而可以生成更優的采樣向量，再根據更優的采樣向量生成第一分子數據，可以得到性能更好的分子。

技術領域

本申請屬于計算機技術領域，尤其涉及分子生成方法、裝置、電子設備及存儲介質。

背景技術

計算機輔助藥物分子設計是一個新興的結合了計算機、人工智能、藥學、生物學的交叉領域，其中基于計算機的分子結構生成是一個重要的研究方向。現有的分子生成方法一般是先訓練一個分子生成模型，將源分子數據輸入訓練好的分子生成模型，從而生成新的分子。為了提高生成分子的多樣性，一般是將源分子數據和采樣向量輸入分子生成模型，生成新的分子，而采樣向量一般是基于標準高斯分布進行多次采樣得到的，因此，采樣向量存在較大的隨機性，會導致不能得到最優的采樣向量，而根據隨機性較大的采樣向量生成分子時，會導致生成的分子的性能不滿足要求。

發明內容

有鑒于此，本申請實施例提供了分子生成方法、裝置、電子設備及存儲介質，可以提高生成的分子的性能。

本申請實施例的第一方面提供了一種分子生成方法，包括：

獲取源分子數據；

在d個時刻中，在每一個時刻分別將所述源分子數據和采樣向量輸入預設的分子生成模型中，輸出與所述源分子數據對應的第一分子數據，其中，第r個時刻輸入的所述采樣向量是根據第r-1個時刻輸出的所述第一分子數據確定的，d≥r＞1，r和d為整數。

在第一方面的一種可能的實現方式中，根據第r-1個時刻輸出的所述第一分子數據確定第r個時刻輸入的所述采樣向量的方法為：

將第r-1個時刻輸出的所述第一分子數據輸入預設的RNN模型中，得到第r個時刻對應的觀測狀態；

將所述第r個時刻對應的觀測狀態輸入預設的智能體中，得到第r個時刻輸入所述分子生成模型的所述采樣向量。

在第一方面的一種可能的實現方式中，第1個時刻輸入的所述采樣向量是根據所述源分子數據確定的，其中，所述第1個時刻輸入的所述采樣向量的確定方法為：

將所述源分子數據輸入所述RNN模型中，得到第1個時刻對應的觀測狀態；

將所述第1個時刻對應的觀測狀態輸入所述智能體中，得到第1個時刻輸入所述分子生成模型的所述采樣向量。

在第一方面的一種可能的實現方式中，在所述將所述第r-1個時刻對應的觀測狀態輸入預設的智能體中之前，所述方法還包括：

根據所述源分子數據和所述分子生成模型訓練所述智能體。

在第一方面的一種可能的實現方式中，所述根據所述源分子數據和所述分子生成模型訓練所述智能體，包括：

在d個時刻中，在每一個時刻根據輸入分子數據和初始的智能體確定隱向量；

將所述源分子數據和所述隱向量分別輸入所述分子生成模型中，輸出與所述源分子數據對應的第二分子數據，其中，第r個時刻的輸入分子數據是根據第r-1個時刻輸出的所述第二分子數據確定的，第1個時刻的輸入分子數據是所述源分子數據；

根據所述第二分子數據優化所述初始的智能體，得到用于輸入所述觀測狀態的智能體。

在第一方面的一種可能的實現方式中，所述根據所述第二分子數據優化所述初始的智能體，包括：

根據所述第二分子數據的屬性值、相似性以及差異性確定獎勵值；