Computer Engineering and Applications ›› 2023, Vol. 59 ›› Issue (19): 140-150.DOI: 10.3778/j.issn.1002-8331.2205-0418

• Graphics and Image Processing • Previous Articles     Next Articles

Lung Cancer Image Classification Using Cell Morphological Feature Contrastive Learning

JIA Wei, JIANG Haifeng, ZHAO Xuefen   

  1. 1.School of Information Engineering, Ningxia University, Yinchuan 750021, China
    2.Department of Pathology, General Hospital of Ningxia Medical University, Yinchuan 750021, China
    3.Xinhua College, Ningxia University, Yinchuan 750021, China
  • Online:2023-10-01 Published:2023-10-01

采用细胞形态特征对比学习的肺癌图像分类

贾伟,江海峰,赵雪芬   

  1. 1.宁夏大学 信息工程学院,银川 750021
    2.宁夏医科大学总医院 病理科,银川 750021
    3.宁夏大学 新华学院,银川 750021

Abstract: Aiming at the problems that the labeled pathological images of lung cancer are few and the cell morphology is complex in lung cancer image classification, a lung cancer pathological image classification method based on cell morphological feature contrastive learning is proposed. Through contrastive learning, the unlabeled data with high confidence is mixed into the training data to solve the problem of insufficient labeled data. Based on the nearest-neighbor contrastive learning, a farthest and nearest-neighbors-based contrastive learning is proposed. The farthest and nearest neighbor images are used for the contrastive learning at the same time. The performance of the contrastive learning is improved by increasing the learning difficulty of the positive samples and the diversity of the data. Deformable convolution and dynamic convolution-based ResNet50 are used as an encoder to enhance the extraction ability of cell morphological features. The experimental results show that, in the case of less labeled data, compared with the existing classification methods, the proposed method can make full use of the cell feature information in the labeled and unlabeled cancer pathology images and obtain better classification results.

Key words: lung cancer, pathological image classification, contrastive learning, deformable convolution, dynamic convolution

摘要: 针对肺癌图像分类中出现的已标记肺癌病理图像较少且细胞形态复杂的问题,提出一种基于细胞形态特征对比学习的肺癌病理图像分类方法,通过对比学习将置信度较高的未标记数据混入到训练数据中,解决已标记数据不足的问题。在最近邻对比学习的基础上,提出基于最远和最邻近的对比学习,将最远和最近邻图像同时用于对比学习,通过增加对比样本的正样本学习难度和数据的多样性,提升对比学习的性能。将基于可变形卷积和动态卷积的ResNet50作为编码器,增强对细胞形态特征的提取能力。实验结果表明,在已标记数据较少的情况下,与现有的分类方法相比,该分类方法能够充分利用已标记和未标记癌症病理图像中的细胞特征信息,获得较好的分类效果。

关键词: 肺癌, 病理图像分类, 对比学习, 可变形卷积, 动态卷积