DeePMD-kit 是一個用 Python/C++ 編寫的軟體包,旨在最大限度地減少構建基於深度學習的原子間勢能和力場模型以及執行分子動力學 (MD) 所需的工作。這為解決分子模擬中準確性與效率的困境帶來了新的希望。 DeePMD 套件的應用範圍從有限分子到擴展系統,從金屬系統到化學鍵結系統。
有關更多信息,請查看文檔。
DeePMD-kit 專案已獲得 GNU LGPLv3.0 授權。如果您在未來的任何出版物中使用此程式碼,請出於一般目的引用以下出版物:
另外,請依照bib檔引用你所使用的方法。
Deep Potential的目標是採用深度學習技術,實現通用、準確、計算高效且可擴展的原子間位能模型。關鍵部分是透過為每個原子分配局部參考系和局部環境來尊重位能模型的廣泛和對稱不變特性。每個環境包含有限數量的原子,其局部座標以保持對稱的方式排列。然後,這些局部座標透過子網路轉換為所謂的原子能。將所有原子能量相加就得到了系統的位能。
最初的概念證明在 Deep Potential 論文中,該論文採用了一種僅使用位能來訓練神經網路模型的方法。對於典型的從頭算分子動力學 (AIMD) 資料集,這不足以重現軌跡。深勢分子動力學 (DeePMD) 模型克服了這個限制。此外,由於引入了一系列靈活的損失函數,DeePMD 中的學習過程比 Deep Potential 方法有了顯著改進。以這種方式建構的神經網路勢能在擴展系統和有限系統中準確地再現經典和量子(路徑積分)的 AIMD 軌跡,其成本隨系統規模線性擴展,並且始終比等效 AIMD 低幾個數量級模擬。
儘管效率很高,但原始的深勢模型滿足了位能模型的廣泛和對稱不變性質,但代價是在模型中引入了不連續性。這對規範採樣的軌蹟的影響可以忽略不計,但可能不足以計算動力學和機械特性。這些點促使我們開發了 Deep Potential-Smooth Edition (DeepPot-SE) 模型,該模型以平滑的自適應嵌入網路取代非平滑的局部框架。 DeepPot-SE 在對物理、化學、生物學和材料科學領域感興趣的多種系統進行建模方面表現出強大的能力。
除了建立位能模型外,DeePMD-kit 還可用於建立粗粒化模型。在這些模型中,我們想要參數化的量是粗粒粒子的自由能或粗粒位能。有關更多詳細信息,請參閱 DeePCG 論文。
se_e2_r
、 se_e3
和se_atten
(DPA-1)。有關 v2.2.3 之前所有功能的詳細信息,請參閱我們的最新論文。
請閱讀線上文檔,以了解如何安裝和使用 DeePMD-kit。
程式碼組織如下:
examples
:例子。deepmd
:DeePMD-kit python 模組。source/lib
:核心庫的原始碼。source/op
:運算子 (OP) 實作。source/api_cc
:DeePMD-kit C++ API 的原始碼。source/api_c
:C API 的原始碼。source/nodejs
:Node.js API 的原始碼。source/ipi
:i-PI 客戶端的原始碼。source/lmp
:LAMMPS 模組的原始碼。source/gmx
:Gromacs 插件的原始程式碼。請參閱 DeePMD-kit 貢獻指南以成為貢獻者! ?