arrow
Return

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

delete2024-01-10
delete14
delete
OA
AI
J
Junyu Liu
M
Minzhao Liu
J
Jin-Peng Liu
Z
Ziyu Ye
Y
Yunfei Wang
Y
Yuri Alexeev
J
Jens Eisert *
L
Liang Jiang
DOI:10.1038/s41467-023-43957-xdelete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
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 x 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.
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Nature Communications cover
Nature Communications
IF:
15.7
Papers:
9.2W
Citations:
91.2W

Organization

F
Free University of Berlin
Scholars:
3.8W
Papers: 3.2W
Citations: 51
U
university of chicago
Scholars:
4.4W
Papers: 3.7W
Citations: 80
B
Brandeis University
Scholars:
4.0K
Papers: 3.8K
Citations: 5.3K
University of California System cover
University of California System
Scholars:
37.5W
Papers: 33.7W
Citations: 6.6K
researcher View more organizations