八、TEBD 与 DMRG 算法

八、TEBD 与 DMRG 算法


1. TEBD 算法


       我们给定一个大小为 W×HW\times H 个张量组成的网络(设 H 为偶数),其图形表示如下:
八、TEBD 与 DMRG 算法

该张量网络由三种不同的张量构成,在四个角上的为二阶张量,边上的为三阶张量,内部的为四阶张量,边上的数字表示第几个指标。很多物理问题可最终等价为类似的张量网络收缩计算问题,即将该张量网络收缩为一个数。如,经典模型热力学配分函数的计算,量子格点模型基态的计算等。严格收缩这种张量网络的计算,其代价随着 WWHH 呈指数增大,属于 NP 难问题,因此必须采用近似方法收缩计算。

       TEBD 算法(time-evolving block decimation) 为一种基于矩阵乘积态的近似收缩张量网络的数值算法。其主要思路为:从处于边界的张量构成的 MPS 开始,一行一行地收缩张量网络。

       将张量网络从中间分成两部分,以下半部分为例,下边界的张量实际上组成了一个长度为 WW 的 MPS 态,记为 φD0|\varphi^{D_0}\rangle ,整个下半部分的收缩结果可记为另一个 MPS 态 φD|\varphi^{D}\rangle ,中间每一层张量构成一个作用在 φD0|\varphi^{D_0}\rangle 上的算符,记为 ρ^\hat{\rho} ,那么我们可以用符号表示下半部分的收缩为:
φD=ρ^H21φD0\left|\varphi^{D}\right\rangle= \hat{\rho}^{\frac{H}{2}-1} \left|\varphi^{D_{0}}\right\rangle

我们可以用图形表示为:
八、TEBD 与 DMRG 算法

类似的,张量的上半部分可以相应的定义为 φU0|\varphi^{U_0}\rangleφU|\varphi^{U}\rangle ,同样的,我们有:
φU=(ρ^T)H21φU0\left|\varphi^{U}\right\rangle= \left(\hat{\rho}^{T}\right)^{\frac{H}{2}-1} \left|\varphi^{U_{0}}\right\rangle

那么整个张量网络的收缩结果为:
Z=φUφD=φU0ρ^H2φD0\mathrm{Z} =\left\langle\varphi^{U} \mid \varphi^{D}\right\rangle =\left\langle\varphi^{U_{0}} \left|\hat{\rho}^{H-2}\right| \varphi^{D_{0}}\right\rangle

所以,张量网络的 收缩变成了计算 φU|\varphi^{U}\rangleφD|\varphi^{D}\rangle

具体的计算思路为:通过进行下图所示的张量收缩,计算 (H21)(\frac{H}{2}-1)ρ^\hat{\rho} 对 MPS 的作用, ρ^\hat{\rho} 被称为转移矩阵(transfer matrix),这里 ρ^\hat{\rho} 具备的张量收缩的形式被称为矩阵乘积算符(matrix product operator , MPO)。
八、TEBD 与 DMRG 算法

张量的每次收缩都会使 MPS 辅助指标的维数扩大 D 倍(D 为 ρ^\hat{\rho} 水平方向指标的维数),也就是说,MPS 辅助指标的维数随收缩次数呈指数增大。所以我们引入 MPS 的最优裁剪,将维数限定为预先设定的截断维数 χ\chi ,也就是对 MPS 收缩后依次每个张量进行中心正交化,然后对该张量进行奇异值分解,保留前 χ\chi 个奇异值和奇异向量,保证每个辅助指标的大小都不超过 χ\chi。具体的算符步骤如下:

  1. 用处于边界的对应张量初始化 φU0|\varphi^{U_0}\rangleφD0|\varphi^{D_0}\rangle
  2. 计算 φUt+1=ρ^TφUt,φDt+1=ρ^φDt\left|\varphi^{U_{t+1}}\right\rangle= \hat{\rho}^{T}\left|\varphi^{U_{t}}\right\rangle, \quad\left|\varphi^{D_{t+1}}\right\rangle= \hat{\rho}\left|\varphi^{D_{t}}\right\rangle ,得到新的 MPS 态
  3. 如果 MPS 辅助指标的维数超过截断维数 χ\chi ,那么进行最优化裁剪,使辅助指标截断到 χ\chi
  4. 进行第二步和第三步 H21\frac{H}{2}-1 次后,计算 Z=φUφDZ=\langle\varphi^U|\varphi^D \rangle

TEBD 算法计算一维格点模型的基态

       我们来考虑 N 个自旋构成的一维海森堡模型:
H^=i,jH^ij,H^ij=α=x,y,zs^iαs^jα\widehat{H}= \sum_{\langle i, j\rangle} \widehat{H}_{i j}, \quad \widehat{H}_{i j}= \sum_{\alpha=x, y, z} \hat{s}_{i}^{\alpha} \hat{s}_{j}^{\alpha}

