Semi-Supervised Learning for Medical Image Classification Based on Anti-Curriculum Learning

Abstract

Although deep learning has achieved great success in image classification, large amounts of labelled data are needed to make full use of the advantages of deep learning. However, annotating a large number of images is expensive and time-consuming, especially annotating medical images, which requires professional knowledge. Therefore, semi-supervised learning has shown its potential for medical image classification. In this paper, we propose a novel pseudo-labelling semi-supervised learning method for medical image classification. Firstly, we utilize the anti-curriculum strategy for model training to prevent the model from producing predictions with a high value from the samples which are similar with existed labeled data. Secondly, to produce more stable and accurate pseudo labels for unlabeled data, we generate the pseudo labels with ensemble predictions provided by the model with samples augmented by different augmentations. In addition, we refine the generated pseudo labels using the prediction of the model at the current epoch in order to make the model learn from itself and improve the model performance. Comparative experiments on the Chest X-ray14 dataset for a multi-label classification task and the ISIC 2018 dataset for a multi-class classification task are performed, and the experimental results show the effectiveness of our method.