物理系在第一性原理计算与人工智能的交叉领域取得新进展
新清华 2023年05月26日 第2290期 学术前沿
本研究被选为《自然·计算科学》封面文章。
本报讯 磁性超结构不仅可以作为研究量子效应的材料平台,也有丰富的潜在应用价值。基于密度泛函理论(DFT)的第一性原理电子结构计算是研究材料性质的重要手段,但受限于计算成本,这些方法难以直接用于研究磁性超结构。近期出现了一系列深度学习第一性原理计算的方法,利用人工神经网络从DFT数据中学习,从而高效地预测大尺度材料的电子结构性质。然而,当前的方法主要集中在非磁性系统上,忽略了对磁性系统的研究。研究磁性系统需要高精度的先进深度学习方法来描述精妙的磁性物理效应,以高效准确地计算磁性材料的电子结构。
在先前的工作中,清华大学物理系教授徐勇、段文晖研究组提出深度学习第一性原理计算的理论框架和算法DeepH(Deep DFT Hamiltonian),该方法使用神经网络由材料原子结构预测DFT哈密顿量,从而可以预测所有单粒子图像下的物理性质,可极大加速非磁性材料的电子结构计算。在最新的工作中,该研究组提出xDeepH(extended DeepH)方法,用于学习磁性材料的DFT哈密顿量对原子结构和磁结构的依赖关系,并高效预测其电子结构与物性。
将物理先验知识融入神经网络架构设计,对深度学习的性能至关重要。DFT哈密顿量对原子结构和磁结构的依赖关系,在对称操作下(如旋转和时间反演)具有等变性。为此,该研究提出了一种拓展的等变神经网络,能考虑电子自旋和轨道自由度,使得神经网络保持在欧几里得群和时间反演操作下的对称性。此外,相对于原子结构,磁结构的改变对DFT哈密顿量的影响有着更强的局域特性。该研究设计了严格局域的神经网络更新函数,用于处理磁结构的输入;沿用传统的消息传递神经网络处理原子结构的输入。对于对称性和局域性这两点先验知识的妥善利用,显著提升了xDeepH方法的精度与泛化能力,并减小训练难度与代价。
通过对三种代表性复杂磁性超结构的准确预测,该研究表明了xDeepH方法的高精度和普适性。测试材料体系包括了单层NiBr2中的螺磁结构、CrI3纳米管中的弯曲磁性、双层莫尔转角CrI3中的磁性斯格明子(原胞中具有4336个原子)。该成果为研究磁性超结构提供了一种高效准确的计算方法,展现了解决DFT精度-效率两难困境的潜力。xDeepH方法有助于更好地理解磁性超结构的基本性质和量子现象,同时也有望为开发新型磁性材料和相关器件提供更高效的计算指导。
近日,研究成果以“用于磁性超结构的深度学习电子结构计算”为题发表于《自然·计算科学》(Nature Computational Science)期刊上,并入选为期刊封面文章。同期,该期刊还发表了以“研究磁性超结构的深度学习方法”为题的研究简报、以“计算探测莫尔磁体”为题的社论、以“揭示莫尔磁体中的磁相互作用”为题的新闻与观点介绍上述成果。徐勇和段文晖为该论文的通讯作者,清华大学物理系2019级博士生李贺、2019级本科生唐泽宸为共同第一作者。(物理系)