统计物理中的机器学习重整化群

Wanda Hou    Yi-Zhuang You Department of Physics, University of California at San Diego, La Jolla, CA 92093, USA
摘要

我们开发了一种机器学习重整化群 (MLRG) 算法来探索和分析统计物理中的多体晶格模型。 利用生成式建模的表征学习能力,MLRG 自动学习最优重整化群 (RG) 变换,这些变换来自自生成的自旋构型,并在没有人工监督的情况下制定 RG 方程。 该算法不专注于模拟任何特定的晶格模型,而是广泛地探索所有与内部和晶格对称性兼容的可能模型,前提是给定现场对称性表示。 它可以发现控制 RG 流的 RG 单调,假设 c-定理的强形式。 这使得一些下游任务成为可能,包括相的无监督分类、相变或临界点的自动定位、临界指数和算子尺度维度的受控估计。 我们在具有伊辛对称性的二维晶格模型中演示了 MLRG 方法,并表明该算法正确地识别并表征了伊辛临界性。

I 引言

重整化群 (RG) 是统计物理和量子场论中一个优雅的概念框架和强大的计算方法。 RG 通过分层地逐步粗化物理系统中的自由度来提取每个给定尺度上的相关特征。 通常,实空间 RG 依赖于人类物理学家根据他们对物理系统的直觉来设计粗粒化变换。 在这项研究中,我们旨在探索人工智能 (AI) 从能量模型中自动学习最优 RG 变换的潜力。 无监督机器学习特别适合这项任务,因为它能够从数据中学习低维表示或相关特征,并在没有监督的情况下去除噪声和无关特征。 这种方法与学习 RG 变换的目标一致,RG 变换旨在通过将细粒度配置转换为粗粒度配置来提取物理系统的相关特征,同时保留必要的信息和相关性。

先前的研究表明,神经网络可以学习在配置级别执行分层特征提取 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11] 然而,RG 更令人着迷的一方面是它能够在模型层面上定量分析物理理论在参数空间中的流动 [12, 13]. 因此,本研究旨在开发一种新颖的机器学习重整化群 (MLRG) 方法,该方法能够自动推导出 RG 流方程,发现 RG 单调函数,提出有效理论,识别临界点并估计临界指数,所有这些都从物理系统的对称性和维度开始。

Refer to caption
图 1: MLRG 算法的高级架构。 涉及三个 AI 模型。 教师 模型和 学生 模型都是基于 RBM 的生成模型,具有不同的预定义神经网络连接,分别代表细粒度和粗粒度局部能量模型。 调节器 模型是一个判别模型,用于预测 RG 单调函数。 调节器首先引导哈密顿蒙特卡罗 (HMC) 采样器来提出最值得训练的教师模型参数,然后根据 RG 流预测学生模型参数。 在两个 RBM 都获得其模型参数后,教师会生成用于训练学生的数据。 梯度从学生反向传播到调节器,以训练 RG 单调网络。 训练后,调节器获得了关于整个参数空间的 RG 流的良好知识,这些知识随后可用于各种下游任务。

在本研究中,我们将重点关注定义在规则格上的统计力学模型,并开发机器学习算法,以在实空间 RG 框架内分析这些模型。 我们提出的 MLRG 算法如图 1 所示。 我们引入了两个受限玻尔兹曼机 (RBM) [14, 15, 16, 17],分别对细粒度和粗粒度格上的不同尺度上的局部能量模型进行建模。 细粒度模型充当教师,通过生成可见配置样本,这些样本随后用于训练粗粒度模型。 粗粒度模型充当学生,并学习其能量模型来描述教师提供的训练数据。 我们还引入了一个第三个模型,RG 单调网络,作为观察师生学习过程并学习预测粗粒度模型的模型参数如何与细粒度模型的模型参数相关的调节器。 这使得调节器模型能够学习 RG 流动并利用其知识来指导哈密顿蒙特卡罗 (HMC) [18, 19, 20] 采样器提出最值得训练的新模型参数。 训练后,我们可以使用机器学习的 RG 流动来识别参数空间中的 RG 固定点,并自动分析这些固定点处的物理性质。

本文将按以下顺序组织。 我们将在 Sec. II 中首先介绍 MLRG 算法,其中包括 (i) 师生学习系统 Sec. II.2 来模拟 RG 流动,以及 (ii) 调节器 Sec. II.4 和 HMC 采样系统 Sec. II.5 用于提取 RG 单调并用它来指导训练。 教师和学生由等变 RBM 建模,如 Sec. II.1 中所述,其点群表示选择在 Sec. II.3 中详细说明。 然后我们在 Sec. III 中演示了 MLRG 在二维 Ising 模型上的应用。 Sec. III.1 描述了问题设置。 Sec. III.2 展示了机器学习的 RG 单调和 RG 流动图。 然后给出了一些定量结果,包括临界点 Sec. III.3、基态简并度 Sec. III.4 和标度维 Sec. III.6 Sec. III.5 解释了如何使用牛顿法来定位 RG 固定点。 我们在 Sec. IV 中总结了 MLRG 的优势和局限性,并评论了它与相关工作的联系。

II 方法论

II.1 统计力学模型

我们从晶格上统计力学系统的通用定义开始。 令晶格由图 𝒢=(𝒱,) 描述,其中 𝒱 表示顶点(位点)集,而 表示边(键)集。 在每个位置 i𝒱 上,我们引入一个向量 sin 来表征该位置上的自由度,通常被称为 自旋 格子上的整个自旋构型 s 可以被视为一个映射 s:𝒱n 平衡统计物理学的一个核心主题是使用局部能量函数 E(s) 对概率分布 p(s)eE(s) 进行建模,例如

EJ(s)=ijϵJ(si,sj), (1)

其中 ϵJ 是一个标量函数,描述了每对自旋 (si,sj) 在边 ij 上的能量。 下标 J 表示参数化能量模型的参数集合。

