Quantum machine learning for image classification

被引:39
作者
Senokosov, Arsenii [1 ]
Sedykh, Alexandr [1 ]
Sagingalieva, Asel [1 ]
Kyriacou, Basil [1 ]
Melnikov, Alexey [1 ]
机构
[1] Terra Quantum AG, Kornhausstr 25, CH-9000 St Gallen, Switzerland
来源
MACHINE LEARNING-SCIENCE AND TECHNOLOGY | 2024年 / 5卷 / 01期
关键词
quantum machine learning; image classification; hybrid quantum neural networks; MNIST; CIFAR-10; MedicalMNIST; Quanvolutional neural network; DEEP;
D O I
10.1088/2632-2153/ad2aef
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Image classification, a pivotal task in multiple industries, faces computational challenges due to the burgeoning volume of visual data. This research addresses these challenges by introducing two quantum machine learning models that leverage the principles of quantum mechanics for effective computations. Our first model, a hybrid quantum neural network with parallel quantum circuits, enables the execution of computations even in the noisy intermediate-scale quantum era, where circuits with a large number of qubits are currently infeasible. This model demonstrated a record-breaking classification accuracy of 99.21% on the full MNIST dataset, surpassing the performance of known quantum-classical models, while having eight times fewer parameters than its classical counterpart. Also, the results of testing this hybrid model on a Medical MNIST (classification accuracy over 99%), and on CIFAR-10 (classification accuracy over 82%), can serve as evidence of the generalizability of the model and highlights the efficiency of quantum layers in distinguishing common features of input data. Our second model introduces a hybrid quantum neural network with a Quanvolutional layer, reducing image resolution via a convolution process. The model matches the performance of its classical counterpart, having four times fewer trainable parameters, and outperforms a classical model with equal weight parameters. These models represent advancements in quantum machine learning research and illuminate the path towards more accurate image classification systems.
引用
收藏
页数:13
相关论文
共 81 条
[1]  
Aaronson S, 2016, Arxiv, DOI [arXiv:1612.05903, 10.48550/ARXIV.1612.05903, DOI 10.48550/ARXIV.1612.05903]
[2]  
Agarap A. F., 2018, arXiv, DOI DOI 10.48550/ARXIV.1803.08375
[3]  
Alam S., 2021, Iccad special session paper: Quantum-classical hybrid machine learning for image classification
[4]  
apolanco3225, 2017, Medical MNIST classification
[5]  
Badrinarayanan V, 2016, Arxiv, DOI [arXiv:1511.00561, DOI 10.1109/TPAMI.2016.2644615]
[6]  
Baek H, 2022, Arxiv, DOI arXiv:2209.12372
[7]   ELEMENTARY GATES FOR QUANTUM COMPUTATION [J].
BARENCO, A ;
BENNETT, CH ;
CLEVE, R ;
DIVINCENZO, DP ;
MARGOLUS, N ;
SHOR, P ;
SLEATOR, T ;
SMOLIN, JA ;
WEINFURTER, H .
PHYSICAL REVIEW A, 1995, 52 (05) :3457-3467
[8]  
Bergholm V, 2022, Arxiv, DOI [arXiv:1811.04968, DOI 10.48550/ARXIV.1811.04968]
[9]   Quantum machine learning [J].
Biamonte, Jacob ;
Wittek, Peter ;
Pancotti, Nicola ;
Rebentrost, Patrick ;
Wiebe, Nathan ;
Lloyd, Seth .
NATURE, 2017, 549 (7671) :195-202
[10]  
BLOCH F, 1946, PHYS REV, V70, P460, DOI 10.1103/PhysRev.70.460