Towards provably efficient quantum algorithms for large-scale machine-learning models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Liu, Junyu, Liu, Minzhao, Liu, Jin-Peng, Ye, Ziyu, Wang, Yunfei, Alexeev, Yuri, Eisert, Jens, Jiang, Liang
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911755479810048
author Liu, Junyu
Liu, Minzhao
Liu, Jin-Peng
Ye, Ziyu
Wang, Yunfei
Alexeev, Yuri
Eisert, Jens
Jiang, Liang
author_facet Liu, Junyu
Liu, Minzhao
Liu, Jin-Peng
Ye, Ziyu
Wang, Yunfei
Alexeev, Yuri
Eisert, Jens
Jiang, Liang
contents Large machine learning models are revolutionary technologies of artificial intelligence whose bottlenecks include huge computational expenses, power, and time used both in the pre-training and fine-tuning process. In this work, we show that fault-tolerant quantum computing could possibly provide provably efficient resolutions for generic (stochastic) gradient descent algorithms, scaling as O(T^2 polylog(n)), where n is the size of the models and T is the number of iterations in the training, as long as the models are both sufficiently dissipative and sparse, with small learning rates. Based on earlier efficient quantum algorithms for dissipative differential equations, we find and prove that similar algorithms work for (stochastic) gradient descent, the primary algorithm for machine learning. In practice, we benchmark instances of large machine learning models from 7 million to 103 million parameters. We find that, in the context of sparse training, a quantum enhancement is possible at the early stage of learning after model pruning, motivating a sparse parameter download and re-upload scheme. Our work shows solidly that fault-tolerant quantum algorithms could potentially contribute to most state-of-the-art, large-scale machine-learning problems.
format Preprint
id arxiv_https___arxiv_org_abs_2303_03428
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Towards provably efficient quantum algorithms for large-scale machine-learning models
Liu, Junyu
Liu, Minzhao
Liu, Jin-Peng
Ye, Ziyu
Wang, Yunfei
Alexeev, Yuri
Eisert, Jens
Jiang, Liang
Quantum Physics
Machine Learning
Large machine learning models are revolutionary technologies of artificial intelligence whose bottlenecks include huge computational expenses, power, and time used both in the pre-training and fine-tuning process. In this work, we show that fault-tolerant quantum computing could possibly provide provably efficient resolutions for generic (stochastic) gradient descent algorithms, scaling as O(T^2 polylog(n)), where n is the size of the models and T is the number of iterations in the training, as long as the models are both sufficiently dissipative and sparse, with small learning rates. Based on earlier efficient quantum algorithms for dissipative differential equations, we find and prove that similar algorithms work for (stochastic) gradient descent, the primary algorithm for machine learning. In practice, we benchmark instances of large machine learning models from 7 million to 103 million parameters. We find that, in the context of sparse training, a quantum enhancement is possible at the early stage of learning after model pruning, motivating a sparse parameter download and re-upload scheme. Our work shows solidly that fault-tolerant quantum algorithms could potentially contribute to most state-of-the-art, large-scale machine-learning problems.
title Towards provably efficient quantum algorithms for large-scale machine-learning models
topic Quantum Physics
Machine Learning
url https://arxiv.org/abs/2303.03428