編輯 | 紫羅

去年，清華大學物理系徐勇課題組開發(fā)了深度學習 ，可極大加速電子結(jié)構(gòu)計算。展現(xiàn)了其解決 DFT 精度-效率困境的巨大潛力。

為了研究更廣泛的磁性材料，需要對該方法進行泛化。

磁性超結(jié)構(gòu)（例如，磁性 Skyrmion）的從頭算研究對于新型材料的研究是不可或缺的，但由于其巨大的計算成本而成為瓶頸。

為了解決這個問題，徐勇團隊開發(fā)了一種深度等變神經(jīng)網(wǎng)絡方法（Deep Equivariant Neural Network)，命名為 xDeepH，將密度泛函理論哈密頓量 HDFT?表示為原子和磁性結(jié)構(gòu)的函數(shù)，并將神經(jīng)網(wǎng)絡應用于高效的電子結(jié)構(gòu)計算。

通過將有關(guān)重要局部性和對稱性的先驗知識納入該方法，可以優(yōu)化神經(jīng)網(wǎng)絡的智能。

特別是，設計了一個神經(jīng)網(wǎng)絡架構(gòu)，充分保留了歐幾里得和時間反轉(zhuǎn)對稱性（E(3) × {I, T }）對 HDFT??的所有等效要求，這對于提高方法性能至關(guān)重要。

該方法可應用于磁性材料研究，并激發(fā)深度學習從頭算方法的發(fā)展。

該研究以《Deep-learning electronic-structure?calculation of magnetic superstructures》為題，發(fā)布在《Nature Computational Science》上。

深度學習從頭算

磁性超結(jié)構(gòu)，如磁性 skyrmion、moiré?磁和自旋螺旋磁體，引起了廣泛的研究興趣，為探索量子材料中的新興物理學提供了機會。基于密度泛函理論（DFT）的從頭計算已成為研究中不可或缺的工具，但由于計算成本高，僅適用于小型上層結(jié)構(gòu)的研究。

深度學習從頭計算方法使用人工神經(jīng)網(wǎng)絡從 ab initio 數(shù)據(jù)中學習，將神經(jīng)網(wǎng)絡應用于材料模擬，而無需調(diào)用從頭計算代碼，從而能夠研究大規(guī)模材料系統(tǒng)。然而，目前的方法通常設計用于處理無磁性的系統(tǒng)，忽略了材料性質(zhì)對磁性結(jié)構(gòu)的依賴性，因此不適合該目的。

深度學習 DFT 計算的一個關(guān)鍵問題是設計深度神經(jīng)網(wǎng)絡模型，將 DFT 哈密頓量 HDFT 表示為原子結(jié)構(gòu) HDFT（{R}）的函數(shù)，以實現(xiàn)高效的電子結(jié)構(gòu)模擬。深度學習 DFT 方法是研究領域迫切需要的。

xDeepH：擴展的深度學習 DFT 哈密頓量方法

擴展的深度學習 DFT 哈密頓量 (xDeepH) 方法，包括理論框架、數(shù)值算法和計算代碼，通過深度等變神經(jīng)網(wǎng)絡（ENN）模型學習自旋極化 DFT 哈密頓量與原子和磁性結(jié)構(gòu)的依賴關(guān)系，實現(xiàn)了大規(guī)模磁性材料的有效電子結(jié)構(gòu)計算。

作為一項關(guān)鍵創(chuàng)新，研究人員設計了一個 ENN 架構(gòu)，以結(jié)合物理見解并尊重?HDFT（{R}, {M}）的基本對稱群 E(3) × {I, T }（歐幾里得和時間反轉(zhuǎn)對稱）， 確保高效準確的深度學習。

xDeePH 的工作流程如圖 1（a）所示。首先，通過約束 DFT 計算具有不同原子和磁性構(gòu)型的小尺寸材料以準備數(shù)據(jù)集。然后在數(shù)據(jù)集上訓練表示?HDFT（{R}, {M}） 的深度神經(jīng)網(wǎng)絡。接下來，神經(jīng)網(wǎng)絡用于預測具有不同原子和磁性結(jié)構(gòu)的材料的?HDFT ?；?HDFT，可以計算單粒子圖片中材料的任何電子特性。通過用深度神經(jīng)網(wǎng)絡代替 DFT SCF 計算，xDeePH 大大降低了電子結(jié)構(gòu)計算的計算成本，并使磁性超結(jié)構(gòu)的研究成為可能。然而，關(guān)鍵問題是設計智能神經(jīng)網(wǎng)絡來建模映射函數(shù)（{R}，{M}HDFT?，使用盡可能多的先驗知識來優(yōu)化神經(jīng)網(wǎng)絡的性能。

通過對從納米管磁體、自旋螺旋磁體到磁性 skyrmion 等磁性超結(jié)構(gòu)的示例研究，系統(tǒng)地測試了該方法以顯示高精度（sub-meV 誤差）和良好的可遷移性。得益于擴展能力和最先進的性能，xDeePH 在材料研究中應用很有前途，可促進深度學習從頭算方法的發(fā)展。

總之，研究人員提出了一個通過深度神經(jīng)網(wǎng)絡表示磁性材料的 DFT 哈密頓量的通用框架，該框架構(gòu)建了從原子結(jié)構(gòu)和磁性構(gòu)型到物理特性的映射。

所有已知的 DFT 哈密頓量的物理對稱性都可以由 xDeepH 處理，這大大降低了訓練的復雜性和所需的訓練數(shù)據(jù)量。該方法為研究大規(guī)模材料系統(tǒng)中的新型磁性和電子-磁振子耦合提供了機會。

此外，可以將 xDeepH 與有效的自旋模型或磁性系統(tǒng)的機器學習潛力相結(jié)合，以研究自旋動力學對電子特性的影響。