对称性在定义自旋自由度和约束能量函数方面起着重要作用。 G 为内部对称群,R 为格子的点群。 假设 GR 可交换,并形成一个直积群 G×R,则该位置上的自旋 si 应构成 G×R 对称性的 n 维线性表示。 在对称性作用 gGrR 下,自旋 si 变换为

si(TrTg)sr(i), (2)

其中 TgTr 分别是内部 g 和点群 r 对称性变换的矩阵表示。 r(i) 表示通过应用点群变换 r 从原始位置 i 获得的结果位置。

能量函数 EJ(s) 必须在内部对称性和格子对称性(包括点群和平移对称性)下保持不变。 这要求 ϵJ 满足以下对称性约束:

ϵJ(si,sj)=ϵJ((TrTg)sr(i),(TrTg)sr(j)) (3)

对于所有 (g,r)G×R 这意味着,在代表性键上定义能量函数 ϵJ 就足够了,并使用格子对称性将其传递到格子上所有其他键(假设格子上所有键都通过格子对称性变换相关联)。

Refer to caption
图 2: (a) 代表性键上的键能函数 ϵJ(si,sj) 的张量网络表示。 (b) 核 K 在内部对称性 gG 下是不变的。

更明确地说,假设自旋分量 siaα 由两个索引 aα 标记,分别索引点群和内部对称性的表示空间基。 在代表性键上,等变键能函数 ϵJ 的一种可能设计是使用张量网络:

ϵJ(si,sj)=JabγKαβγsiaαsjbβ, (4)

其中重复的索引自动累加,如图 2(a) 所示。 K 是一个 G 不变张量,满足对称性约束 TgααKαβγTgββ=Kαβγ (gG),如图 2(b) 所示。 张量 J 包含所有决定能量函数的参数。 它们可以被视为不同对称表示之间的耦合常数。 虽然等式 (4) 还没有代表最一般的等变能量模型,但如果表示空间维度 n 足够大,它已经足够表达。 所以我们不会深入研究更复杂的等变神经网络模型 [21, 22, 23, 24, 25, 26],而是采用这种张量网络设计来构建等变受限玻尔兹曼机。

总之,给定内部对称群 G 和晶格图 𝒢(晶格对称性由 𝒢 的自同构群给出),可以通过其在内部和点群对称性下的表示来指定现场自旋 si 统计力学模型通常由对称能量函数 EJ(s) 定义,由 J 参数化,如等式 (1) 所示。 RG 分析旨在确定模型参数 J 如何在不同尺度上有效地变化。

II.2 实空间重整化群

为具体起见,我们将重点关注二维晶格模型。 特别地,我们将考虑如图 3(a,b) 所示的 Lieb 晶格 [27](键插入方形晶格)。 在 Lieb 格子上定义的 Ising 模型在物理上等效于简单的方格,但 Lieb 格子更便于描述实空间 RG 方案。 一般来说,将我们的方法扩展到其他格子和更高维度是可能的。

Refer to caption
图 3: (a) 细粒度和 (b) 粗粒度 Lieb 格子。 阴影方块标记出局部块。 在每个块中,教师和学生 RBM 分别定义在 (c) 方形和 (d) 十字图上。 箭头标记键方向。 所有键都通过 C4v 点群变换相关,包括四重旋转 C4 和镜像反射 σ

从在 Lieb 格子上定义的能量模型开始,我们的 RG 方案可以描述如下:(i) 将格子划分为重叠的 2×2 块,如图 3(a) 所示。 (ii) 在每个块中,用方形图 3(c) 上的局部能量模型替换十字图 3(d) 上的新模型,使得它们在角自旋上的边缘分布尽可能匹配。 (iii) 将新的局部能量模型嵌入回原始格子中。 新格子 3(b) 成为粗粒度 Lieb 格子,晶格常数增大为 2 递归地重复该过程,能量模型参数将被重新归一化到越来越大的格子尺度。

此 RG 方案的主要目标是学习新的局部能量模型。 为此,我们将方形和十字图局部能量模型视为两个受限玻尔兹曼机 (RBM) [14, 15, 16, 17] 我们将细粒度方形图上的 RBM 称为 教师 机,其中角自旋 xi 是可见变量,装饰自旋 zi 是隐藏变量,参见图 3(c)。 它的能量模型如下:

EJtch(x,z)=ijsquareϵJtch(zi,xj). (5)

在粗粒度交叉图上的 RBM 将被称为 学生 机,其中角自旋 xi 仍然是可见变量,但中心只有一个隐藏自旋 z0,见图 3(d)。 它的能量模型是:

EJstd(x,z)=ijcrossϵJstd(xi,zj). (6)

教师和学生 RBM 分别由 JtchJstd 参数化。 在等式 (5) 和等式 (6) 中,键能函数 ϵJ(xi,xj) 在沿 ij 方向的代表性键上定义,使用张量网络模型等式 (4)。 在图 3(c,d) 中,代表性键用蓝色着色,箭头表示键方向。

正方形和交叉图都符合 C4v 点群对称性,该对称性由四重旋转 C4 和镜像反射 σ 生成,见图 3(c,d)。 镜像反射 σ 始终被分配以保留代表性键(蓝色)。 它仅根据等式 (3) 的要求对键能模型 ϵJ 强加对称性约束。 四重旋转 C4 进一步将代表性键带到其他键(其他颜色),在该键下,键能模型将根据等式 (3) 等变地变化。 这样,RBM 将根据设计尊重内部 G 和点群 R=C4v 对称性。

目标是训练学生 RBM 从教师 RBM 生成的可见自旋配置中学习粗粒度局部能量模型。 给定教师模型参数 Jtch,我们通过最小化两个模型的可见变量分布之间的 Kullback-Leibler (KL) 散度来优化学生模型参数 Jstd

DKL(ptch(x)pstd(x))=xptch(x)logptch(x)pstd(x), (7)

