[논문 리뷰] Adaptive Patching for Tensor Train Computations

Gianluca Grosso, Marc K. Ritter외 6명|arXiv (Cornell University)|2026. 02. 25.

Quantum Computing Algorithms and Architecture인용 수 0

한 줄 요약

본 논문은 양자 텐서 트레인(QTT)의 적응 패칭 스킴을 도입하여 국소 특성을 갖는 텐서를 더 작은 패치로 분할하고 결합 차원과 계산량을 줄여 버블 다이어그램 및 Bethe-Salpeter 방정식과 같은 대규모 QTT 기반 계산의 효율을 가능하게 한다.

ABSTRACT

Quantics Tensor Train (QTT) operations such as matrix product operator contractions are prohibitively expensive for large bond dimensions. We propose an adaptive patching scheme that exploits block-sparse QTT structures to reduce costs through divide-and-conquer, adaptively partitioning tensors into smaller patches with reduced bond dimensions. We demonstrate substantial improvements for sharply localized functions and show efficient computation of bubble diagrams and Bethe-Salpeter equations, opening the door to practical large-scale QTT-based computations previously beyond reach.

연구 동기 및 목표

• 양자 다체 문제에서 큰 결합 차원에 대한 QTT 연산의 높은 비용을 동기 부여하고 해결한다.
• 블록-희소 QTT 구조를 활용하는 적응적 분할-정복 패칭 전략을 개발한다.
• TT/패치된 TT 연산의 효율 증가와 Green’s 함수 및 Bethe-Salpeter 방정식에 대한 실용적 응용을 입증한다.
• 패치 순서 설정, 복잡도 분석 및 overpatching 완화에 대한 프레임워크를 제공한다.

제안 방법

• 개별적으로 더 작은 결합 차원을 가진 패치로 텐서를 분할하여 QTT에 대한 적응형 패칭을 도입한다.
• 패치를 TT 분해로 표현하고 패치를 합산하여 전체 텐서를 재구성한다.
• 패칭을 Tensor Cross Interpolation(TCI)와 결합하여 TT 표현을 효율적으로 구성한다.
• 패치 기반 TT 랭크를 갖는 메모리, 함수 평가, 산술 비용 등의 계산 복잡도를 분석한다.
• 현실적인 2D Green’s 함수 및 Bethe-Salpeter 유형 계산에 패치된 QTCI를 적용한다.
• 성능 최적화를 위한 패치 순서, overpatching, 패칭 패턴에 대해 논의한다.

실험 결과

연구 질문

• RQ1 국소적 특징을 갖는 QTT 표현에서 적응형 패칭이 효과적 결합 차원과 메모리 요구량을 줄일 수 있는가?
• RQ2 패칭이 TT/MPO 수축 및 관련 텐서 연산의 계산 비용에 어떤 영향을 미치는가?
• RQ3 Green’s 함수 및 Bethe-Salpeter 방정식 계산에서 패치된 QTT의 실용적 이득은 무엇인가?
• RQ4 과도한 패칭(overpatching)을 방지하고 속도 향상을 극대화하려면 패치를 어떻게 순서대로 정하고 크기를 어떻게 정해야 하는가?

주요 결과

• 적응형 패칭은 블록-희소 텐서 코어를 활용하여 패치당 결합 차원을 감소시키고 전체 메모리 절감을 달성한다.
• 패치된 TT 표현은 여러 경우에서 표준 QTT 정확도에 맞추면서도 매개변수 수를 줄이고 런타임을 단축시킬 수 있다.
• 두 가지 패치 클래스가 나타난다: 매끄러운 영역용으로 큰 저랭크 패치와 급격한 특징용으로 작은 고랭크 패치로, 타깃 리소스 할당을 가능하게 한다.
• 패치된 MPO–MPO 수축은 분할 정복 방식으로 효율적으로 수행될 수 있어 대규모 QTT 기반 계산의 실현 가능성을 높인다.
• 패치 품질과 순서는 성능에 크게 영향을 미치며, 총 비용을 최소화하는 패치당 최적의 bond-cap이 존재하고 overpatching은 이점을 상쇄할 수 있다.
• 시연된 응용은 현실적인 2D Green’s 함수 및 버블 다이어그램과 Bethe-Salpeter 방정식의 탐색을 포함한다.