Semi-supervised Medical Image Classification Combining Metric Pseudo-Label and Classification Pseudo-Label

Boya Ke,Huijuan Lu,Wanli Huo,Yanbin Wang
DOI: https://doi.org/10.1109/icceai55464.2022.00166
2022-01-01
Abstract:Deep learning networks with good performance often need a lot of data for training. However, in medical image classification, obtaining high-quality label data is time-consuming and depends on the professional knowledge of doctors. In this paper, a new semi-supervised medical image classification framework based on pseudo-labeling, which combines feature metric pseudo-labeling and model classification pseudo-labeling, and obtains supplementary information in the training process through prior information marked by model from unlabeled data. Considering a lot of information is abandoned in feature map extraction, we introduce an effective pseudo-label combination method, which combines the pseudo-labels obtained by different methods to potential information from unlabeled data. Compared with existing pseudo-labeling methods, our method obtains feature metric pseudo-labels by extracting image features, measuring image feature prototype from labeled data, and combines them with the pseudo-labels obtained by model classification. The performance of this model on skin lesion classification is better than most advanced semi-supervised learning methods.
What problem does this paper attempt to address?