其中边缘分布 ptch(x)pstd(x) 是通过消去隐藏自旋来定义的:

ptch(x)=zeEJtch(x,z)x,zeEJtch(x,z),pstd(x)=zeEJstd(x,z)x,zeEJstd(x,z). (8)

虽然直接评估等式 (7) 中的 KL 散度是不可行的,但优化可以通过遵循 RBM 的标准对比发散 (CD) [15] 训练技术的随机梯度下降来执行。

通过这种方式,学生机器会自动学习粗粒度模型参数 Jstd,这些参数由教师机器的细粒度模型参数 Jtch 给定。 这与传统的真实空间 RG 方法形成对比,在传统的真实空间 RG 方法中,人类物理学家必须设计粗粒化规则,将可见自旋配置 x 映射到每个块中的隐藏自旋 z0(x) 我们的方法在两个方面有所不同:(i)粗粒化变量 z0 不再是 x 的确定性映射,而是通过学生机器给出的条件分布 pstd(z0|x) 以概率方式定义的。 (ii)条件分布 pstd(z0|x) 不是由人类指定的,而是从数据中学习的。 这使学生机器能够在其可用的表示空间维度内开发最佳的 RG 变换,从可见配置 x 中提取最相关的特征 z0,而无需监督。

训练完成后,我们可以用学生模型参数 Jstd 替换教师模型参数 Jtch,并继续训练下一代学生。 因此,一代一代地,师生学习迭代将追踪参数空间中 J 的 RG 流。

II.3 选择点群表示

为了证明学生 RBM 可以很好地逼近教师 RBM,我们考虑一个基于二维 Ising 模型在方形格子上具体例子。 在这种情况下,内部对称性是 G=2,点群对称性是 R=C4v 然后,两个 RBM 中的每个自旋变量(无论可见还是隐藏)都将被指定为 2×C4v 的表示。 2 群只有一个非平凡表示,即奇(带符号)表示,变换为 ss 因此,我们将假设自旋的每个分量在 2 Ising 对称性下变换为该奇表示。 C4v 点群具有更丰富的不可约表示,总结在表 1 中。 RBM 的表达能力取决于为每个自旋选择的 C4v 点群表示,这将在下面详细说明。

irrep dim transforms as TC4 Tσ
𝖠1 1 1 [1] [1]
𝖠2 1 xy(x2y2) [1] [1]
𝖡1 1 x2y2 [1] [1]
𝖡2 1 xy [1] [1]
𝖤 2 (x,y) [0110] [1001]
表 1: C4v 群的不可约表示。 列显示了它们的维度、笛卡尔积 (x 轴沿代表键) 以及 r=C4,σ 的变换矩阵 TrC4v 的两个生成元)。

在传统的 Ising 模型中,单个 Ising 自旋对应于一个 2 奇数变量,该变量携带 𝖠1 表示,它在点群对称性下根本不发生变换。 然而,在 RG 下,粗粒度自旋(例如学生 RBM 中的 z0 自旋)可以携带扩大的点群表示。 从图 3(c,d) 可以清楚地看到,RG 过程实际上是将教师模型中的四个隐藏自旋(方图)合并为学生模型中的一个隐藏自旋(十字图)。 因此,学生模型中的隐藏自旋 z0 必须包含更多内部结构并携带轨道角动量(即 C4v 的非平凡表示)以解决教师模型中不均匀隐藏自旋配置的复杂性。

例如,通过说一个 2 自旋变量 z0 携带 𝖠1𝖤 表示,我们隐含地认为 z03 是一个三维向量,由三个 Ising 变量组成:一个 Ising 变量形成 𝖠1 表示,在晶格旋转下保持不变,两个 Ising 变量形成二维 𝖤 表示,可以像向量一样旋转。 然后,C4σ 变换表示为

TC4z0=[100001010]z0,Tσz0=[100010001]z0, (9)

根据表 1 中列出的变换矩阵。

假设我们从一个细粒度的教师模型开始,该模型的可见自旋和隐藏自旋都是 𝖠1 表示中的普通 Ising 自旋,即 xi,zi{±1} 考虑到对称性约束,参数张量 Jtch 将只包含一个分量。 根据图 3(c) 中的方图,教师模型由以下能量函数描述

EJtch(x,z)=Jtchi=03zi(xi+x(i+1)mod4). (10)

隐藏自旋 zi 可以立即在配分函数中被迹出(边缘化),留下我们一个针对可见自旋 xi 的有效模型,其描述如下

EJtch(x)=Jeff(x0x1+x1x2+x2x3+x3x0), (11)

其中 Jeff=12logcosh(2Jtch) 是一个取决于 Jtch 的有效耦合。 相应地,可见自旋分布为 ptch(x)=Z1eEJtch(x)

要构建一个粗粒度的学生模型来近似 ptch(x),第一步是指定学生 RBM 中粗粒度变量 z0 的点群表示的选择。 例如,如果选择 z0 来携带 𝖠1𝖤 表示,该表示具有总共三个分量作为 z0=(z01,z02,z03),其中每个分量都是一个 2 Ising 变量,该变量 z0a{±1},那么学生模型可以写成(见图 3(d) 了解这些变量的连通性)

EJstd(x,z)=Jstd𝖠1𝖠1(x0+x1+x2+x3)z01Jstd𝖠1𝖤((x0x2)z02+(x1x3)z03). (12)

𝖠1表示z01的波动在每对可见自旋之间介导等量的正相关xixj,这确实是铁磁中的主要相关模式伊辛模型。 然而,为了更好地匹配教师模型 ptch(x)x0x2x1x3 应该相对于其他相关性进一步减弱,因为在原始教师模型等式 11 中,这些对角自旋之间没有直接耦合。 这就是 𝖤 表示发挥作用的地方,因为 z02z03 的波动将在 x0x2x1x3 中介导一些负相关,这是由于等式 12 中的负号,这有助于使学生模型的可见分布 pstd(x) 更接近教师模型 ptch(x) 的分布。 这种论点解释了为什么在学生模型中将更多点群表示包含到隐藏自旋中通常有助于提高其表达能力。

