arXiv:2503.21536cs.LGcond-mat.dis-nn2025-03

用倒空间视角揭示RBMs的能级结构与对称性破缺机制

Exploring the Energy Landscape of RBMs: Reciprocal Space Insights into Bosons, Hierarchical Learning and Symmetry Breaking

  • 引入倒空间分析,发现RBMs初始时处于鞍点,曲率由奇异值决定
  • 训练中对称性破缺导致多层抽象特征逐步涌现,符合朗道理论特征
  • 适用于理解生成模型内在学习机制,适合关注理论深度的研究者

深度生成模型因其从复杂分布中学习与采样的能力而广泛应用。尽管各类框架层出不穷,但其相互关系仍不明确,制约了统一人工智能学习理论的发展。本文聚焦受限玻尔兹曼机(RBMs),该模型具备离散分布的通用逼近能力。通过引入倒空间表述,我们揭示了RBMs、扩散过程与耦合玻色子之间的联系。初始化时,RBMs处于鞍点,局部曲率由奇异值决定,其分布遵循马尔琴科-帕斯图尔定律并具有旋转对称性。训练过程中,由于分层学习机制,不同自由度逐步捕获多层次抽象特征,导致能量景观中的对称性破缺,类似朗道理论。这一破缺由奇异值及权矩阵特征向量矩阵表征,并在平均场近似下导出对应自由能。在无限尺寸极限下,倒空间变量呈高斯分布,此时部分模式的扩散过程无法收敛至玻茨曼分布。基于MNIST数据集,我们训练了不同隐层大小的RBMs副本以验证结果。研究弥合了不同生成框架间的鸿沟,深化了对生成模型学习机制的理解。

原文摘要 · Abstract (English)

Deep generative models have become ubiquitous due to their ability to learn and sample from complex distributions. Despite the proliferation of various frameworks, the relationships among these models remain largely unexplored, a gap that hinders the development of a unified theory of AI learning. We address two central challenges: clarifying the connections between different deep generative models and deepening our understanding of their learning mechanisms. We focus on Restricted Boltzmann Machines (RBMs), known for their universal approximation capabilities for discrete distributions. By introducing a reciprocal space formulation, we reveal a connection between RBMs, diffusion processes, and coupled Bosons. We show that at initialization, the RBM operates at a saddle point, where the local curvature is determined by the singular values, whose distribution follows the Marcenko-Pastur law and exhibits rotational symmetry. During training, this rotational symmetry is broken due to hierarchical learning, where different degrees of freedom progressively capture features at multiple levels of abstraction. This leads to a symmetry breaking in the energy landscape, reminiscent of Landau theory. This symmetry breaking in the energy landscape is characterized by the singular values and the weight matrix eigenvector matrix. We derive the corresponding free energy in a mean-field approximation. We show that in the limit of infinite size RBM, the reciprocal variables are Gaussian distributed. Our findings indicate that in this regime, there will be some modes for which the diffusion process will not converge to the Boltzmann distribution. To illustrate our results, we trained replicas of RBMs with different hidden layer sizes using the MNIST dataset. Our findings bridge the gap between disparate generative frameworks and also shed light on the processes underpinning learning in generative models.

生成模型能量景观对称性破缺受限玻尔兹曼机

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。