中国物理B ›› 2024, Vol. 33 ›› Issue (4): 40303-040303.doi: 10.1088/1674-1056/ad1926

• • 上一篇    下一篇

Analysis of learnability of a novel hybrid quantum—classical convolutional neural network in image classification

Tao Cheng(程涛)2, Run-Sheng Zhao(赵润盛)1, Shuang Wang(王爽)2, Rui Wang(王睿)1, and Hong-Yang Ma(马鸿洋)1,†   

  1. 1 School of Sciences, Qingdao University of Technology, Qingdao 266033, China;
    2 School of Information and Control Engineering, Qingdao University of Technology, Qingdao 266033, China
  • 收稿日期:2023-11-03 修回日期:2023-12-19 接受日期:2023-12-28 出版日期:2024-03-19 发布日期:2024-03-27
  • 通讯作者: Hong-Yang Ma E-mail:hongyang_ma@aliyun.com
  • 基金资助:
    Project supported by the Natural Science Foundation of Shandong Province, China (Grant No. ZR2021MF049) and the Joint Fund of Natural Science Foundation of Shandong Province (Grant Nos. ZR2022LLZ012 and ZR2021LLZ001).

Analysis of learnability of a novel hybrid quantum—classical convolutional neural network in image classification

Tao Cheng(程涛)2, Run-Sheng Zhao(赵润盛)1, Shuang Wang(王爽)2, Rui Wang(王睿)1, and Hong-Yang Ma(马鸿洋)1,†   

  1. 1 School of Sciences, Qingdao University of Technology, Qingdao 266033, China;
    2 School of Information and Control Engineering, Qingdao University of Technology, Qingdao 266033, China
  • Received:2023-11-03 Revised:2023-12-19 Accepted:2023-12-28 Online:2024-03-19 Published:2024-03-27
  • Contact: Hong-Yang Ma E-mail:hongyang_ma@aliyun.com
  • Supported by:
    Project supported by the Natural Science Foundation of Shandong Province, China (Grant No. ZR2021MF049) and the Joint Fund of Natural Science Foundation of Shandong Province (Grant Nos. ZR2022LLZ012 and ZR2021LLZ001).

摘要: We design a new hybrid quantum—classical convolutional neural network (HQCCNN) model based on parameter quantum circuits. In this model, we use parameterized quantum circuits (PQCs) to redesign the convolutional layer in classical convolutional neural networks, forming a new quantum convolutional layer to achieve unitary transformation of quantum states, enabling the model to more accurately extract hidden information from images. At the same time, we combine the classical fully connected layer with PQCs to form a new hybrid quantum—classical fully connected layer to further improve the accuracy of classification. Finally, we use the MNIST dataset to test the potential of the HQCCNN. The results indicate that the HQCCNN has good performance in solving classification problems. In binary classification tasks, the classification accuracy of numbers 5 and 7 is as high as 99.71%. In multivariate classification, the accuracy rate also reaches 98.51%. Finally, we compare the performance of the HQCCNN with other models and find that the HQCCNN has better classification performance and convergence speed.

关键词: parameterized quantum circuits, quantum machine learning, hybrid quantum—classical convolutional neural network

Abstract: We design a new hybrid quantum—classical convolutional neural network (HQCCNN) model based on parameter quantum circuits. In this model, we use parameterized quantum circuits (PQCs) to redesign the convolutional layer in classical convolutional neural networks, forming a new quantum convolutional layer to achieve unitary transformation of quantum states, enabling the model to more accurately extract hidden information from images. At the same time, we combine the classical fully connected layer with PQCs to form a new hybrid quantum—classical fully connected layer to further improve the accuracy of classification. Finally, we use the MNIST dataset to test the potential of the HQCCNN. The results indicate that the HQCCNN has good performance in solving classification problems. In binary classification tasks, the classification accuracy of numbers 5 and 7 is as high as 99.71%. In multivariate classification, the accuracy rate also reaches 98.51%. Finally, we compare the performance of the HQCCNN with other models and find that the HQCCNN has better classification performance and convergence speed.

Key words: parameterized quantum circuits, quantum machine learning, hybrid quantum—classical convolutional neural network

中图分类号:  (Quantum algorithms, protocols, and simulations)

  • 03.67.Ac
03.67.Lx (Quantum computation architectures and implementations) 03.67.-a (Quantum information)