Refer to caption
图 4: 最小 KL 散度 DKL(ptch(x)pstd(x))(以比特为单位)与教师 RBM 参数 Jtch 的关系,针对学生 RBM 中 z0 的不同点群表示选择。

在这种设置下,我们测试了一系列学生 RBM 的性能,这些 RBM 是用不同的隐藏自旋 z0 点群表示选择构建的。 性能通过优化学生机器后实现的最小 KL 散度 DKL(ptch(x)pstd(x)) 来评估。 在图 4 中,我们展示了在 z0 的不同表示选择下,教师模型参数 Jtch 的一个较大范围内,最小 KL 散度的值。 一般表示是通过组合不可约表示 𝖠1𝖡1𝖤 来构建的。 其他两个不可约表示 𝖠2𝖡2 未在本特定计算中使用,因为由于镜像反射对称性 σ 的约束,它们不会与 𝖠1 中的 xi 耦合。 但在更一般的情况下,当 xi 的表示更大时,原则上所有不可约表示都可能出现在 z0 上。 此计算定量地证明了学生 RBM 可以学习从教师 RBM 逼近可见自旋分布。 通过在隐藏自旋 z0 上引入更大的表示,逼近可以逐步改进,即使精确匹配可能只在较大表示维数限制 n 下才能实现。

在实践中,RBM 只能处理有限的表示空间维数 n。因此,我们必须在最大维数 nmax 处截断表示空间。 这将在每个 RG 步中引入不精确性,但希望是截断引入的误差在 RG 流下变得无关紧要,这样截断只以可控的方式影响 RG 固定点行为。 这也是许多真实空间 RG 方案的潜在假设,其中每个 RG 步不需要精确地保留配分函数。

当我们通过递归地用教师 RBM 训练学生 RBM 并用训练好的学生替换教师来执行 MLRG 程序时,RG 流应该将我们(接近)带到某个固定点 RBM 模型,在这个模型中,我们可以进一步研究 RG 固定点的普适性质(如临界指数和算符尺度维数)。 我们将提供数值证据来表明,当表示维度 n 变大时,MLRG 算法可以更准确地定位 RG 固定点并估计其普适性质,因此它可以提供一种有用且可控的数值方法,在没有人工监督的情况下自动绘制相图并研究统计物理模型中的临界现象。

II.4 学习 RG 单调性

虽然上述用于真实空间 RG 的师生学习方法很有吸引力,但它在实际训练中面临着严峻的挑战。 这种挑战来自 RBM 训练的随机性,因此学生 RBM 参数 Jstd 将在每个 RG 步的最佳值附近波动。 这不可避免地将随机噪声注入整个 RG 流,导致模型在参数空间中进行随机游走。 由于临界点(即不稳定 RG 固定点)附近的 RG 流对微小扰动特别敏感,因此随机 RG 流几乎总是会错过临界点。 因此,如果不解决参数空间随机游走的问题,上述简单的 MLRG 算法对于研究任何临界现象都是无用的。

为了应对这一挑战,我们引入了第三个 AI 系统,称为 主持人,它在师生系统之外运行。 主持人的目标是在许多不同的场景下监控随机 RG 流随时间的变化,并学习整个参数空间的潜在确定性 RG 流。 这里的主要思想是假设存在一个 RG 单调性 C(J),它是一个关于 RBM 模型参数 J 的实标量函数,这样 RG 流就可以被表述为 RG 单调性 dJ/d=JC(J) 的梯度流,其中 参数化了 RG 步。 这是 c-定理 [28, 29, 30, 31] 的最强形式。 我们没有试图用人工来构建这种 RG 单调性,而是引入了一个前馈神经网络 Cθ(J),称为 RG 单调性网络,来建模函数 C(J) 并与 RG 流中的 RBM 联合训练。 这里 θ 表示神经网络中所有参数的集合。

训练从参数空间中教师模型参数 Jtch 的随机选择开始。 调节器采用初始条件 J(=0)=Jtch 并通过求解 RG 方程将 J()=0 进化到 =1

dJd=JCθ(J), (13)

遵循 RG 单调网络 Cθ 提供的梯度信号。 这里采用神经常微分方程 (neuralODE) 技术 [32, 33] 来实现通过 ODE 求解器的梯度反向传播。 然后,调节器将解传递给学生 RBM 作为其模型参数 Jstd=J(=1) 然后,教师 RBM 开始采样可见自旋配置并将它们发送给学生 RBM。 学生 RBM 接收训练数据并评估等式 ( 7 ) 中的 KL 散度作为总损失函数。 所有模型通过使用 CD 训练技术最小化 KL 散度来联合训练。 损失函数梯度最终将反向传播到 θ 以更新 RG 单调网络。 训练在许多随机选择的初始参数 Jtch 上进行,这样可以聚合大量训练数据以优化整个参数空间的 RG 单调性。

尽管训练的本质仍然是随机的,但随机噪声将在拟合 RG 单调性时被平均掉,这样在训练之后,最佳拟合 Cθ(J) 可用于生成遵循等式 ( 13 ) 的确定性 RG 流。 通过这种方式,可以避免参数空间中的随机游走行为,从而使 MLRG 算法能够准确地定位 RG 不动点。

II.5 参数空间采样

接下来,我们将讨论如何更有效地对 Jtch 进行采样以加快训练速度。 由于 RBM 模型参数 J 存在于高维参数空间中,均匀采样可能不是一种有效的策略。 我们提出了一种重要性采样策略,该策略专注于 RG 不动点。

采样器利用有关 RG 单调性的知识从以下概率分布中对参数 J 进行采样

p(J)eβJCθ(J)2, (14)

