利用条件生成对抗神经网络加速密度泛函理论计算

超能小量子
2026-09-16 18:15:35
人工智能
化学科学
算法解析

密度泛函理论(DFT)计算常常要在同一个环节上来回打磨:先猜一个电荷密度,再求解Kohn–Sham方程,更新密度,直到自洽场(self-consistent fieldSCF)收敛。体系扩展到数百个原子,或者要连续筛选成千上万个结构时,这些迭代会积成一笔不小的计算开销。


这项工作的着力点很具体:不让机器学习替代最终的DFT结果,而是让它给SCF一个更接近终点的初始电荷密度。研究团队提出多通道体素到体素条件生成对抗网络(multi-channel voxel-to-voxel conditional generative-adversarial networkmcv2v-cGAN)。在三个测试集上,以网络预测密度启动SCF后,迭代步数的中位降幅分别为40%25%50.9%



网络不直接猜最终密度


SIESTA框架下,常规初始猜测是原子密度叠加ρSAD。模型把ρSAD与中性原子势VNA作为两个三维网格输入,预测自洽密度与初始密度之差ΔρSCSCρSAD,再与ρSAD相加得到ρcGAN。这里的密度对应价电子电荷密度。


预测差值而非直接预测ρSC,是方法中的关键取舍。ρSADρSC本就相近,直接做密度到密度的映射,生成器容易退化为复制ρSADVNA则补入了不同元素赝势的信息。模型只使用局部网格量,不需要额外的量子力学计算来生成输入。


Image1mcv2v-cGAN用于DFT加速与代理建模的整体方案。ρSADVNA共同输入网络,模型输出ΔρSC;预测密度既可用于启动完整SCF,也可在权衡精度与效率后用于非自洽性质估算和代理分子动力学。


生成器采用三维U-Net;研究还在QM9上测试了使用嵌套跳跃连接的U-Net++。判别器沿用PatchGAN思路,训练目标由条件对抗损失和逐体素L1损失共同组成。论文同时说明,受GPU显存限制,U-Net++没有用于网格分辨率更高的NFA数据集。


从轮廓成形到细节收敛


2NFA数据集中的有机硅分子C28H24N4O7SSi为例。前102步,预测结果几乎还看不出分子的整体轮廓;在103—104步之间,主要特征逐渐出现。接下来的约104步学到ΔρcGAN的关键细节,此后直到106步主要是细调,图像上的变化已经很难直接分辨。


Image2C28H24N4O7SSi分子的学习过程。白底图为不同训练阶段预测的ΔρcGAN及其与目标ΔρSC的对照;右侧黑底图给出两个输入通道ρSADVNA


作者用三类体系检验方法:QM9小分子、非富勒烯受体(NFA)分子,以及自行构建的含空位簇硅晶体Si245。经尺寸筛选后,NFA子集包含18,735个分子结构,分子最多有106个原子。Si245数据从5×5×5、共250个原子的硅超胞出发,在半径8.5 Å的球形区域内随机移除5个硅原子,再去除几何重复的空位构型;每个最终结构含245个原子。训练、验证和测试集使用彼此未见的化学结构或空位构型,论文给出的验证集样本数为10


真正省下的是多少步


3报告的不是平均步数,而是相对ρSAD基线的SCF迭代步数分布及中位降幅。QM9NFASi245分别减少40%25%50.9%。其中Si245ρSAD启动时的SCF步数中位数约为51,因此降幅也更容易呈现为连续分布;小分子的步数较少,相对降幅只能取若干离散值。


Image3QM9NFASi245中,以ρcGAN替代ρSAD作为初始密度后的SCF步数分布与相对降幅。直接以最终自洽密度ρSC输入时只需少量步骤。


论文还将QM9结果与既有工作作了文献层面的比较:本研究的中位降幅为40%,此前两项研究报告的是26.7%35.5%。不过,这不是同一软件与同一设置下的受控基准——本研究使用SIESTA,所比较文献使用VASP,不能据此下模型全面优于既有方法的结论。


ρcGAN在这里主要是完整SCF的起点,计算仍然按既定阈值迭代到自洽,因此最终结果保持DFT层级的精度。至于实际墙钟时间能省多少,还取决于体系、SCF设置和模型推理开销;论文的核心量化指标是迭代步数,而不是统一的总耗时加速倍数。


跳过SCF后,误差有多大


预测密度也能直接送入一次非自洽计算,估算动能、Hartree能、交换-相关能和总能。图4给出的总能绝对误差中位数按每原子计:QM9中,ρSAD2.05×101 eV/atomρcGAN降至3.75×103 eV/atomNFA中分别为5.37×1016.76×102 eV/atomSi245中分别为7.68×1033.23×103 eV/atom


Image4|由ρSADρcGAN和参考ρSC得到的每原子能量误差分布。从左至右依次为动能、Hartree能、交换-相关能和总能,图中标出了各分布的中位数。


ρcGAN相对ρSC的电荷密度误差中位数为:QM9 1.05%NFA 2.6%Si245 0.18%。预测密度比ρSAD更接近参考值,但作者明确指出,直接由预测密度构造的能量泛函尚未达到完整DFT计算的精度。因此,这一路径适合启发式估算,不能与ρcGAN初始化并完成SCF”混为一谈。


近似等变,也只是近似


mcv2v-cGAN没有在网络结构中强制旋转等变性,也没有用结构旋转做数据增强。作者将同一个C28H24N4O7SSi分子沿极角和方位角分别以11.25°递增,覆盖完整旋转。不同取向下预测电荷的相对偏差低于3%;将旋转后再预测的结果与原预测密度的旋转版本比较,误差低于2%。作者把这种从大规模训练数据中涌现出的统计一致性称为伪等变性pseudo-equivariance),它并不等于由架构严格保证的等变性。


