Implementation of Trained Factorization Machine Recommendation System on Quantum Annealer

Chen-Yu Liu

Hsin-Yu Wang

Pei-Yen Liao

Ching-Jui Lai

Min-Hsiu Hsieh

Proceedings of the 2024 International Joint Conference on Neural Networks (IJCNN), pp. 1–8, 2024. DOI: 10.1109/IJCNN60899.2024.10650298

出版日期

March 15, 2024

摘要

Factorization Machine (FM) is the most commonly used model to build a recommendation system since it can incorporate side information to improve performance. However, producing item suggestions for a given user with a trained FM is time-consuming. It requires a run-time of O((NmlogNm)2), where Nm is the number of items in the dataset. To address this problem, we propose a quadratic unconstrained binary optimization (QUBO) scheme to combine with FM and apply quantum annealing (QA) computation. Compared to classical methods, this hybrid algorithm provides a faster than quadratic speedup in finding good user suggestions. We then demonstrate the aforementioned computational advantage on current NISQ hardware by experimenting with a real example on a D-Wave annealer.

研究中心

量子計算研究所

內容目錄