作为超参数,使用一些反温度 β 我们最初设置 β=0,使得采样器将在参数空间中均匀地提出模型参数 J 随着训练的进行,RG 单调网络逐渐积累了关于 RG 流的知识。 RG 固定点作为梯度 JCθ(J)=0 消失的局部鞍点出现。 然后,我们逐渐增加 β,因此采样器将被偏向于更多地围绕 RG 固定点(包括稳定和不稳定)进行采样。 这种设计鼓励 RG 单调网络在 RG 固定点附近接受更密集的训练,以便可以更准确地估计固定点位置,这是统计力学系统中自动发现临界点(相变)的理想特征。

由于 J 连续变化,哈密顿蒙特卡罗 (HMC) 方法 [18, 19, 20] 成为采样方法的自然选择。 HMC 是一种马尔可夫链蒙特卡罗算法,它使用哈密顿动力学来有效地提出目标分布 p(J) 参数空间中的移动。 它是用于对高维空间中连续变量的复杂分布进行采样的强大方法。 我们使用多个具有副本交换的 HMC 采样器来实现该方法,以减轻陷入单个固定点的可能性。

总之,RG 单调网络 Cθ(J) 在 MLRG 算法中起着至关重要的作用,如图 1 所示。 它首先引导 HMC 采样器根据 Eq. (14) 中的概率分布提出新的教师模型参数 Jtch,然后通过求解 Eq. (13) 中的 RG 流微分方程来预测学生模型参数 Jstd 从这个意义上说,它确实充当了一个调节器,来调节教学-学习过程。 相反,当学生 RBM 学习生成与教师 RBM 生成的相似的自旋配置时,RG 单调网络会得到训练。 值得一提的是,HMC 采样的教师模型参数 Jtch 与计算图分离,因此 RG 单调网络参数 θ 将只接收来自学生模型参数 Jstd 的梯度信号。 通过这种方式,RG 单调的训练不会受到 HMC 采样的影响。

III 结果与分析

III.1 对称赋值和耦合参数

为了应用 MLRG 方法,我们专注于具有 Ising 对称性的二维晶格模型。 我们不需要指定任何特定的 Ising 模型哈密顿量,因为 MLRG 可以自动探索所有与内部和晶格对称性一致的模型,前提是给定了现场对称性表示。

以下,我们将始终采用 G=2 内部对称性和 R=C4v 点群对称性。 关于现场自旋的对称性表示,我们认为每个自旋(无论可见还是隐藏)在 2 下都是奇数,并且在 C4v 下具有 𝖠1𝖠1𝖤2𝖠1𝖤 表示。 这些点群表示的选择基于我们的经验,即它们是各自表示维度中最有效的表示,可以最小化教师和学生 RBM 之间的 KL 散度,如图 4 所示。 上述对称性赋值总结在表 2 中,这是我们设置等变 RBM 并准备 MLRG 模型进行训练所需的全部信息。

symmetry representation
G 2 1
R C4v 𝖠1,𝖠1𝖤,2𝖠1𝖤
表 2: 二维 Ising MLRG 的对称性和自旋表示设置。

对于 2 自旋,Eq. (4) 中的(代表性的)键能函数简化为以下形式

ϵJ(si,sj)=Jabsiasjb, (15)

其中 a,b 标注 C4v 对称性表示的基础。 Eq. (4) 中的内部对称性表示标签 α,β 被省略,因为 2 群只有一个非平凡的不可约表示。 耦合 J 是一个矩阵,它可以在不同点群表示的选择下参数化如下:

  • 𝖠1 表示(一维,1 个耦合参数)

    J=[J0], (16)
  • 𝖠1𝖤 表示(三维,5 个耦合参数)

    J=[J0J10J2J3000J4], (17)
  • 2𝖠1𝖤 (4 维,10 个耦合参数)

    J=[J0J1J20J3J4J50J6J7J80000J9]. (18)

在上述矩阵中,我们使用线条将 C4v 对称性的不同不可约表示分开。 一些矩阵元素为零,因为镜面反射对称性 σ 施加了限制,正如方程式 ( 3 ) 在一般情况下所要求的那样。 在所有情况下,参数 J0 始终表示第一个 𝖠1 之间的伊辛耦合,直接连接到晶格模型中的裸伊辛耦合。

III.2 RG 单调性和 RG 流

选择最小的点群表示 𝖠1,耦合矩阵 J 由单个变量 J0 参数化。 在这种情况下,RG 流在单维参数空间中简单地定义。 通过 MLRG 方法确定的 RG 单调性 C(J0) 如图 5 (a) 所示。 J0=0.82±0.02 处观察到 RG 单调性的局部最大值。 由于 RG 流旨在遵循 RG 单调性的梯度下降轨迹,如方程式 (13 ) 所示,这意味着参数 J0 将从不稳定的固定点 J0 出发,并流向两个稳定固定点之一,即 J0=0J0 RG 流也可以从图 5 (b) 中 J0C 的图中理解,该图对应于 RG 理论背景下的 β 函数。

Refer to caption
图 5: (a) RG 单调性 C(J0) 和 (b) 它的负梯度 J0C(J0) (β 函数),作为 𝖠1 中唯一耦合参数 J0 的函数。 阴影区域表示基于十个独立训练的 MLRG 模型估计的两个标准差的不确定性范围。 垂直实线标记估计的临界点 J0,而精确的临界点 J0c 显示为垂直虚线。 箭头表示 RG 流方向。

RG 单调 C(J0) 的最大位置 J0:=argmaxJ0C(J0) 提供了分离顺磁相 (J0=0) 和铁磁相 (J0)。 Lieb 格子模型的精确伊辛临界点位于

J0c=12arccosh(1+2)0.7643. (19)

估计值 J0=0.82±0.02 接近但仍偏离精确值 J0c 这是因为 𝖠1 表示的维度很小,这限制了学生 RBM 在 MLRG 算法中逼近教师 RBM 的能力。 我们应该期望当现场自旋涉及点群的更大表示时,估计值会得到改进。

