Title
Semi-supervised medical image classification based on CamMix
Abstract
Collecting a large amount of labeled data is crutial for training deep neural network, which is a limitation for medical image classification because it necessarily involves expert knowledge. To mitigate this problem of insufficient labeled medical data, in this work, we propose a novel semi-supervised framework for medical image classification. For unlabeled data, we apply the consistency-based strategy to produce high-quality pseudo label, which encourages model to output the same predictions under different perturbations. In addition, we present a novel mixed sample data augmentation CamMix to effectively exploit the relation between samples, mixing pairs of input data and labels according to the class activation map mask. We have evaluated our proposed method on two public medical image datasets, interstitial lung disease dataset and ISIC 2018 skin lesion analysis dataset. The results demonstrate superior performance of our method over other existing methods on the two datasets. Meanwhile, our proposed CamMix performs better than the current mixed sample data augmentation methods.
Year
DOI
Venue
2021
10.1109/IJCNN52387.2021.9534222
2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN)
Keywords
DocType
ISSN
Semi-supervised learning, CamMix, Medical image classification
Conference
2161-4393
Citations 
PageRank 
References 
0
0.34
14
Authors
7
Name
Order
Citations
PageRank
Lingchao Guo100.34
Changjian Wang2911.97
Dongsong Zhang301.35
Kele Xu44621.80
Zhen Huang55720.78
Li Luo601.35
Yuxing Peng719445.66