Discriminative kernel convolution network for multi-label ophthalmic disease detection on imbalanced fundus image dataset

Abstract

It is feasible to recognize the presence and seriousness of eye disease by investigating the progressions in retinal biological structures. Fundus examination is a diagnostic procedure to examine the biological structure and anomalies present in the eye. Ophthalmic diseases like glaucoma, diabetic retinopathy, and cataracts are the main cause of visual impairment worldwide. Ocular Disease Intelligent Recognition (ODIR-5K) is a benchmark structured fundus image dataset utilized by researchers for multi-label multi-disease classification of fundus images. This work presents a Discriminative Kernel Convolution Network (DKCNet), which explores discriminative region-wise features without adding extra computational cost. DKCNet is composed of an attention block followed by a Squeeze-and-Excitation (SE) block. The attention block takes features from the backbone network and generates discriminative feature attention maps. The SE block takes the discriminative feature maps and improves channel interdependencies. Better performance of DKCNet is observed with InceptionResnet backbone network for multi-label classification of ODIR-5K fundus images with 96.08 AUC, 94.28 F1-score, and 0.81 kappa score. The proposed method splits the common target label for an eye pair based on the diagnostic keyword. Based on these labels, over-sampling and/or under-sampling are done to resolve the class imbalance. To check the bias of the proposed model towards training data, the model trained on the ODIR dataset is tested on three publicly available benchmark datasets. It is observed that the proposed DKCNet gives good performance on completely unseen fundus images also.

Keywords: Channel shuffle; Discriminative kernel convolution (DKCNet); Fundus image; Multi-label classification; ODIR-5K.