Refer to caption
图 6: 具有 𝖠1𝖤 表示的模型的 RG 流图。 J0 参数化 𝖠1 自旋之间的伊辛耦合。 J1,2, 参数化等式 ( 17 ) 中剩余的耦合,使其处于最相关的方向。 背景颜色(和轮廓)表示 RG 单调函数 C(J) 箭头描绘了 RG 流方向。 稳定(不稳定)不动点用黑点(红点)标记。

超越 𝖠1 表示,RG 流将在高维参数空间中定义。 为了可视化 RG 单调函数 C(J),我们将参数 J0 与耦合矩阵 J 中的剩余参数 J1,2, 分开。 我们始终通过将 J0 设置为格子模型的裸伊辛耦合并允许 J1,2, 在 RG 流下生成来初始化 RG 流。 在由 J0J1,2, 的最相关流方向(表示 J1,J2, 参数的特定线性组合,其中 RG 单调函数的梯度最大)所跨越的子流形上,我们可以绘制由 MLRG 方法获得的 RG 单调函数 C(J),以及其梯度方向(RG 流方向)。 一个例子如图 6 所示,该图是通过使用 𝖠1𝖤 现场表示进行训练得到的。

MLRG 算法在整个参数空间中学习 RG 单调 C(J),在此基础上可以获得 RG 流图。 当我们沿着水平轴将裸耦合 J0 调到临界点 J0 时,RG 流将我们带到远离水平轴的真实 RG 固定点 J 参数 J 定义了一个统计力学模型,该模型近似于 Ising 共形场理论 (CFT)。 预计随着现场点群表示的增大,近似效果会更好。

III.3 确定临界点

为了估计 Ising 临界点 J0,我们首先用给定的 J0J1,2,=0 初始化耦合矩阵 J,记为

J(J0,=0)=[J0000]. (20)

然后,我们通过从 =0 到某个较大的 求解 RG 方程 dJ/d=JC(J) 来使耦合矩阵 J 流动。 我们将该解记为 J(J0,),因为它取决于初始条件 J0 和 RG 尺度 在我们的 RG 方案中,线性系统大小将在 RG 尺度 上有效地放大 2/2。 图 7 显示了 RG 单调 C(J(J0,)) 作为不同 RG 尺度 J0 不同初始值的函数。

Refer to caption
图 7: RG 单调 C(J(J0,)) 的流动,从给定的 J0 参数开始,在表示选择 (a) 𝖠1、(b) 𝖠1𝖤 和 (c) 2𝖠1𝖤 下。 表示 RG 流动的步骤。 垂直实线表示估计的临界点 J0,而精确的临界点 J0c 则显示为垂直虚线。

可以看到,RG 单调在临界点处达到峰值,并且随着 RG 流动 的延长,峰值变得更尖锐。 对于足够大的 ,我们可以通过找到 RG 单调的局部最大值来估计临界点 J0

J0=argmaxJ0C(J(J0,)), (21)

只要 RG 规模足够大以致 RG 流收敛,结果 J0 对 RG 规模 不敏感。 我们为每个现场点群表示选择训练了几个不同的MLRG模型,并使用上述方法估计了伊辛临界点 J0 我们的结果如图 8 所示。 我们可以看到一个明显的趋势,即估计的临界点 J0c 更大的点群表示(因此更强的RBM模型)而收敛到其精确值。

Refer to caption
图8: 在不同的现场点群表示选择下估计的临界点 J0 虚线表示精确值 J0c=12arccosh(1+2)

III.4 玻尔兹曼权重张量

为了进一步分析不同RG不动点的物理性质,我们根据耦合参数 J 从RBM能量模型中定义了玻尔兹曼权重张量 T(J) 张量元素被指定为

T(J)x0,x1,x2,x3:=zeEJ(x,z), (22)

其中 x=(x0,x1,x2,x3) 可见自旋共同,z 隐藏自旋共同。 T(J) 一个秩4张量,每个张量元素编码可见自旋 x 特定配置的玻尔兹曼权重,如图 9 所示。 我们可以使用教师RBM(如公式 5)或学生RBM(如公式 6)作为能量模型 EJ(x,z) 只要模型参数 J 收敛到一个RG不动点,教师和学生在理想极限情况下应该表现相同,它们应该没有太大差异。 然而,在以下分析中,我们将始终采用教师模型,因为我们发现它比学生模型更准确一些。

Refer to caption
图9: RBM 能量模型的张量表示,其中每个张量元素编码可见自旋配置的玻尔兹曼权重。

玻尔兹曼权重张量 T(J) 使我们能够方便地定义统计力学模型的各种物理性质。 例如,基态简并度 Z(J)(正则化配分函数)[34, 35] 可以定义为以下张量收缩的比率(重复索引自动求和)

Z(J)=(T(J)abab)2T(J)acbcT(J)bdad=([Uncaptioned image])2[Uncaptioned image]. (23)

对于 Ising 模型,基态简并度表征了破缺对称群的阶数:Z=|1|=1 在顺磁(无序)不动点,其中内部 2 对称性保持,以及 Z=|2|=2 在铁磁(有序)不动点,其中内部 2 对称性被破坏。

Refer to caption
图 10: 基态简并度 Z 的流动,从给定的 J0 参数开始,在 (a) 𝖠1、(b) 𝖠1𝖤 和 (c) 2𝖠1𝖤 的表示选择下。 标记 RG 流动的步骤。 垂直实线标记估计的临界点 J0,而精确的临界点 J0c 显示为垂直虚线。

遵循 Sec. III.3 中描述的方法,我们从用单个参数 J0 初始化耦合矩阵 J 开始,如等式 ( 20 ) 中,并遵循 RG 流获得 J(J0,) 在不同的 RG 尺度 ,我们绘制基态简并度 Z(J(J0,)) 作为初始参数 J0 的函数,结果显示在图 10 中。 J0 穿过临界值 J0 时,基态简并度从 Z=1(顺磁体)过渡到 Z=2(铁磁体)。 沿 RG 流动,过渡随着 的增加变得更尖锐。 可以使用不同 RG 尺度下基态简并度曲线的交点来推断 Ising 临界点 J0 当我们扩大点群表示时,J0 逐渐接近图 8 中所示的 J0c 处的精确 Ising 临界点。

