非监督学习高维多体波函数矢量轨迹所在低维子空间

黄志伟 杨宏宇 翟峰 陆肖励 卢军强† 吴健

1) (浙江师范大学物理系，金华 321004)

2) (清华大学物理系，北京 100084)

Frame 等提出采用特征矢量延拓方法求解关联量子模型的高维多体波函数:当模型哈密顿矩阵包含光滑变化的参数时，其特征矢量随参数变化的轨迹集中在1 个低维的子空间中，因此可以将哈密顿量投影到该子空间的一组基矢量来简化求解(Frame D，He R Z，Ipsen I，Lee D，Lee D，Rrapaj E 2018 Phys.Rev.Lett.121 032501).但是他们没有明确给出轨迹子空间的维度以及其与模型大小之间的联系.本文系统研究了大小不同的反铁磁Heisenberg 链模型，其交换相互作用随参数光滑变化.首先通过主成分分析方法分别确定了包含4 个自旋的模型和包含6 个自旋的模型的基态多体波函数矢量轨迹子空间，并分别绘制了子空间中的轨迹.然后分析了包含 8，···，14 个自旋的模型基态矢量轨迹的主成分分量，并指出:当采用特征矢量延拓方法求解反铁磁Heisenberg 链模型基态时，所需基矢数目随模型所包含自旋个数的增加而增加.本文研究可用于指导采用特征矢量延拓方法求解包含更多自旋的反铁磁Heisenberg 链模型哈密顿量.

1 引言

关联量子模型的希尔伯特空间维度通常随模型所含粒子数呈指数增长，如果不采取任何近似，即使包含较少粒子数目的模型哈密顿矩阵也很容易超出计算机的内存空间而不能载入，严格的数值求解更是无从谈起.现有的关联量子模型哈密顿量近似求解手段包括变分方法[1-6]、量子蒙特卡罗方法[7-11]以及图展开方法[12-15]等;这些方法的有效性依赖于模型哈密顿量所包含的某些参数的取值范围:在一定的参数区间范围内，可以利用这些方法进行有效求解，超出了这些范围，采用这些方法求解所导致的误差往往无法控制而导致求解失败.

最近，Frame 等[16]提出了一种新方法求解关联量子模型哈密顿量:特征矢量延拓.其关键思路是:尽管高维哈密顿矩阵的特征矢量在高维空间中，当哈密顿矩阵随参数光滑变化时，相应特征矢量的变化轨迹集中在1 个低维的子空间中;因此可以选取少量对应不同参数的特征矢量作为基矢来构建这个低维子空间，将哈密顿量在此子空间中展开，从而只需求解低维哈密顿矩阵.他们将1 个具有可调参数的 4×4×4 晶格的Bose-Hubbard 模型哈密顿量分别展开到1 个由3 个参数互相远离的特征矢量为基矢构成的三维子空间和1 个由5 个参数互相接近的特征矢量为基矢构成的五维子空间，通过求解三维矩阵和五维矩阵均可获得高精度的近似解.

然而，文献[16]中子空间基矢数量主要是以猜测的方式进行选取，因此，当模型粒子数目稍有增加，比如从4×4×4晶格增加到 6×6×6 晶格，获得高精度近似解需要的基矢是3 个还是5 个，或者说轨迹子空间到底应该是三维还是五维，亦或更高维，文献中并没能给出明确答案.确定或至少估计轨迹子空间维度大小是重要的:一方面，如果轨迹子空间维度大小随模型所含粒子数目的增加而迅速增加，也就是采用特征矢量延拓方法所需的基矢数量迅速增加，特征矢量延拓方法将很难奏效;另一方面，如果在特征矢量延拓计算中选用的基矢数目大于轨迹子空间的维度，基矢有可能是线性相关的，从而导致不合理的错误结果.

确定或估计轨迹子空间维度大小的困难在于:尽管随参数光滑变化的高维哈密顿矩阵的特征矢量变化轨迹所在的子空间是低维的，低维子空间的维度与多体模型所含粒子数目之间并不显含简单直接的函数关系.本文系统性地研究了反铁磁Heisenberg 链模型哈密顿矩阵的特征矢量随参数光滑变化的轨迹，通过对特征矢量轨迹进行主成分分析画出其所在低维子空间，指出采用特征矢量延拓求解反铁磁Heisenberg 链模型哈密顿量所需基矢数目随其所包含自旋个数的增加而增加.

2 研究内容和方法

采用由L个(L为偶数)自旋格点组成的，具有周期边界条件的一维Heisenberg 链模型(以下简称为模型)，如图1 所示.模型哈密顿量为

图1 一维Heisenberg 链模型示意图Fig.1.Schematic diagram of a one-dimensional Heisenberg chain model.

当参数θ∈[0°，90°]，Ji≥0，模型为反铁磁Heisenberg 链.

