计算机与现代化 ›› 2024, Vol. 0 ›› Issue (02): 81-87.doi: 10.3969/j.issn.1006-2475.2024.02.013

• 人工智能 • 上一篇    下一篇

基于对比学习MocoV2的COVID-19图像分类#br#

  

  1. (1.重庆师范大学计算机与信息科学学院,重庆 401331; 2.西南大学计算机与信息科学学院,重庆 400715)
  • 出版日期:2024-02-19 发布日期:2024-03-19
  • 作者简介: 作者简介:许跃雯(1997—),女,辽宁大连人,硕士研究生,研究方向:人工智能,图像处理,E-mail: 3172830439@qq.com; 通信作者:李明(1966—),男,教授,博士,研究方向:人工智能,网络技术与教育应用,E-mail: 20131052@cqnu.edu.cn; 李莉(1967—),女,教授,博士,研究方向:机器学习,数据挖掘与分析,E-mail: lily@swu.edu.cn。
  • 基金资助:
    国家自然科学基金资助项目(61877051, 61170192); 重庆市科委重点项目(cstc2017zdcy-zdyf0366); 重庆市教委项目(113143); 重庆市研究生教改重点项目(yjg182022)
       

Image Classification of COVID-19 Based on Contrast Learning MocoV2

  1. (1. College of Computer and Information Science, Chongqing Normal University, Chongqing 401331, China;
    2. College of Computer and Information Science, Southwest University, Chongqing 400715, China)
  • Online:2024-02-19 Published:2024-03-19

摘要: 摘要:肺炎是一种常见多发感染性疾病,老年人和免疫力较弱者容易感染,尽早发现有助于后期治疗。肺部病变的位置、密度和清晰度等因素会影响肺炎图像分类的准确性。随着深度学习的发展,卷积神经网络被广泛应用于医学图像分类任务中,然而网络的学习能力依赖训练样本的数量和标签。针对电子计算机断层扫描(Computed Tomography, CT)的肺炎图像分类研究,提出一种基于自监督对比学习的网络模型(MCLSE),可以从无标记的数据中学习特征,提高网络模型的准确率。本文模型(MCLSE)首先设计辅助任务,从无标记的图像中挖掘表征完成预训练,提高模型在向量空间中学习数据映射关系的能力。其次,使用卷积神经网络提取特征,为了有效捕获更高层次的特征信息选择SENet网络改进分类模型,建模特征通道的相关性。最后,用训练好的权重加载改进后的分类模型中,下游任务中使用标记数据再次训练网络。在公开数据集SARS-CoV-2 CT和CT Scans for COVID-19 Classification上进行实验,实验结果表明MCLSE对整体样本分类的准确率分别达到99.19%和99.75%,较主流模型有很大提升。

关键词: 关键词:COVID-19图像, 医学图像分类, 卷积神经网络, 自监督学习, 对比学习

Abstract: Abstract: Pneumonia is a common multi-infectious disease that predisposes the elderly and those with weakened immune systems to infection, and early detection can help with later treatment. Factors such as the location, density and clarity of lung lesions can affect the accuracy of pneumonia image classification. With the development of deep learning, convolutional neural network is widely used in medical image classification tasks, however, the learning ability of the network depends on the number of training samples and labels. Aiming at the classification of pneumonia images in computed tomography (CT), a network model based on self-supervised comparative learning (MCLSE) is proposed, which can learn features from unmarked data and improve the accuracy of the network model. Firsly, auxiliary tasks were designed to mine representations from unmarked images to complete pre-training, improving the ability of the model to learn data mapping relationships in vector space. Secondly, the convolutional neural network is used to extract features. In order to effectively capture higher level feature information, the compression excitation network is selected to improve the classification model and the correlation between the feature channels is modeled. Finally, the trained weights are loaded into the improved classification model, and the network is trained again with marked data in the downstream task. Experiments were carried out on open data sets, SARS-CoV-2 CT and CT Scan for COVID-19 Classification. The results show that the accuracy of the MCLSE model in this paper for the overall sample classification reached 99.19% and 99.75%, respectively, which was greatly improved compared with the mainstream model.

Key words: Key words: COVID-19 image, medical image classification, convolutional neural network, self-supervised learning, contrastive learning

中图分类号: