[논문 리뷰] Block belief propagation algorithm for two-dimensional tensor networks

Chu Guo, Dario Poletti외 1명|arXiv (Cornell University)|2023. 01. 14.

Quantum many-body systems참고 문헌 58인용 수 6

한 줄 요약

이 논문은 두 차원 텐서 네트워크를 효율적으로 수축하기 위해 블록 신뢰 전파(블록BP) 알고리즘을 제안한다. 이는 유한하고 무한한 2차원 양자 시스템에서 고체 상태를 정확하게 근사할 수 있게 한다. 시스템을 겹치지 않는 블록으로 나누고, 블록 간 메시지를 전달함으로써 블록 내부의 상관관계를 반영하면서도 고도로 병렬화 가능하게 하여, 기존 방법에 비해 훨씬 더 빠른 수렴 속도로 2D 허이젠베르크 및 횡방향 이징 모델에 대해 최신 기술 수준의 정확도를 달성한다.

ABSTRACT

Belief propagation is a well-studied algorithm for approximating local marginals of multivariate probability distribution over complex networks, while tensor network states are powerful tools for quantum and classical many-body problems. Building on a recent connection between the belief propagation algorithm and the problem of tensor network contraction, we propose a block belief propagation algorithm for contracting two-dimensional tensor networks and approximating the ground state of $2D$ systems. The advantages of our method are three-fold: 1) the same algorithm works for both finite and infinite systems; 2) it allows natural and efficient parallelization; 3) given its flexibility it would allow to deal with different unit cells. As applications, we use our algorithm to study the $2D$ Heisenberg and transverse Ising models, and show that the accuracy of the method is on par with state-of-the-art results.

연구 동기 및 목표

• 강한 상관관계를 가진 양자 시스템에 특히 적합한 스케일러블하고 정확한 2차원 텐서 네트워크 수축 방법을 개발하는 것.
• 표준 신뢰 전파가 순환 네트워크에서 가지는 한계를 극복하기 위해 일반화된 메시지 전달 프레임워크를 통해 블록 수준의 상관관계를 통합하는 것.
• 유한 시스템과 무한 시스템을 동일한 알고리즘 프레임워크 내에서 통합적으로 다루어 별도의 구현을 피하는 것.
• 텐서 네트워크 수축의 효율적 병렬화를 가능하게 하여 계산 시간을 단축하면서도 높은 정확도를 유지하는 것.
• 다양한 단위 세포와 격자 기하 구조(양자 화학에서의 비정규 구조 포함)에 적응 가능한 영리한 접근법을 제공하는 것.

제안 방법

• 알고리즘은 2차원 텐서 네트워크를 겹치지 않는 스핀 블록으로 나누며, 각 블록을 메시지 전달 과정에서 단일 실체로 간주한다.
• 각 블록 내부의 상관관계를 정확한 블록 텐서 네트워크 수축을 통해 반영하는 수정된 신뢰 전파 프레임워크를 사용하여 블록 간 메시지를 전달한다.
• 블록 환경은 이웃 블록들로부터 수렴한 메시지의 곱으로 근사하며, 표준 BP의 평균장 근사보다 향상된 성능을 보인다.
• 하이브리드 접근법을 사용한다: 각 블록 내부의 국소 업데이트는 고유 차원 χ를 가진 경계 행렬 제품 상태(bMPS) 기반 방법으로 수행하며, 블록 간 메시지는 별도의 고유 차원 χm을 사용하여 계산한다.
• 에너지 및 관측 가능도 변화를 모니터링하면서, 이미상 시간 진화(ITE)를 반복 적용하여 고체 상태에 수렴한다.
• 경계 조건과 블록 구조를 조정함으로써, 주기적 또는 개방 경계 조건을 가진 유한 시스템과 iPEPS 기반 무한 시스템을 자연스럽게 지원한다.

실험 결과

연구 질문

• RQ1 스핀을 블록으로 묶음으로써 2차원 텐서 네트워크의 순환 구조를 다룰 수 있도록 신뢰 전파를 일반화할 수 있는가?
• RQ2 블록 수준의 메시지 전달이 2차원 양자 시스템에서 평균장 신뢰 전파에 비해 정확도를 크게 향상시키는가?
• RQ3 동일한 알고리즘이 아키텍처 변경 없이도 유한 및 무한 2차원 시스템을 모두 효율적으로 처리할 수 있는가?
• RQ4 기존 텐서 네트워크 방법에 비해 블록BP는 얼마나 높은 수준의 병렬화를 가능하게 하고 실행 시간을 줄이는가?
• RQ5 에너지 및 관측 가능도 정확도 측면에서, 2D 허이젠베르크 및 횡방향 이징 모델과 같은 벤치마크 모델에서 이 방법은 어떤 성능을 보이는가?

주요 결과

• 무한한 횡방향 이징 모델에서, 블록 크기 k=3일 때 블록BP는 에너지 수렴을 다섯 번째 소수 자리까지 달성하며, 표준 bMPS에 비해 수렴 속도가 빠르다.
• 2D 허이젠베르크 모델에서, 블록BP는 최신 기술 수준의 방법과 비교해 유사한 고체 상태 에너지를 달성하며, D=2,3,4의 다양한 고유 차원에서 결과가 안정적이다.
• 무한 시스템에서 bMPS에 비해 빠른 수렴 속도를 보이며, 작은 시스템 크기에서도 높은 정확도를 달성한다.
• PEPS 고유 차원 D에 따라 런타임이 급격히 증가하지만, D≤5 범위에서는 여전히 실행 가능하며, 작은 D에서는 전체 업데이트 알고리즘과 경쟁 가능한 성능을 보인다.
• 메시지 계산에 대해 χm=D²를 기본 설정으로 사용할 경우, 테스트된 모든 모델에서 정확한 결과를 얻었으며, 하이퍼파rameter 조정에 대해 뛰어난 내구성을 보였다.
• 알고리즘은 매우 높은 수준의 병렬화가 가능하며, 스레드 수에 따라 런타임 스케일링이 유리하게 나타나, 특히 큰 시스템에서 유리하다.