其主要思路为:利用基于 TRotter - Suzuki 分解的退火算法,将随机初始化的 MPS 态演化到基态,我们将一次一次的演化过程转换为张量网络的收缩问题,并使用 TEBD 算法求解该收缩。其具体步骤为

  1. 获得演化算符 eτH^ije^{-\tau \widehat{H}_{i j}} 的系数张量(τ\tau 为小的正实数),当哈密顿算符确定后,我们就能用下式计算出每个演化算符
    Gs1s2s1s2=s1s2eτH^ijs1s2G_{s_1's_{2}' s_{1} s_{2}}= \left\langle s_{1}^{\prime} s_{2}^{\prime} \left|e^{-\tau \widehat{H}_{i j}}\right| s_{1} s_{2}\right\rangle
    用图形可以表示为:
    八、TEBD 与 DMRG 算法 通过计算,我们可以将基态演化算符
    P^=limK(eτi,jH^ij)K\widehat{P}= \lim _{K \rightarrow \infty} \left(e^{-\tau \sum_{\langle i, j\rangle} \hat{H}_{i j}}\right)^K

    表示为由四阶张量 G 构成的张量网络,如下图
    八、TEBD 与 DMRG 算法

  2. 随机初始化一个 MPS 态,放置于张量底部,整个张量网络变为下图所示
    八、TEBD 与 DMRG 算法

  3. 将靠近MPS的半层张量与MPS中相关的张量进行收缩,部分近邻的两个张量被收缩成为一个四阶张量,如下图
    八、TEBD 与 DMRG 算法

  4. 利用规范变换,将 MPS 变换为中心正交形式,正交中心为第一个四阶张量。

  5. 对中心张量进行奇异值分解,如果奇异谱的维数大于预设的截断维数 χ\chi,则仅保留 χ\chi 个最大的奇异值与相应的奇异向量
    八、TEBD 与 DMRG 算法

  6. 移动正则中心到下一个四阶张量,重复第五步,直到所有四阶张量被分解为三阶张量

  7. 进行完上述步骤后,MPS 被变换称为演化前的形式,即都是三阶张量的形式,重复第三步到第六步,收缩下一个半层张量直到MPS收敛。

TEBD 算法的具体实施方案并不唯一,我们可以每次只演化一个局域门,将一个局域门演化为三阶张量后再去演化下一个,如下图所示:
八、TEBD 与 DMRG 算法


2. DMRG 密度矩阵重整化群算法


       TEBD 算法计算基态的思路是退火,密度矩阵重整化群算法(DMRG)采用基于最大本征态求解对应的最优化问题,即求解下面的极小化问题:
Eg=minφφ=1φH^φE_{g}=\min _{\langle\varphi \mid \varphi\rangle=1}\langle\varphi|\widehat{H}| \varphi\rangle

当哈密顿量为 H^=i,jH^ij\widehat{H}=\sum_{\langle i, j\rangle} \widehat{H}_{i j} ,有:
Eg=minφφ=1i,jφH^ijφE_{g}=\min _{\langle\varphi \mid \varphi\rangle=1} \sum_{\langle i, j\rangle}\left\langle\varphi\left|\widehat{H}_{i j}\right| \varphi\right\rangle

他们每一项对应的张量网络图如下所示,以 φH^23φ\left\langle\varphi\left|\widehat{H}_{23}\right| \varphi\right\rangle 为例可以用图表示为:
八、TEBD 与 DMRG 算法DMRG 的策略是更新各个张量,使能量达到最小,具体的跟新策略也不唯一。

       在 单点 DMRG 中,每次更新 MPS 中的一个张量,其余张量看成是给定的参数,一起来更新一个张量,该方法称为交替最小二乘算法(ALS)。

下面我们将单个张量的优化问题等价为局域矩阵最大本征值的问题,以更新第二个张量为例,定义有效哈密顿量 h2^\hat{h_2} 为下面各项的求和:
八、TEBD 与 DMRG 算法

其中每一项收缩之后的结果为六阶张量,h2^\hat{h_2} 为多个六阶张量的求和,所以也是六阶张量,我们表示为
八、TEBD 与 DMRG 算法

上面的过程就是我们先计算跟第二个张量无关的收缩,得到张量,下面我们将第二个张量和得到的六阶张量收缩,得到系统的能量变为:
E=A(2)h^2A(2)E= \left\langle A^{(2)} \left| \widehat{h}_{2}\right| A^{(2)}\right\rangle

可以用图形表示为:
八、TEBD 与 DMRG 算法

最优化问题变为了:
minφφ=1A(2)h^2A(2)\min _{\langle\varphi \mid \varphi\rangle=1}\left\langle A^{(2)}\left|\hat{h}_{2}\right| A^{(2)}\right\rangle

此时我们可以通过规范变换,将正则中心移动至 A(2)A^{(2)} ,有:
φφ=1A(2)=1\langle\varphi \mid \varphi\rangle=1 \Leftrightarrow\left|A^{(2)}\right|=1

