arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2204.05693cond-mat.str-elquant-ph
收藏

密度矩阵重整化群与张量处理单元

Density Matrix Renormalization Group with Tensor Processing Units

Martin Ganahl, Jackson Beall, Markus Hauru, Adam G. M. Lewis, Jae Hyeon Yoo, Yijian Zou, Guifre Vidal

AI总结:

本文利用谷歌TPU加速并扩展密度矩阵重整化群(DMRG)方法,在二维晶格模型上达到键维数65536,显著提升计算规模与效率。

摘要:

谷歌的张量处理单元(TPU)是专门为加速和扩展机器学习工作负载而设计的集成电路。它们能够执行快速的分布式矩阵乘法,因此可以被重新用于其他计算密集型任务。在这项工作中,我们展示了使用TPU来加速和扩展密度矩阵重整化群(DMRG)方法,这是一种用于计算局部量子多体哈密顿量基态的强有力数值方法。DMRG的计算成本随系统规模$N$按$O(ND^3)$增长,其中所谓的键维数$D$决定了底层矩阵乘积态(MPS)变分拟设的表达能力。我们考虑了二维空间中的晶格模型,包括大小为$10\times 10$(自由费米子)和$20\times 20$(横场伊辛模型)的方形晶格,对于这些模型,所需的MPS键维数已知至少按$\exp(\sqrt{N})$增长。使用半个TPU v3 pod(即$1,\\!024$个TPU v3核心),我们达到了前所未有的巨大键维数$D = 2^{16} = 65,\\!536$,在此情况下优化单个MPS张量大约需要2分钟。

英文摘要:

Google's Tensor Processing Units (TPUs) are integrated circuits specifically built to accelerate and scale up machine learning workloads. They can perform fast distributed matrix multiplications and therefore be repurposed for other computationally intensive tasks. In this work we demonstrate the use of TPUs for accelerating and scaling up the density matrix renormalization group (DMRG), a powerful numerical approach to compute the ground state of a local quantum many-body Hamiltonian. The cost of DMRG scales with system size $N$ as $O(ND^3)$, where the so-called bond dimension $D$ regulates how expressive the underlying matrix product state (MPS) variational ansatz is. We consider lattice models in two spatial dimensions, with square lattices of size $10\times 10$ (free fermions) and $20\times 20$ (transverse field Ising model), for which the required MPS bond dimension is known to scale at least as $\exp(\sqrt{N})$. Using half of a TPU v3 pod (namely $1,\!024$ TPU v3 cores) we reached an unprecedentedly large bond dimension $D = 2^{16} = 65,\!536$, for which optimizing a single MPS tensor took about 2 minutes.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