MLRG 可以使用此分析自动通过其基态简并度标记不同的对称性破缺相。 这可以被视为一种无监督相分类方法。 基态简并曲线 Z(J(J0,)) 也提供了一种通过寻找其交点来估计临界点 J0 的方法。

III.5 定位单调鞍点

我们的目标是估计临界点的通用性质。 这要求我们在高维参数空间中准确地定位 RG 固定点参数 J,如 Fig. 6 所示。

但是,如果我们只是沿着 RG 流动来逼近 J,我们几乎不可避免地会错过它。 这是由于 RG 流动有偏离不稳定固定点的趋势。 幸运的是,MLRG 算法已被训练来学习 RG 单调函数 C(J) 此功能使我们能够通过牛顿法定位 RG 单调的鞍点来找到 RG 固定点。 通过这样做,我们可以有效地规避参数空间中微调参数的复杂性。

给定 RG 单调函数 C(J) 并从 RG 固定点附近的初始猜测 J 开始,牛顿法通过以下更新递归地改进近似值

JJ(J2C)1JC, (24)

其中 JC 表示梯度向量,J2C 表示 C(J)J 处的 Hessian 矩阵。 Fig. 11 展示了 J 沿着 (J2C)1JC 的向量场流动。 与 Fig. 6 中的 RG 流动不同,牛顿法从各个方向收敛到 RG 固定点的邻域,无论 RG 固定点是稳定还是不稳定。 这使我们能够从附近的初始猜测中精确定位 RG 固定点。

Refer to caption
图 11: 牛顿法迭代公式(24)对具有 𝖠1𝖤 表示的模型的局部 RG 固定点的流程图。 稳定和不稳定的 RG 固定点在牛顿流下都是吸引的。 虚线标记了不同吸引盆之间的边界。

为了在参数空间中定位 Ising 临界不动点 J,可以设置矩阵 JJ0 分量为其临界值 J0 来构造初始猜测,该临界值可以通过 Sec. II.4 或 Sec. III.4 中描述的方法进行估计。 可以通过评估 C(J) 来监控收敛性。 当连续步骤之间 C(J) 没有明显变化时,我们停止迭代。 预计迭代将收敛到 RG 单调的鞍点 J,其中 JC=0

III.6 计算标度维数

一旦我们通过 Eq. (24) 中的迭代找到 RG 不动点 J,就可以将其代入 Eq. (22) 来构建不动点玻尔兹曼权重张量 T(J) 然后可以使用该张量来计算传递矩阵 M,其矩阵元素如下:

Mab:=T(J)acbc=[Uncaptioned image]. (25)

传递矩阵 M 的特征值与 CFT [34, 35, 36, 13, 37] 中算子的标度维数 Δk (k=0,1,) 相关,

TrMnke2πn(Δkc/12), (26)

其中 c 是中心电荷。 比例因子可以通过要求恒等算子(即真空态)具有零标度维数 Δ0=0 来确定。 这使得我们可以提取与最相关算子相关的最低几个标度维数。

Refer to caption
图 12: 在不同的现场点群表示选择下估计的标度维数 Δ1 虚线表示精确值 Δ1=1/8

12 显示了使用不同点群表示选择进行 Ising MLRG 对最低标度维数 Δ1 的估计。 这是 2 奇数伊辛序参数的缩放维度,其确切值为 Δ1=1/8 我们可以看到,MLRG 可以通过更大的点群表示来逼近精确值。

IV 摘要和讨论

在这项研究中,我们提出了 MLRG 算法,并用它来分析统计物理学中的多体晶格模型。 我们的方法结合了多种现代机器学习技术,包括使用等变神经网络 [21, 22, 23, 24, 25, 26] 用于 RBM 和神经常微分方程 [32, 33] 用于模拟 RG 流。 源代码和原始数据可在 MLRG GitHub 仓库中获取 [38]

MLRG 算法展示了机器学习在自动化和增强统计物理系统研究方面的强大功能。 我们利用生成式建模提供的表示学习能力,直接从数据中学习最佳的 RG 变换,而无需直接的人为干预或关于系统先验知识(除了对称性和维度)。

MLRG 算法的设计体现了 内省学习 [39] 的范式,这是机器学习中用于科学发现的一种有效方法。 它指的是算法在学习过程中能够自我分析和自我适应的能力,利用多个层次的学习和分析来生成对当前问题的更全面、更深入的理解。 MLRG 的设计通过其多级、多模型架构体现了这种内省性质。 它使用两个“低级”机器,教师和学生模型,来模拟通过表示学习提取相关特征的重整化群过程。 它还包含一个“高级”机器,调节器模型,它学习 RG 流并利用其知识来指导对最值得训练的新模型参数进行采样。 这种多级、多模型设计允许在 任务执行(由教师和学生模型执行)和 知识提取(由调节器模型执行)之间进行清晰的区分。 这种分离赋予了高级模型错误校正能力,使其能够抵抗 RBM 训练中噪声对学习 RG 流的影响。 这种压缩知识和纠正错误的能力对于 AI 进行科学发现至关重要 [40]