所以最优化问题最终变成了:
minA(2)=1A(2)h^2A(2)\min _{\left|A^{(2)}\right|=1}\left\langle A^{(2)}\left|\hat{h}_{2}\right| A^{(2)}\right\rangle

所以 A(2)A^{(2)} 就是算符 h2^\hat{h_2} 的最低本征态,我们应该将 A(2)A^{(2)} 更新为 h2^\hat{h_2} 的最低本征态,求解 h2^\hat{h_2} 的最低本征态的计算复杂度是可控的,也就是求解维数为 dχ2×dχ2d \chi^{2} \times d \chi^{2} 的厄密矩阵的本征态。我们直接将 h2^\hat{h_2} 转换为矩阵,进行本征值分解,得到最小的本征态替换为 A(2)A^{(2)}

我们可以用上面的方式,不断地重复更新 A(1)A(2)A(3)A(4)A(5)A(6)A(7)A^{(1)}、A^{(2)}、A^{(3)}、A^{(4)}、A^{(5)}、A^{(6)}、A^{(7)} ,直到 MPS 态收敛。

得到基态MPS后,我们可利用中心正交形式简化对算符观测量的计算,当正交中心与观测量所在格点重合时,我们可以将下面左图的计算简化为右图:
八、TEBD 与 DMRG 算法当正交中心不与观测量所在格点重合时,在不移动正交中心的情况下,计算可做例如下图的简化:
八、TEBD 与 DMRG 算法

综上所述,DMRG 算法,对第 n 个张量 A(n)A^{(n)} 的更新步骤为:

  1. 将MPS正交中心移动到第 n 个张量
  2. 计算第 n 个张量对应的有效哈密顿量 hn^\hat{h_n}
  3. A(n)A^{(n)} 更新为 hn^\hat{h_n} 的最低本征态

整个 DMRG 算法的步骤为:

  1. 随机初始化 MPS 态中的各个张量 {A(n)}\left\{A^{(n)}\right\}
  2. 按上面的步骤一次更新第 1 到 n 个张量,再依次更新第 n 到第 1 个张量
  3. 如果 MPS 收敛,计算完成,否则重复进行第二步

DMRG算法的关键在于将整个MPS的优化问题化成多个有效哈密顿量的最小本征问题,循环优化直至MPS收敛。


3. 基于自动微分的基态变分算法


       自动微分技术用来在机器学习中计算模型变分参数关于损失函数的梯度,我们称之为反向传播算法。

       这里我们可以用自动微分算法求解实对称矩阵的最大本征向量,即求解如下的极大值问题
maxv=1vTMv\max _{|v|=1}\left|v^{\mathrm{T}} M v\right|

       我们可以在上面的式子前添加一个符号将求解最大值问题转换为求解极小值问题,我们定义损失函数
f=vTMvv2f=-\frac{v^{\mathrm{T}} M v}{|v|^{2}}

其中的 vv\frac{v}{|v|} 是对向量的归一化,因此该式满足上面的 v=1|v|=1 的条件,该损失函数的最小本征向量就是 vv\frac{v}{|v|}

       为了计算 v ,我们可以计算 vv 关于 ff 的梯度 dfdv\frac{df}{dv} ,要使得 ff 减小,需要将 vv 沿负梯度方向进行更新,即每次将 vv 做如下改变:
vvηdfdvv \Rightarrow v-\eta \frac{d f}{d v}

其中,η\eta 为认为给定的常数,被称为更新步长或学习率。

       进行多次迭代更新之后,得到收敛的 vv ,那么最大本征向量 v~\tilde{v} 与最大本征值 λ\lambda 满足:
v~=vv,λ=v~TMv~\tilde{v}=\frac{v}{|v|}, \quad \lambda=\tilde{v}^{\mathrm{T}} M \tilde{v}

我们可以用程序实现上述算法,上述算法中的梯度 dfdv\frac{d f}{d v} 可以直接使用自动微分技术计算,可以使用优化器让程序自适应地控制学习率,以达到较好的稳定性和收敛速率。

       根据最大本征值问题与基态问题地等价性,显然,自动微分方法可用于求解多体系统地基态,对应的优化问题可以写为:
E=min{A(n)}φH^φφφE=\min _{\{A^{(n)}\}} \frac{\langle\varphi|\widehat{H}| \varphi\rangle}{\langle\varphi \mid \varphi\rangle}

其中,变分参数为 MPS 中地各个张量 {A(n)}\{A^{(n)}\} ,梯度更新地公式为:
A(n)A(n)ηEA(n)A^{(n)} \Rightarrow A^{(n)}-\eta \frac{\partial E}{\partial A^{(n)}}

这里我们可以更加直接地将 MPS 看作是对基态地一种特殊的参数化形式,能量就是变分的损失函数。之后我们就可以使用 BP 算法和各种优化器进行梯度的更新了。