Pseudo-Labeling for Small Lesion Detection on Diabetic Retinopathy Images

TL;DR

This paper presents a semi-supervised CNN-based approach with pseudo-labeling and enhanced feature extraction to improve small lesion detection in diabetic retinopathy fundus images, addressing label scarcity and small object challenges.

Contribution

It introduces an iterative pseudo-labeling training algorithm and extends feature pyramid networks to better detect small lesions in diabetic retinopathy images.

Findings

Significant performance improvement over baseline methods.

Effective discovery of unlabeled lesion instances via pseudo-labeling.

Enhanced detection of small lesions through larger CNN feature maps.

Abstract

Diabetic retinopathy (DR) is a primary cause of blindness in working-age people worldwide. About 3 to 4 million people with diabetes become blind because of DR every year. Diagnosis of DR through color fundus images is a common approach to mitigate such problem. However, DR diagnosis is a difficult and time consuming task, which requires experienced clinicians to identify the presence and significance of many small features on high resolution images. Convolutional Neural Network (CNN) has proved to be a promising approach for automatic biomedical image analysis recently. In this work, we investigate lesion detection on DR fundus images with CNN-based object detection methods. Lesion detection on fundus images faces two unique challenges. The first one is that our dataset is not fully labeled, i.e., only a subset of all lesion instances are marked. Not only will these unlabeled lesion instances not contribute to the training of the model, but also they will be mistakenly counted as false negatives, leading the model move to the opposite direction. The second challenge is that the lesion instances are usually very small, making them difficult to be found by normal object detectors. To address the first challenge, we introduce an iterative training algorithm for the semi-supervised method of pseudo-labeling, in which a considerable number of unlabeled lesion instances can be discovered to boost the performance of the lesion detector. For the small size targets problem, we extend both the input size and the depth of feature pyramid network (FPN) to produce a large CNN feature map, which can preserve the detail of small lesions and thus enhance the effectiveness of the lesion detector. The experimental results show that our proposed methods significantly outperform the baselines.