Reducing Memory Requirements of Quantum Optimal Control
Abstract
Quantum optimal control problems are typically solved by gradient-based algorithms such as GRAPE, which suffer from exponential growth in storage with increasing number of qubits and linear growth in memory requirements with increasing number of time steps. These memory requirements are a barrier for simulating large models or long time spans. We have created a nonstandard automatic differentiation technique that can compute gradients needed by GRAPE by exploiting the fact that the inverse of a unitary matrix is its conjugate transpose. Our approach significantly reduces the memory requirements for GRAPE, at the cost of a reasonable amount of recomputation. We present benchmark results based on an implementation in JAX.
Cite
@article{arxiv.2203.12717,
title = {Reducing Memory Requirements of Quantum Optimal Control},
author = {Sri Hari Krishna Narayanan and Thomas Propson and Marcelo Bongarti and Jan Hueckelheim and Paul Hovland},
journal= {arXiv preprint arXiv:2203.12717},
year = {2022}
}
Comments
14 pages, 6 figures, 4 listings, 1 table, accepted for publication in the proceedings of the International Conference on Computational Science (ICCS) 2022