论文标题
用于糖尿病性视网膜病变图像的小病变检测的伪标记
Pseudo-Labeling for Small Lesion Detection on Diabetic Retinopathy Images
论文作者
论文摘要
糖尿病性视网膜病(DR)是全球劳动者失明的主要原因。每年大约有3至400万糖尿病患者失明。通过彩色底面图像诊断DR是减轻此类问题的常见方法。但是,DR诊断是一项困难且耗时的任务,它要求经验丰富的临床医生确定许多小型特征在高分辨率图像上的存在和意义。卷积神经网络(CNN)最近被证明是自动生物医学图像分析的一种有希望的方法。在这项工作中,我们使用基于CNN的对象检测方法对DR EDER图像的病变检测进行了研究。眼底图像的病变检测面临两个独特的挑战。第一个是我们的数据集没有完全标记,即仅标记所有病变实例的子集。这些未标记的病变实例不仅会导致模型的训练,而且还会错误地将它们算作虚假负面,从而使模型转向相反的方向。第二个挑战是病变实例通常很小,因此很难被普通对象探测器找到。为了应对第一个挑战,我们引入了一种迭代培训算法,以针对伪标记的半监督方法进行迭代,其中可以发现大量未标记的病变实例,以提高病变探测器的性能。对于小尺寸目标问题,我们同时扩展了特征金字塔网络(FPN)的输入大小和深度,以产生一个大型CNN特征图,该图可以保留小病变的细节,从而提高病变检测器的有效性。实验结果表明,我们提出的方法显着优于基准。
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.