A quantum-inspired classical algorithm for recommendation systems

• Published in: arXiv.org 2019-05
• Main Author: Tang, Ewin
• Format: Article
• Language: English
• Subjects: Algorithms; Machine learning; Recommender systems; Weight
• ISSN: 2331-8422
• DOI: 10.48550/arxiv.1807.04271

We give a classical analogue to Kerenidis and Prakash's quantum recommendation system, previously believed to be one of the strongest candidates for provably exponential speedups in quantum machine learning. Our main result is an algorithm that, given an \(m \times n\) matrix in a data structure supporting certain \(\ell^2\)-norm sampling operations, outputs an \(\ell^2\)-norm sample from a rank-\(k\) approximation of that matrix in time \(O(\text{poly}(k)\log(mn))\), only polynomially slower than the quantum algorithm. As a consequence, Kerenidis and Prakash's algorithm does not in fact give an exponential speedup over classical algorithms. Further, under strong input assumptions, the classical recommendation system resulting from our algorithm produces recommendations exponentially faster than previous classical systems, which run in time linear in \(m\) and \(n\). The main insight of this work is the use of simple routines to manipulate \(\ell^2\)-norm sampling distributions, which play the role of quantum superpositions in the classical setting. This correspondence indicates a potentially fruitful framework for formally comparing quantum machine learning algorithms to classical machine learning algorithms.