哈密顿量(1)的每个特征波函数可以对应于1 个N维矢量.由于单个自旋的波函数可以由1 个2 维矢量表示，两个维度可以分别对应于Sz向上和向下的状态，从而N(L)2L，波函数矢量和哈密顿矩阵的维度随L的增加而指数增加.在本文中，研究该模型的基态，它对应的总自旋角动量z分量0，因此自旋算符Sz向上和向下的状态数都是L/2个，所以波函数的维度N减 少 为N(L).例如N(4)=6，N(6)=20，N(14)=3432，N(16)=12870，N依然随自旋个数L的增加而迅速增加.只有对自旋个数L不太大的情形，多体哈密顿量(1)的特征波函数和特征能量可通过精确对角化哈密顿矩阵的数值方法严格求解.

在本文中，对L4，6，···，14 的模型哈密顿量的基态矢量和能量进行了系统性研究.对每一个L，将哈密顿矩阵的参数θ从0°到90°每隔1°连续变化，通过对角化求解基态矢量和能量，所获得的M个基态矢量轨迹构成一个M ×N的矩阵，记为XL，其中M91，N.

我们的目标是通过对具有不同自旋个数L的模型基态矢量轨迹矩阵XL分别进行主成分分析[17-20]，找出轨迹所在低维子空间的维度大小随自旋个数L的变化关系，从而用于指导利用特征矢量延拓法求解模型哈密顿矩阵的基态矢量和能量.主成分分析是一种非监督学习算法，被广泛应用于数据特征提取[17-20].在轨迹矩阵XL中，每一行是1 个基态波函数，对应于1 个N维空间中的矢量;根据特征矢量延拓法的思路[16]，XL中的M个基态矢量所构成的轨迹仅在部分基矢方向具有比较大的投影，而且与其他基矢方向近似正交.通过主成分分析找出这部分基矢，并将原始高维基态矢量在其方向上投影，从而可以在低维空间中直观而生动的展示轨迹图像.

由于矩阵XL不是方阵，首先对其进行变换Y，然后求解Y矩阵的特征值和特征矢量，即

其中特征值λn按从大到小的顺序排列λ1≥λ2≥λ3≥···≥λN≥0，而对应的特征矢量v1，v2，v3，···，vN是相互正交的，称具有较大特征值λn的特征矢量方向vn为主成分方向，原始高维波函数轨迹可以投影到相应主成分vn方向进行展示，即

其中PCn是波函数轨迹在第n(n1，2，3，···，N)个主成分方向上的投影.

3 研究结果

首先计算了L4的模型基态矢量随θ变化的轨迹并将其投影到各主成分方向，发现其仅在2 个主成分方向v1，v2的投影PC1，PC2不为零，其中

而在其他主成分方向的投影都小于 10-8.图2(a)用v1，v2构成的2 维空间中单位圆上的一段圆弧完整直观地展示了L4的模型基态矢量随θ变化的轨迹，说明尽管基态矢量是6 维的，其随θ变化的轨迹仅处于由前两个主成分方向所张成的2 维空间中.接着计算了L6的模型基态矢量随θ变化的轨迹并将其投影到各主成分方向，发现其不为0的主成分分量个数不再为2，而是增加到了3.图3(a)用这3 个主成分方向构成的3 维空间中单位球上的一段弧线完整直观地展示了L6 的模型基态矢量随θ变化的轨迹.

图2 L=4的Heisenberg 链 (a) 基态矢量随 θ 变 化 的轨迹(从红到蓝是0°到90°);(b) 在6 维和2 维空间中分别求得的基态能量Fig.2.The Heisenberg chain with L=4 :(a) Ground state vector trajectory (from red to blue:θ =0° to θ=90°);(b) ground state energies calculated in a 6-dimensional space and in a 2-dimensioanl space，respectively.

图3 L=6的Heisenberg 链 (a) 基态矢量随 θ 变化的轨迹(从红到蓝是0°到90°);(b) 在20 维和3 维空间中分别求得的基态能量Fig.3.The Heisenberg chain with L=6 :(a) Ground state vector trajectory (from red to blue:θ=0° to θ=90°);(b) ground state energies calculated in a 20-dimensional space and in a 3-dimensioanl space，respectively.