MLRG 方法与现有工作(并与之区别)在以下方面相关:

  • 蒙特卡洛重整化群 (MCRG): MCRG [41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54] 使用蒙特卡罗采样来促进非微扰 RG 技术。 此过程从给定模型参数的细粒度模型中采样配置开始。 然后,它采用局部 RG 变换来生成粗粒度配置,并相应地估计新模型参数。 此方法与我们从教师 RBM 训练学生 RBM 的方法一致。 特别是,最近的进展 [4, 7, 8] 引入了生成模型来学习最佳 RG 变换,这是一个与我们目标一致的目标。

    然而,我们的方法利用特定晶格结构,如 Lieb 晶格,将 RG 操作限制在小的自旋簇内。 在缺乏这种设计的情况下,MCRG 必须在更大的晶格上执行采样,并对许多长程多自旋耦合进行建模,这会增加计算复杂性。 此外,MCRG 不会学习 RG 单调性,也不能自动找到 RG 固定点,因此在研究临界点性质时需要微调模型参数,从而增加了计算负担。

  • 深度学习与 RG: 多项研究 [55, 56, 57, 58, 59] 在深度神经网络和 RG 之间发现了平行关系。 值得注意的是,他们认识到生成是重整化的逆过程 [6, 11] 因此,生成模型可用于实现数据驱动的 RG。 这些讨论侧重于使用深度学习优化 RG 变换,这通常依赖于蒙特卡罗模拟的自旋配置进行分层特征提取。 尽管它们在提取稳定相的特征方面非常有效,但它们在预测相变的普适性质方面缺乏可控精度,因为它们没有学习 RG 方程或 RG 单调性。

  • 基于 RG 流的生成模型: 神经 RG [2, 6] 和 RG 流 [10, 11] 等技术将 RG 变换嵌入到多级基于流的生成模型 [60, 61, 62] 中,通过最小化自由能应用深度学习方法来学习模型哈密顿量中的最佳 RG 变换。 这些方法基于 可逆 RG 框架,该框架将局部 RG 变换设计为从自旋配置到相关和不相关特征的双射(可逆)确定性映射。

    然而,双射的要求限制了 RG 变换的可能性。 此外,基于流的模型难以对离散变量概率分布进行建模,限制了它们在各种统计力学问题中的应用。 它们也缺乏渐近精确极限,这意味着它们通常只能用作配置更新提议者来加速蒙特卡罗计算,而不是取代蒙特卡罗作为无偏模拟方法。

    相反,MLRG 适用于离散变量和连续变量,提供更灵活的 RG 变换,具有随机粗粒度映射,允许扩展现场自由度,并且在现场自由度趋于无穷大时是精确的。 这些特性使 MLRG 成为研究统计物理学的宝贵方法。

  • 信息论方法 RG:一些研究探索了最佳 RG 的信息论标准,表明 RG 变换应该最大化相关特征与环境的互信息 [1, 5, 63, 64, 65, 66] 或最小化无关特征之间的互信息 [6] 虽然 MLRG 不与这些原则冲突,但它没有明确地将它们用作优化标准。 相反,它使用教师模型和学生模型在其共同自旋上的边际分布的匹配来定义最佳 RG 变换,这种方法更直接且更容易优化。

作为一种解决统计力学问题的数值算法,MLRG 展示了几个优势:

  • 小簇抽样效率:与传统的蒙特卡罗 (MC) 模拟相比,MLRG 算法在更小、更轻量级的规模上运行。 它只需要对一小簇自旋(在晶胞内)内的自旋配置进行采样。 此外,不同自旋簇中的吉布斯抽样可以在现代计算设备上有效地并行化。 这种紧凑的操作允许该算法比更大规模的模拟更有效地执行。

  • 探索完整的参数空间:MLRG 算法旨在在一个训练过程中有效地遍历整个参数空间。 这与 MC 模拟和张量网络重整化群 (TNRG) [67, 68, 34, 69, 70, 35] 存在很大差异,后者必须在不同参数下多次求解哈密顿量。 MLRG 算法能够探索完整的参数空间,降低了计算成本和扫描相图所花费的时间,这在参数空间维数很大时可能是有益的。

  • 关键点的发现和分析:MLRG 利用关于 RG 单调性的知识来识别关键点(即不稳定的 RG 固定点)。 它使用机器学习的 RG 流来计算固定点属性,从而消除了对有限尺寸缩放的需要。 这种自动发现和分析过程增强了算法分析复杂物理系统中关键属性的能力,尤其是在具有多个不稳定方向的多临界点的情况下。

  • 算法的受控收敛:与张量网络方法类似,MLRG 算法对物理属性的估计随着潜在空间维度的增加而收敛。 这种行为确保随着更多计算资源的投入,算法的预测变得越来越准确和可靠。

  • 可解释性:与其他用于 RG 的机器学习方法相比,MLRG 方法通过使用对称表示对模型中的隐藏自旋进行标记,从而提供更好的可解释性,从而得出物理上有意义的耦合参数和 RG 规则。 在物理科学中,这一特性尤其宝贵,因为目标不仅是准确预测,还要获得物理见解。

目前用于对 MLRG 中局部能量进行建模的 RBM 训练是一个随机过程。 这会导致波动和噪声,从而导致 RG 单调性网络不稳定。 RG 单调性网络的稳定性直接影响关键属性预测的可靠性。 提高这种稳定性的一个潜在方法是用张量网络替换 RBM,如 Fig. 9 所示。 这将允许基于张量网络优化的确定性训练方法,从而减少噪声并提高 RG 单调性网络的稳定性。 随着稳定性和可靠性的提高,MLRG 可以扩展到处理更复杂的自旋模型。 这些模型涉及更大的内部对称群,例如 Sn 群,这对研究分类对称 [71, 72] 和纠缠跃迁 [73, 74, 75, 76, 77, 78] 至关重要。

致谢。
我们感谢 Rokas Veitas、John McGreevy、Roger Melko、Han Ma 和 Lei Wang 的讨论。 WH 和 YZY 由加州大学圣地亚哥分校的创业基金和美国国家科学基金会 (NSF) 资助,资助编号为: DMR-2238360。 该研究最初于 2021 年夏季在 Swarma Club 的因果涌现阅读小组中发表,并受益于活动期间与观众的互动。 我们感谢 OpenAI GPT4 模型在撰写本文的过程中提供了编辑建议。

参考文献