[论文解读] Hidden Unit Interpretability in RBM Quantum States:Encoding Antiferromagnetic Order in Heisenberg Spin Rings

Bharadwaj Chowdary Mummaneni, Manas Sajjan|arXiv (Cornell University)|Mar 25, 2026

Quantum many-body systems被引用 0

一句话总结

该论文分析受限玻尔茨曼机（RBM）如何在爱因斯坦-海森堡自旋环中编码反铁磁序，展示隐藏单元集体专门化以捕捉错位磁化并考察随系统尺寸的扩展性。

ABSTRACT

We investigate how Restricted Boltzmann Machines (RBMs) encode antiferromagnetic order when trained as variational ansätze for one-dimensional Heisenberg spin rings with periodic boundary conditions. Through systematic hidden unit analysis and ablation studies on $N=4$ and $N=8$ spin systems, we show that individual hidden units spontaneously specialize to capture staggered magnetization patterns characteristic of antiferromagnetic ground states. Hidden units naturally segregate into two classes: those essential for ground-state energy and correlation structure, and supplementary units providing smaller corrections. Removing important units induces clear energy penalties and disrupts the staggered correlation pattern in $C_{zz}(r)$, whereas removing supplementary units has modest effects. Single-unit analysis confirms that no individual hidden unit reproduces the full antiferromagnetic correlations, indicating that quantum order emerges through collective encoding across the hidden layer. Extending this analysis to $N=8$ through $20$ with hidden unit densities $α= 2$ to $5$ and ten independent seeds per configuration, we find that the fraction of important hidden units decreases with system size, consistent with sublinear growth $m' \sim N^k$ ($k \approx 0.4$). The energy-correlation impact relationship persists for small to moderate system sizes, though it weakens for the largest systems studied. These results provide a quantitative framework for RBM interpretability in quantum many-body systems.

研究动机与目标

• 通过考察具有周期边界条件的一维海森堡自旋环来促进对神经量子态可解释性的研究。
• 通过消融与权重分析来识别单个隐藏单元如何编码反铁磁序。
• 在系统尺寸N=4到N=20以及隐藏单元密度α变化下，表征重要隐藏单元的作用和尺度缩放。

提出的方法

• 用于周期边界条件的海森堡自旋环的RBM变分下确界。
• 使用随机重配置的变分蒙特卡罗优化以最小化能量。
• 对隐藏单元进行消融研究，以评估对能量和相关性的影响。
• 新的AFM模式相关度量S_AFM^(j)，用于量化每个隐藏单元对错位序的编码。
• 分析C_zz(r)以在训练过程中跟踪反铁磁相关性。

实验结果

研究问题

• RQ1 RBM中的单个隐藏单元如何在爱因斯坦-海森堡自旋环中编码反铁磁序？
• RQ2 隐藏单元是否专门化为对能源和相关结构具有不同功能的群体，且这在尺寸扩大时如何演化？
• RQ3 随着N增加，重要隐藏单元的数量的尺度缩放如何，隐藏单元密度α如何影响可解释性？
• RQ4 反铁磁序是否来自于集体编码而非单一单元的贡献？
• RQ5 将重要隐藏单元与辅助隐藏单元消融对能量和C_zz(r)模式有何影响？

主要发现

• 隐藏单元自发地实现专业化；有些编码带有高S_AFM的错位AFM模式，其他则提供较小的修正。
• 消融表明移除重要单元会导致较大的能量惩罚并破坏AFM C_zz(r)模式，而移除辅助单元影响较小。
• 单一隐藏单元分析表明不存在单一隐藏单元能够再现全部AFM相关性，表明隐藏单元之间存在集体编码。
• 对于N=8到20且α=2–5，重要隐藏单元的比例随系统尺寸减少，遵循次线性尺度m′ ∼ N^k，k约为0.4。
• 提高α可以改善能量收敛，但可解释性在α=3–4附近达到峰值，适度密度达到最佳平衡。
• 能量–相关性影响关系在各尺寸中保持，但在研究的最大系统中减弱。