Enhancing multi-class diabetic retinopathy detection using tuned hyper-parameters and modified deep transfer learning

Yeganeh Modaresnia,Farhad Abedinzadeh Torghabeh,Seyyed Abed Hosseini,Hosseini, Seyyed Abed
DOI: https://doi.org/10.1007/s11042-024-18506-3
IF: 2.577
2024-03-09
Multimedia Tools and Applications
Abstract:Diabetes retinopathy (DR) is the primary cause of blindness worldwide. Computer-aided diagnosis methods for early detection of DR fundus images are time and effort-saving and have a low risk of a misdiagnosis compared with a manual diagnosis of DR by clinical experts and ophthalmologists. Recently, transfer learning strategies of pre-trained models have been increasingly employed in medical image detection as a deep learning technique, and they effectively speed up the training phase. However, the effect of image enhancement and hyper-parameter tuning in DR detection was not reported in the literature. Furthermore, conventional transfer learning strategies, while widely employed, have exhibited limitations in achieving high accuracies in this context. Here, we employed the publicly available APTOS dataset for training models. Three different image enhancement techniques, including contrast enhancement, color constancy, and contrast-limited adaptive histogram equalization (CLAHE), are used on DR and its severity detection. Bayesian optimization method was employed for hyper-parameter tuning. Three different pre-trained convolutional neural networks, namely AlexNet, GoogLeNet, and SqueezeNet, are investigated on the binary and multi-classification of DR fundus images. The AlexNet model with tuned training hyper-parameters showed an accuracy of 99.10% for multi-classification of DR. The modified AlexNet + genetic algorithm and tuned hyper-parameters model achieved an impressive accuracy of 99.81% in the same multi-level classification of DR severity. The CLAHE enhancement technique outperforms other techniques in these approaches. The reliable performance of the proposed method would enhance the rapid and robust detection of DR utilizing fundus images to intervene effectively before vision loss occurs.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?