Argonne研究人員將啟用AI的超級計算用于藥物發(fā)現(xiàn)

世界上的超級計算機正進行緊急的尋寶游戲，研究盡可能多的分子，以期找到能與有效結合并可用作藥物的分子。但是，有大量需要測試的分子積壓下來，數(shù)量達到數(shù)十億?，F(xiàn)在，阿貢實驗室(Argonne National Laboratory)的研究人員正在利用超級計算支持的AI快速識別最有希望的分子。

“我們正在嘗試構建基礎設施，以將AI和機器學習工具與基于物理的工具集成在一起，” Argonne實驗室數(shù)據(jù)科學與學習部門的計算生物學家Arvind Ramanathan在接受TACC的Aaron Dubrow采訪時解釋說。“我們將這兩種方法結合起來，以期獲得更大的收益。”

該研究小組使用了DeepDriveMD(蛋白質折疊的深度學習驅動的自適應分子模擬的縮寫)，該工具最初是為Exascale計算項目開發(fā)的。大流行發(fā)生時，DeepDriveMD被用于癌癥藥物分析，此后研究人員轉向分析。DeepDriveMD從簡單的蛋白質分子模型開始，逐漸將模型與新因素和更復雜的分析復雜化，使研究人員能夠利用深度學習發(fā)現(xiàn)蛋白質的各個方面，這些方面使其成為更強的結合候選者。

Ramanathan說：“我們構建了用于在線進行深度學習的工具包，使我們可以隨時隨地進行采樣。”“我們首先用一些數(shù)據(jù)訓練它，然后允許它非?？焖俚赝茢鄠魅氲哪M數(shù)據(jù)。然后，根據(jù)其標識的新快照，該方法會自動決定是否需要修改培訓。”

為了訓練和運行這些重型模型，研究人員轉向了一個而不是兩個，而是四個超級計算機：圣地亞哥超級計算機中心(SDSC)的2.8峰petaflop Comet系統(tǒng);德州高級計算中心(TACC)的2.3 Linpack petaflop Longhorn系統(tǒng);TACC的23.5 Linpack petaflop Frontera系統(tǒng);最后，是橡樹嶺實驗室(ORNL)的148.6 Linpack petaflop Summit系統(tǒng)，該系統(tǒng)在最新的Top500列表中被評為全球最強大的公開排名超級計算機。

Ramanathan說：“ TACC對我們的工作至關重要，尤其是Frontera機器。”“我們已經花了一段時間了，使用Frontera的CPU來最大程度地進行快速篩選：將虛擬分子放在蛋白質旁邊，以查看其是否結合，然后從中推斷出其他分子是否也會起作用相同。”(目前，該團隊每小時在Frontera上模擬300,000個配體。)

使用DeepDriveMD，研究人員從十億個分子向下鉆取到25億個分子，再向下延伸到六百萬個到幾千個，最終找到了具有最大結合能力的30個分子。這些結果將與研究合作者共享，并將很快在開放獲取報告中發(fā)布?，F(xiàn)在，研究人員正在著手分析主要蛋白酶和更大，更復雜的蛋白質。

TACC執(zhí)行董事Dan Stanzione表示：“在這樣的全球需求時代，重要的是不僅要利用我們所有的資源，而且要以盡可能創(chuàng)新的方式做到這一點。”“我們已經將許多資源用于抗擊的關鍵研究，但是在該項目中支持新的AI方法將使我們有機會更有效地利用這些資源。”