以上通过主成分分析方法获得了L=4 或6的模型基态矢量随θ变化的轨迹子空间的维度为2 或3，也就确定了采用特征矢量延拓方法求解具有其他角度θ的基态所需的基矢数目为2 或3.于是对L4 的模型哈密顿量分别在原6 维空间(N(4)6)中和2 维轨迹子空间中求解基态能量，记为E4，6和E4，2，画在图2(b)当中.毫不奇怪地，E4，2和E4，6几乎一样，它们的最大绝对误差，δE|E4，2(θm)-E4，6(θm)|＜10-15.同样地，对L=6 的模型哈密顿量分别在原20 维空间(N(6)20)中和3 维轨迹子空间中求解基态能量，并记为E6，20和E6，3，如图3(b);E6，3和E6，20的最大绝对误差 δE也小于 10-15.然而，确定采用特征矢量延拓方法求解L=4 或6 的模型所需的基矢数目并不是非常有用，因为利用计算机求解6×6或20×20 的哈密顿矩阵并不非常困难.我们的目标是能够确定或估计采用特征矢量延拓方法求解具有更大L的模型所需的基矢数目.随着模型中自旋数目L的增加，基态矢量的原空间维度N(L)迅速增长，例如，N(16)12870，利用计算机求解N ×N的哈密顿矩阵变得困难，因此需要找出轨迹子空间维度，从而用于指导特征矢量延拓方法的求解计算.

尽管可以用主成分分析方法确定对L=4 或6的模型基态能量进行特征矢量延拓方法求解所需的基矢数目，当模型变大以至于用计算机求解N ×N的哈密顿矩阵变得非常困难甚至不可能时，主成分分析方法将不再适合用于确定或估计采用特征矢量延拓方法求解所需的基矢数目;因为首先需要进行N ×N的哈密顿矩阵求解获得足够数量M的基态矢量，比如本文中，M91，然后才能对这M个矢量进行主成分分析.为解决这个困难，进一步研究了采用特征矢量延拓求解模型基态能量所需基矢数目与其所包含的自旋个数之间的关系.

对L8的模型基态矢量随θ变化的轨迹进行主成分分析获得各主成分方向，并分别在原70 维空间(N(8)70)中和由前Nsub个主成分方向构成的子空间中求解基态能量，记为E8，70和E8，Nsub.当Nsub6 ，E8，70和的最大绝对误差 δE小于10-7.同样，对L10的模型，当Nsub8 ，E10，252和的最大绝对误差 δE小于10-7;对L12的模型，当Nsub8 ，E12，924和的最大绝对误差 δE小于10-7;对L14的模型，当Nsub9，E12，924和E12，Nsub的最大绝对误差 δE小于10-7.明显地，随着L的增加，Nsub也在增加，这个事实必须在采用特征矢量延拓进行计算时加以考虑.将Nsub作为L的函数画在图4(a)中，发现Nsub随L的增加而迅速增长，尽管其总是远远小于N.Nsub的快速增长对于特征矢量延拓方法的应用是非常不利的.幸运的是，当我们稍稍降低计算结果的精度要求，比如从δE ＜10-7到 δE ＜10-4，尽管对L4和6，Nsub(L)保持不 变，Nsub(8) 则从6 降到了5，Nsub(10)从8 降到了5，Nsub(12) 从8 降到了6，Nsub(14) 从9 降到了6.从图4(a)可以看出，如果采用 δE ＜10-4，当L≤8时，Nsub随L的增加而快速增长，而当L≥8时，Nsub的 增长速度开始变得缓慢，从而为特征矢量延拓方法的应用提供了可能性.

图4 (a) 采用特征矢量延拓方法求解随 θ 变化的Heisenberg 链基态能量所需基矢数目与其所含自旋个数之间的关系;(b) L=16 的Heisenberg 链在10 维和6 维空间中分别求得的基态能量Fig.4.(a) Relationship between the number of basis vectors needed to calculate θ -dependent ground state energies of the Heisenberg chain by eigenvector continuation and the number of spins in the chain;(b) ground state energies of the Heisenberg chain with L=16 calculated in a 10-dimensional space and in a 6-dimensioanl space，respectively.

图4(a)清楚地展示了采用特征矢量延拓方法求解随θ变化的模型基态能量所需基矢数目随其自旋个数增加而增加的关系图像，因此可用于指导求解具有更多自旋的模型基态.通过延伸图4(a)中的关系曲线(图中的虚线)，可以预测，如果采用δE ＜10-4，Nsub(16)6;而如果采用 δE ＜10-7，Nsub(16)10.于是分别计算了随θ变化的L16的模型基态能量E16，6，E16，10，如图4(b)，其中6 维子空间的基矢由θ=0°，18° ，···，72°，90°的基态矢量构成，10 维子空间的基矢由θ=0°，10°，···，80°，90°的基态矢量构成.从图4(b)可以看出，E16，6和E16，10非常吻合，计算可得E16，6和E16，10的最大绝对误差 δE小于 10-4，证实了图4(a)中关系曲线的预测能力.

4 结论

本文通过采用主成分分析方法研究反铁磁Heisenberg 链基态多体波函数矢量随参数变化的轨迹，画出了采用特征矢量延拓方法求解其基态能量所需基矢数目随其所包含自旋个数增加而增加的函数曲线，用于指导采用特征矢量延拓方法求解包含更多自旋的Heisenberg 链模型哈密顿量.将类似的方法应用于指导求解高维Heisenberg 模型和其他关联量子模型以总结更一般的函数关系，将在后续文章中报道.