A multi-class fundus disease classification system based on an adaptive scale discriminator and hybrid loss

Abstract

Fundus images are crucial in the observation and detection of ophthalmic diseases. However, detecting multiple ophthalmic diseases from fundus images using deep learning techniques is a complex and challenging task One challenge is the complexity of fundus disease structures, which leads to low detection accuracy. Another challenge is the class imbalance problem common in multi-label image classification, which increases the difficulty of algorithm training and evaluation. To address these issues, this study leverages deep learning to propose an ophthalmic disease classification system. We first employ ResNet50 as the backbone network to extract image features, and then use our designed multi-dimensional attention module and adaptive scale discriminator to enhance the network's ability to detect disease features. During training, we innovatively propose a hybrid loss function method to improve the detection capability on imbalanced data. Finally, we conducted experiments on the ODRI-5K dataset with the proposed classification system. In the test set, our method achieved an AUC of 98.53 and an F1-score of 89.73. This result fully demonstrates the excellent disease classification capability of our method. In summary, the multi-label fundus image disease classification system we proposed exhibits outstanding recognition capability, providing an effective solution for the diagnosis of multi-label fundus image diseases.