Diabetic Retinopathy Detection Based on Weakly Supervised Object Localization and Knowledge Driven Attribute Mining

Abstract

Disease grading and lesion identification are two important tasks for diabetic retinopathy detection. Disease grading uses image-level annotation but lesion identification often needs the fine-grained annotations, which requires a lot of time and effort of professional doctors. Therefore, it is a great challenge to complete disease grading and lesion identification simultaneously with the limited labeled data. We propose a method based on weakly supervised object localization and knowledge driven attribute mining to conduct disease grading and lesion identification using only image-level annotation. We first propose an Attention-Drop-Highlight Layer (ADHL), which enables the CNN to accurately and comprehensively focus on the various lesion features. Then, we design a search space and employ neural architecture search (NAS) to select the best settings of the ADHL, to maximize the performance of the model. Finally, we regard the lesion attributes corresponding to different disease grades as weakly supervised classification labels representing prior knowledge, and propose an Attribute Mining (AM) method to further improve the effect of disease grading and complete lesion identification. Extensive experiments and a user study have proved that our method can capture more lesion features, improve the performance of disease grading, and obtain state-of-the-art results compared to the methods only using image-level annotation.