Journal of Computer Applications ›› 2024, Vol. 44 ›› Issue (6): 1905-1910.DOI: 10.11772/j.issn.1001-9081.2023050656

Special Issue: 先进计算

• Advanced computing • Previous Articles     Next Articles

Hybrid classical-quantum classification model based on DenseNet

Feiyu ZHAI, Handa MA()   

  1. School of Computer Science and Communication Engineering,Jiangsu University,Zhenjiang Jiangsu 212013,China
  • Received:2023-05-26 Revised:2023-08-03 Accepted:2023-08-08 Online:2023-08-10 Published:2024-06-10
  • Contact: Handa MA
  • About author:ZHAI Feiyu, born in 1999, M. S. candidate. His research interests include image classification, quantum computing.
  • Supported by:
    Zhenjiang Key R&D Program(GY2023034)

基于DenseNet的经典-量子混合分类模型

翟飞宇, 马汉达()   

  1. 江苏大学 计算机科学与通信工程学院,江苏 镇江 212013
  • 通讯作者: 马汉达
  • 作者简介:翟飞宇(1999—),男,江苏南通人,硕士研究生,主要研究方向:图像分类、量子计算;
  • 基金资助:
    镇江市重点研发计划项目(GY2023034)

Abstract:

Existing image classification models are becoming more and more complex, and the hardware resources and computation time required for computation are increasing. A hybrid Classical-Quantum classification model based on DenseNet (CQDenseNet model) was proposed to address this problem. First, a Variational Quantum Circuit (VQC) that could operate on a Noisy Intermediate-Scale Quantum (NISQ) device was used as a classifier to replace the fully connected layer of DenseNet. Secondly, by using transfer learning, a pre-trained DenseNet model on the ImageNet dataset was utilized as a pre-training model for CQDenseNet. Finally, CQDenseNet model was compared with the benchmark models AlexNet, GoogLeNet, VGG19, ResNet and DenseNet-169 on Chinese Medicine and CIFAR-100 datasets. Experimental results show that CQDenseNet model is more effective than the best-performing benchmark model, with improvements of 2.2 and 7.4 percentage points in accuracy, 2.2 and 7.3 percentage points in precision, 2.2 and 7.1 percentage points in recall, and 2.3 and 6.4 percentage points in F1-score, respectively. It shows that the performance of the hybrid classical-quantum model is better than the classical models.

Key words: DenseNet, hybrid classical-quantum model, image classification, transfer learning, Variational Quantum Circuit (VQC)

摘要:

现有的图像分类模型越来越复杂,计算时所需的硬件资源和计算时间不断增加。针对该问题提出一种基于DenseNet的经典-量子混合分类模型(CQDenseNet模型)。首先,使用一个可在噪声中尺度量子(NISQ)设备上运行的变分量子电路(VQC)作为分类器,替换DenseNet全连接层;其次,使用迁移学习,利用在ImageNet数据集上预先训练好的DenseNet模型作为CQDenseNet的预训练模型;最后,将CQDenseNet模型在中草药分类数据集和CIFAR-100数据集上与基准模型AlexNet、GoogLeNet、VGG19、ResNet和DenseNet-169进行对比。实验结果表明,CQDenseNet模型比所有基准模型中表现最好的基准模型:准确率分别提高了2.2、7.4个百分点,精确率分别提高了2.2、7.3个百分点,召回率分别提高了2.2、7.1个百分点,F1值分别提高了2.3、6.4个百分点,说明了经典-量子混合模型的性能优于经典模型。

关键词: DenseNet, 经典-量子混合模型, 图像分类, 迁移学习, 变分量子电路

CLC Number: