论文精选

一种用于加速材料 DFT 的完整神经网络电子初始化方法

Complete Neural Electronic Initialization Accelerates Materials DFT

精选理由

这个研究很硬核,用神经网络解决了材料计算中的 DFT 问题,能显著加速计算,对做材料科学或计算化学的研究者应该有用。

我们提出了一种全新的机器学习方法来加速平面波密度泛函理论(DFT)在材料计算中的应用。该方法通过引入 AugNet 模型,解决了之前方法中缺失的结构相关组件,包括 Augmentation Occupancies 和自旋初始化,从而实现了完全参考无关的电子初始化。在未见过结构上,该方法可将端到端 DFT 计算时间减少高达 25%。

原文 · arXiv cs.LG

Complete Neural Electronic Initialization Accelerates Materials DFT

We present the first complete machine learning method for accelerating plane-wave density functional theory (DFT) in materials under the projector augmented wave (PAW) formalism. We formalize seven criteria that a \textit{Complete Neural Electronic Initializer} must satisfy for practical end-to-end PAW DFT acceleration. Applying these criteria to prior work reveals two missing structure-dependent components, augmentation occupancies and spin initialization, that prevent existing methods from providing complete reference-free initialization. Controlled ablations show that omitting these components can eliminate or reverse the acceleration obtained via models that only predict the smooth valence density. We satisfy these missing requirements by introducing AugNet, the first general equivariant model for PAW augmentation occupancies, and the first general spin density model for materials, which predicts the smooth spin-difference density and spin-difference PAW augmentation occupancies using predicted magnetic moments to constrain the global magnetic state. Combined with existing valence density models, these components satisfy all seven criteria and form a fully reference-free electronic initializer for materials DFT, requiring no electronic quantities from a converged target calculation. Our method reduces end-to-end DFT wall time by up to ~25% on unseen structures while preserving converged energies.