[논문 리뷰] NCON: A tensor network contractor for MATLAB

Robert N. C. Pfeifer, Glen Evenbly외 2명|arXiv (Cornell University)|2014. 02. 05.

Computational Physics and Python Applications인용 수 25

한 줄 요약

이 논문은 다중 텐서 네트워크의 수축을 최적화된 이원 수축을 통해 permute 및 reshape 연산을 활용하여 효율적으로 계산하는 MATLAB 기반 텐서 네트워크 수축기 ncon()을 소개한다. 이는 이전의 레거시 텐서 네트워크 도구들에 비해 현대적이고 잘 문서화되고 유지보수 가능한 대안을 제공하며, 추적, 외적, 분리된 네트워크와 같은 기능을 지원한다. 주로 양자 다체계 시뮬레이션에서 성능과 정확성을 중시한다.

ABSTRACT

A fundamental process in the implementation of any numerical tensor network algorithm is that of contracting a tensor network. In this process, a network made up of multiple tensors connected by summed indices is reduced to a single tensor or a number by evaluating the index sums. This article presents a MATLAB function ncon(), or "Network CONtractor", which accepts as its input a tensor network and a contraction sequence describing how this network may be reduced to a single tensor or number. As its output it returns that single tensor or number. The function ncon() may be obtained by downloading the source of this preprint.

연구 동기 및 목표

• MATLAB에서 scon()과 같은 레거시 텐서 네트워크 수축 도구를 대체할 현대적이고 유지보수 가능하며 잘 문서화된 대안을 제공하기 위해.
• 양자 다체계 물리학에서 사용되는 텐서 네트워크의 효율적이고 정확한 수축을 지원하기 위해, 추적, 외적, 분리된 네트워크를 포함한 기능을 제공하기 위해.
• 이전 도구의 문법을 유지하면서도 새로운 기능을 추가한 견고하고 후행 호환성 있는 인터페이스를 구현하기 위해.
• permute 및 reshape 연산을 통해 BLAS 가속 행렬 곱셈을 활용하여 성능과 신뢰성을 향상시키기 위해.
• 모든 공유 인덱스를 한 번에 수축하지 않는 비최적의 수축 순서를 탐지하고 경고하기 위해.

제안 방법

• ncon() 함수는 MATLAB의 permute 및 reshape 기능을 사용하여 인덱스를 재구성한 후 행렬 곱셈을 수행하는 방식으로 이원 수축을 반복적으로 적용함으로써 텐서 수축을 수행한다.
• 이원 수축은 reshape 및 permute를 통해 구현되며, 고성능을 위해 BLAS를 활용한다.
• 레그 레이블링 표기법을 지원하여 어떤 인덱스가 수축되는지 지정할 수 있으며, legLinks의 음수 인덱스가 출력 인덱스 순서를 정의한다.
• 추적은 반복 인덱스를 합산하는 데 사용되는 전용 doTrace() 함수를 통해 계산되며, 추적이 완전히 수축되지 않은 경우 경고를 출력한다.
• 0-인-시퀀스 표기법은 zisOuterProduct()를 통해 파싱되어 외적과 단순(차원 1) 인덱스를 처리한다.
• 수축 순서는 단계별로 처리되며, 각 연산 후 인덱스가 제거되고, 최종 출력은 지정된 finalOrder 또는 legLinks에 따라 재정렬된다.

실험 결과

연구 질문

• RQ1 MATLAB에서 표준 수축과 더불어 추적, 외적과 같은 고급 기능을 지원하는 텐서 네트워크 수축 기능을 어떻게 구현할 수 있는가?
• RQ2 기본 함수와 BLAS 최적화된 연산을 활용하여 MATLAB에서 이원 텐서 수축을 가장 효율적으로 수행하는 방법은 무엇인가?
• RQ3 레거시의, 문서화되지 않은, 분기된 유사 도구들에 비해 현대적이고 유지보수 가능하며 잘 문서화된 텐서 네트워크 수축기를 어떻게 설계할 수 있는가?
• RQ4 도구가 비최적의 수축 순서를 탐지하고, 불필요한 계산 오버헤드를 유발하는 경우에 경고를 출력할 수 있는가?
• RQ5 모든 공유 인덱스를 한 번에 수축하는 것과 일부 공유 인덱스만 수축하는 것의 영향은 무엇이며, 이러한 비효율성은 어떻게 식별할 수 있는가?

주요 결과

• ncon() 함수는 permute 및 reshape를 통한 최적화된 행렬 곱셈을 활용하여 고성능이며 후행 호환성 있는 MATLAB 기반 텐서 네트워크 수축기를 성공적으로 구현하였다.
• 이전 도구에서 부족하거나 잘 지원되지 않았던 고급 기능인 추적, 외적, 분리된 네트워크, 1차원/0차원 객체를 모두 지원한다.
• 모든 공유 인덱스를 한 번에 수축하지 않는 수축 순서가 발견되면 경고를 출력하여 잠재적인 비효율성을 알린다.
• 구현은 완전히 문서화되어 있으며 중심적으로 유지보수되며, 이전 scon() 버전에서 부족했던 기여자 정보와 문서화 문제를 해결한다.
• BLAS 가속 행렬 곱셈을 사용함으로써 고성능을 확보하였으며, 양자 다체계 물리학의 다양한 텐서 네트워크 알고리즘에 대해 강력한 성능을 발휘한다.
• 반복 인덱스를 포함한 복잡한 인덱스 패턴과 외적을 위한 0-인-시퀀스 표기법을 정확히 처리한다.