Image5|模型的伪等变性与近似电荷守恒。a为旋转有机硅分子后ρcGAN的相对偏差分布;bQM9NFASi245三个数据集上的电荷守恒误差。


模型同样没有硬性施加总电荷守恒和SIESTA局域基组对应的空间支撑约束。三组数据上的电荷守恒误差约在103%101%之间。若跳过密度归一化和空间支撑校正,SCF步数降幅会下降不到5%。这说明网络学到了较好的近似约束,但近似二字不能删掉。


代理动力学能走多远


论文用NFA数据集训练的模型测试了N-甲基-2-甲基丙胺(C5H13N);该分子并不在NFA数据集中。NVT模拟温度为300 K,时间步长0.5 fs,共2000步、总计1 ps。研究比较了完整自洽ρSC、非自洽ρSAD和非自洽ρcGAN三种方案。


Image62000步分子动力学中C–CC–NH–CH–N原子对的径向分布函数。三行依次对应ρSADρcGAN和自洽参考ρSCρcGAN总体接近参考轨迹,但H–N第一峰仍可见小幅振荡。


ρSAD方案出现明显结构不稳定和峰展宽;ρcGAN大体保持分子结构,其径向分布与ρSC接近。差异主要出现在较大的原子对距离,对应长时间尺度上一些官能团的意外旋转;H–N第一峰也有小幅振荡。这个1 ps单分子测试支持将模型用于快速、探索性的从头算分子动力学,但还不足以证明它能稳定替代更长时间、更多化学体系上的自洽动力学。


适用范围写在计算设置里


本研究的DFT数据由SIESTA计算,采用GGA-PBE泛函和Troullier–Martins模守恒赝势。作者从一开始就把目标限定为特定计算设置和特定数据集下的加速。现有证据覆盖QM9、筛选后的NFA分子和含5空位簇的Si245,并没有验证过渡金属、强关联材料、表面或任意DFT软件与参数间的直接迁移。


因此,这项工作的可靠用途有两层:以ρcGAN启动完整SCF,可以在所测体系中减少迭代而保留收敛后的DFT精度;完全跳过SCF,则换来更高效率,但只能作性质预估或探索性动力学。50.9%的中位降幅来自Si245这一特定数据集,不应扩写成对所有DFT计算都能提速一半


参考文献


Pantis-Simut, C.-A., Filipoiu, N., Manolescu, A. & Nemnes, G. A. Accelerating density functional theory calculations using conditional generative-adversarial neural networks. npj Comput Mater (2026). https://doi.org/10.1038/s41524-026-02308-2

87
0
0
0
关于作者
相关文章
  • Nature_Materials必读综述:AI材料设计路线全景
    了解详情 
  • 金属显微组织图像识别大综述-从分割检测到性能逆向设计 ...
    了解详情 
  • 从基因到完整细胞:AI虚拟细胞正在走向多尺度“数字孪生” ...
    了解详情 
  • Buchwald-Hartwig反应的鲁棒性分布外预测
    了解详情 
领取成功
本月5个550bit真机配额已发放给您,配额将在2个月后到期,请及时使用哦~
活动中心
联系我们
二维码
返回顶部
返回
活动中心

完成任务,轻松获取真机配额

×
每日必做
新手任务
长期任务
其他任务
快速回复 返回顶部 返回列表
玻色有奖小调研
填写问卷,将免费赠送您1个1000bit真机配额
(单选) 您是从哪个渠道得知我们的?*
您是从哪个社交媒体得知我们的?*
您是通过哪个学校的校园宣讲得知我们的呢?
取消

提交成功

真机配额已发放到您的账户,可前往【云平台】查看

量子AI开发者认证

考核目标

开发者能够成功搭建Kaiwu-PyTorch-Plugin项目基础环境,并成功运行示例代码,根据示例提示,输出指定的值并填写至相应的输入框中。

通过奖励

5个一年效期的1000量子比特真机配额

专属「量子AI开发者」社区认证标识

开发者权益

每月固定权益:5个550量子比特真机配额
前往考核

第一步

按照README提示成功安装Kaiwu-PyTorch-Plugin库环境依赖
前往GitHub

第二步

运行 community-assessment 分支下的 run_rbm.py 代码示例

第三步

理解示例代码,手动打印并填写如下数值:

正相采样的状态

负相采样的状态

正相的能量值

负相的能量值

*

提交答案

开发者权益

每月固定权益:5个550量子比特的真机配额

恭喜您完成考核

您将获得量子AI开发者认证标识及考核奖励

1000 bit*5

配额

Quantum AI Developer Certification

Assessment Objectives

Developers should successfully set up the basic environment for the Kaiwu-PyTorch-Plugin project, run the QBM-VAE sample code, and calculate the correct FID value based on the random seed value provided by the system.

Pass Rewards

10 quotas for 550-qubit real quantum machines with a one-year validity period

Exclusive "Quantum AI Developer" Community Certification Badge

Developer Benefits

Fixed Monthly Benefits: 5 quotas for 550-qubit real quantum machines
Proceed to Assessment

Step 1

Install the environment dependencies for the Kaiwu-PyTorch-Plugin library according to the README instructions
Go to GitHub

Step 2

Replace the Seed Value

Your seed value is

Step 3

Enter the FID Value You Calculated

*

Submit Answer

Developer Benefits

Fixed Monthly Benefits: 5 quotas of 550-qubit real machines

Congratulations on Completing the Assessment

You will receive the Quantum AI Developer Certification Badge and Assessment Rewards

550bit*10

Quotas