ArXiv提出基于神经算子的Kohn-Sham映射学习，革新密度泛函理论计算效率

摘要：ArXiv团队提出了一种基于神经算子的新方法，通过学习Kohn-Sham映射来革新密度泛函理论（DFT）的计算效率。该方法通过训练一个SE(3)等变傅里叶神经算子，直接从Kohn-Sham势预测电子密度，从而避免了传统DFT中重复的轨道对角化计算，实现了准线性缩放。这一突破使得在单个GPU上处理包含超过82,500个价电子的镁团簇成为可能，同时保持了Kohn-Sham DFT的准确性，为量子化学计算提供了新的高效解决方案。

---

核心突破

ArXiv团队提出了一种基于神经算子的新方法，通过学习Kohn-Sham映射来革新密度泛函理论（DFT）的计算效率。DFT是电子结构模拟的基础，但其计算复杂度为三次方，导致量子计算只能处理中小规模系统。该方法的核心创新点包括：

• Kohn-Sham映射学习：通过训练一个SE(3)等变傅里叶神经算子，直接从Kohn-Sham势预测电子密度，避免了传统DFT中重复的轨道对角化计算。
• 准线性缩放：该方法实现了准线性缩放，使得在单个GPU上处理包含超过82,500个价电子的镁团簇成为可能。
• 跨系统通用性：模型在训练时联合使用了8504个分子和固体数据，能够推广到分布外的有机分子、绝缘体和金属系统。

技术亮点

1. SE(3)等变性：模型在空间变换下保持等变性，确保了对不同系统结构的适应性。
2. 傅里叶神经算子：利用傅里叶空间中的算子表示，提升了模型的表示能力和计算效率。
3. 联合训练数据：模型在多样化的数据集上进行训练，增强了其在不同化学环境下的泛化能力。

行业影响

• 计算效率提升：该方法显著降低了DFT的计算复杂度，使得更大规模的量子化学计算成为可能。
• 应用场景扩展：在材料科学、药物设计和催化等领域具有广泛的应用前景。
• GPU加速：利用GPU的强大计算能力，进一步提升了计算效率。

开发者建议

• 模型优化：开发者可以进一步优化神经算子的架构，以提升其在更复杂系统中的表现。
• 数据扩展：增加训练数据的种类和规模，以增强模型的泛化能力。
• 应用集成：将模型集成到现有的量子化学软件包中，提供更高效的计算工具。

---

消息来源：ArXiv cs.AI (2026-08-24)

—— 完 ——

主题标签： #密度泛函理论 #神经算子 #量子化学 #GPU加速 #计算材料科学