当前位置:首页 >> 科研进展 >> 正文

张量网络的微分编程

来源:admin时间:2021-02-02浏览:6次

张量网络方法经过多年的发展已经成为经典统计、量子多体物理等领域重要的理论和计算工具。近年来,张量网络方法也被应用于机器学习问题中。然而,针对张量网络的优化仍是一个长期存在的难点。这个瓶颈阻碍了张量网络方法在一系列复杂问题中的充分应用。

最近中国科学院物理研究所/北京凝聚态物理国家研究中心凝聚态理论与计算重点实验室的廖海军副研究员、刘金国博士、王磊副研究员和向涛研究员合作,将机器学习中发展的"自动微分"技术引入到张量网络优化领域中,从而克服了这一障碍。借助机器学习领域发展的微分编程工具,自动微分可以直接而准确地计算变分能量对于张量参数的梯度。结合基于梯度的张量网络优化方法,可以更准确和有效地获得量子多体基态波函数的张量网络表示。基于这些进展,作者得到了二维正方晶格量子海森堡模型最低的变分结果(如附图所示),并进一步研究了更复杂的量子磁性问题。

这项进展为张量网络方法更广泛地应用于物理和机器学习问题扫清了一个障碍。该研究结果近日发表在物理评论X(Phys. Rev. X)上。论文链接https://journals.aps.org/prx/abstract/10.1103/PhysRevX.9.031041。可在https://github.com/wangleiphy/tensorgrad获得开源的程序实现。上述工作得到了国家重点研发计划(2016YFA0302400、2017YFA0302901),国家自然科学基金委(11888101, 11774398)和中国科学院先导专项(XDB28000000)的支持。

图一 (a)不同方法计算二维自旋1/2反铁磁海森堡模型基态能量的相对误差的比较;(b) 不同方法计算交错磁化强度的比较。