Hyper-optimized tensor network contraction
- Creators
- Gray, Johnnie
- Kourtis, Stefanos
Abstract
Tensor networks represent the state-of-the-art in computational methods across many disciplines, including the classical simulation of quantum many-body systems and quantum circuits. Several applications of current interest give rise to tensor networks with irregular geometries. Finding the best possible contraction path for such networks is a central problem, with an exponential effect on computation time and memory footprint. In this work, we implement new randomized protocols that find very high quality contraction paths for arbitrary and large tensor networks. We test our methods on a variety of benchmarks, including the random quantum circuit instances recently implemented on Google quantum chips. We find that the paths obtained can be very close to optimal, and often many orders or magnitude better than the most established approaches. As different underlying geometries suit different methods, we also introduce a hyper-optimization approach, where both the method applied and its algorithmic parameters are tuned during the path finding. The increase in quality of contraction schemes found has significant practical implications for the simulation of quantum many-body systems and particularly for the benchmarking of new quantum chips. Concretely, we estimate a speed-up of over 10,000× compared to the original expectation for the classical simulation of the Sycamore 'supremacy' circuits.
Additional Information
This Paper is published in Quantum under the Creative Commons Attribution 4.0 International (CC BY 4.0) license. Copyright remains with the original copyright holders such as the authors or their institutions. Published: 2021-03-15. We thank S. Boixo and B. Villalonga for useful feedback on the manuscript. JG acknowledges the Samsung Advanced Institute of Technology Global Research Partnership. SK was supported in part by funding from the Canada First Research Excellence Fund.Attached Files
Published - q-2021-03-15-410.pdf
Files
Name | Size | Download all |
---|---|---|
md5:7003ba1d09e45a1184c94311243ae285
|
4.5 MB | Preview Download |
Additional details
- Eprint ID
- 108992
- Resolver ID
- CaltechAUTHORS:20210506-130409930
- Samsung Advanced Institute of Technology
- Canada First Research Excellence Fund
- Created
-
2021-05-06Created from EPrint's datestamp field
- Updated
-
2021-11-16Created from EPrint's last_modified field