Weakly-Supervised Learning With Complementary Heatmap for Retinal Disease Detection.

Abstract

There are many types of retinal disease, and accurately detecting these diseases is crucial for proper diagnosis. Convolutional neural networks (CNNs) typically perform well on detection tasks, and the attention module of CNNs can generate heatmaps as visual explanations of the model. However, the generated heatmap can only detect the most discriminative part, which is problematic because many object regions may exist in the region beside the heatmap in an area known as a complementary heatmap. In this study, we developed a method specifically designed multi-retinal diseases detection from fundus images with the complementary heatmap. The proposed CAM-based method is designed for 2D color images of the retina, rather than MRI images or other forms of data. Moreover, unlike other visual images for disease detection, fundus images of multiple retinal diseases have features such as distinguishable lesion region boundaries, overlapped lesion regions between diseases, and specific pathological structures (e.g. scattered blood spots) that lead to mis-classifications. Based on these considerations, we designed two new loss functions, attention-explore loss and attention-refine loss, to generate accurate heatmaps. We select both "bad" and "good" heatmaps based on the prediction score of ground truth and train them with the two loss functions. When the detection accuracy increases, the classification performance of the model is also improved. Experiments on a dataset consisting of five diseases showed that our approach improved both the detection accuracy and the classification accuracy, and the improved heatmaps were closer to the lesion regions than those of current state-of-the-art methods.