Algorithms for tensor network contraction ordering

Schindler, Frank and Jermyn, Adam S (2020) Algorithms for tensor network contraction ordering. Machine Learning: Science and Technology, 1 (3). 035001. ISSN 2632-2153

[thumbnail of Schindler_2020_Mach._Learn.__Sci._Technol._1_035001.pdf] Text
Schindler_2020_Mach._Learn.__Sci._Technol._1_035001.pdf - Published Version

Download (2MB)

Abstract

Contracting tensor networks is often computationally demanding. Well-designed contraction sequences can dramatically reduce the contraction cost. We explore the performance of simulated annealing and genetic algorithms, two common discrete optimization techniques, to this ordering problem. We benchmark their performance as well as that of the commonly-used greedy search on physically relevant tensor networks. Where computationally feasible, we also compare them with the optimal contraction sequence obtained by an exhaustive search. Furthermore, we present a systematic comparison with state-of-the-art tree decomposition and graph partitioning algorithms in the context of random regular graph tensor networks. We find that the algorithms we consider consistently outperform a greedy search given equal computational resources, with an advantage that scales with tensor network size. We compare the obtained contraction sequences and identify signs of highly non-local optimization, with the more sophisticated algorithms sacrificing run-time early in the contraction for better overall performance.

Item Type: Article
Subjects: Science Global Plos > Multidisciplinary
Depositing User: Unnamed user with email support@science.globalplos.com
Date Deposited: 29 Jun 2023 04:11
Last Modified: 27 Oct 2023 04:59
URI: http://ebooks.manu2sent.com/id/eprint/1263

Actions (login required)

View Item
View Item