用机器学习精简量子化学计算的配置空间,提升精度与效率
Machine-Learned Compact Subspace Generation for Quantum Selected Configuration Interaction within Density Matrix Embedding Framework

- 用受限玻尔兹曼机学习量子采样配置的概率分布
- 仅用4%配置空间即达化学精度,传统方法需20%仍不达标
- 适合大规模生物分子量子模拟,降低经典计算负担
基于量子采样的量子对角化(SQD)作为计算分子基态能量的混合量子-经典范式,已展现出潜力。通过量子采样而非变分优化,避免了梯度消失问题,并可直接重构关联电子波函数。然而,现有构型选取方法主要依赖对称性约束,未保证最优物理相关构型的选择,常导致子空间过大和经典对角化成本增加。本文提出一种基于受限玻尔兹曼机(RBMs)的机器学习紧凑子空间生成方法(QSCI-RBM),并集成至密度矩阵嵌入理论(DMET)框架。该模型在量子采样构型上训练,学习主导单粒子组态的概率分布,从而定向生成高概率构型。我们在卡莫呋尔抑制剂结合新冠病毒主蛋白酶(M^pro)的蛋白质-配体复合物上进行测试。结果表明,DMET-QSCI-RBM仅需约4%的构型子空间即可达到化学精度;而标准的DMET-SQD即使访问高达20%的子空间,仍无法达到化学精度,且化学势已近乎收敛。这说明,基于RBM的构型生成能显著压缩子空间,同时保持物理准确性,有效降低经典计算开销,推动复杂生物系统量子嵌入模拟的可扩展实现。
原文摘要 · Abstract (English)
Sample-based Quantum Diagonalization (SQD), an extension of Quantum Selected Configuration Interaction (QSCI), has emerged as a promising hybrid quantum-classical paradigm for computing molecular ground state energies. By leveraging quantum sampling instead of variational optimization, QSCI avoids barren plateaus and enables direct reconstruction of correlated electronic wavefunctions. However, existing configuration recovery techniques primarily enforce symmetry constraints without guaranteeing optimal selection of the most physically relevant configurations, often leading to unnecessarily large subspaces and increased classical diagonalization costs. In this work, we introduce a machine-learned compact subspace generation protocol based on Restricted Boltzmann Machines (RBMs), termed QSCI-RBM, and integrate it within the Density Matrix Embedding Theory (DMET) framework. The RBM is trained on quantum-sampled configurations to learn the underlying probability distribution of dominant determinants, enabling the targeted generation of high-probability configurations. We apply this framework to the simulation of a protein-ligand complex involving the inhibitor Carmofur bound to the SARS-CoV-2 main protease ($M^{\text{pro}}$). Our results demonstrate that DMET-QSCI-RBM achieves energies within the chemical accuracy threshold by accessing only approximately 4% of the configuration subspace. In contrast, standard DMET-SQD simulations failed to reach chemical accuracy while accessing up to 20% of the subspace, even as the chemical potential itself nearly converged. These findings highlight that RBM-assisted configuration generation produces significantly more compact subspaces while preserving physical accuracy, thereby reducing classical computational overhead and enabling the scalable quantum embedding simulation of complex biological systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。