Abstract
Diabetic retinopathy, a prevalent eye disease in diabetic patients, poses a high risk of blindness. Current computer-aided diagnostic methods require extensive labeled datasets, which are labor-intensive and time-consuming. Given the scarcity of large labeled datasets in medical imaging, self-supervised learning presents a promising alternative, capable of leveraging unlabeled data to improve diagnostic accuracy. Addressing this challenge, our paper introduces a classification method for diabetic retinopathy using self-supervised learning. The method encompasses three stages: data preprocessing, self-supervised learning, and classification. In preprocessing stage, fundus images are resized to 255 × 255 pixels, divided into nine sub-images each, and labeled with serial numbers. Additionally, masks are applied to sub-image edges to minimize model reliance on edge features. The self-supervised learning stage employs a VGG16-based network with nine branches to learn intrinsic features of fundus images, thus decreasing the requirement for labeled samples. The network inputs are these sub-images, shuffled in order, with output being their sequence numbers. This stage produces a pre-trained network. For the classification stage, this pre-trained network is further fine-tuned using a small labeled dataset (300 images), modifying the final fully connected layer for either binary or five-category classification. Our approach has demonstrated impressive results: 92.5% accuracy for binary and 66.7% for five-category classification on APTOS dataset, and 92.7% for binary and 62.4% for five-category classification on the Kaggle EyePACS dataset. These outcomes underscore the substantial potential of self-supervised learning in diabetic retinopathy diagnosis, offering a significant reduction in the dependency on extensive labeled datasets and thereby enhancing the diagnostic process.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Tsiknakis, N., et al.: Deep learning for diabetic retinopathy detection and classification based on fundus images: a review. Comput. Biol. Med. 135, 104599 (2021). https://doi.org/10.1016/J.COMPBIOMED.2021.104599
Ting, D.S.W., et al.: Development and validation of a deep learning system for diabetic retinopathy and related eye diseases using retinal images from multiethnic populations with diabetes. JAMA 318, 2211–2223 (2017). https://doi.org/10.1001/JAMA.2017.18152
Zhu, J., Li, Y., Hu, Y., Ma, K., Zhou, S.K., Zheng, Y.: Rubik’s Cube+: a self-supervised feature learning framework for 3D medical image analysis. Med. Image Anal. 64, 101746 (2020). https://doi.org/10.1016/J.MEDIA.2020.101746
Chen, L., Bentley, P., Mori, K., Misawa, K., Fujiwara, M., Rueckert, D.: Self-supervised learning for medical image analysis using image context restoration. Med. Image Anal. 58, 101539 (2019). https://doi.org/10.1016/J.MEDIA.2019.101539
Vives-Boix, V., Ruiz-Fernández, D.: Diabetic retinopathy detection through convolutional neural networks with synaptic metaplasticity. Comput. Methods Programs Biomed. 206, 106094 (2021). https://doi.org/10.1016/j.cmpb.2021.106094
Zhou, H.-Y., Lu, C., Yang, S., Han, X., Yu, Y.: Preservational Learning Improves Self-supervised Medical Image Models by Reconstructing Diverse Contexts, pp. 3499–3509 (2021)
Kumar, N.S., Ramaswamy Karthikeyan, B.: Diabetic retinopathy detection using CNN, transformer and MLP based architectures. In: ISPACS 2021 - International Symposium on Intelligent Signal Processing and Communication Systems: 5G Dream to Reality, Proceeding (2021). https://doi.org/10.1109/ISPACS51563.2021.9651024
Lahmar, C., Idri, A.: Deep hybrid architectures for diabetic retinopathy classification. Comput. Methods Biomech. Biomed. Eng. Imaging Vis. 11, 166–184 (2023). https://doi.org/10.1080/21681163.2022.2060864
El-Ateif, S., Idri, A.: Single-modality and joint fusion deep learning for diabetic retinopathy diagnosis. Sci. Afr. 17, e01280 (2022). https://doi.org/10.1016/J.SCIAF.2022.E01280
Chen, W., Yang, B., Li, J., Wang, J.: An approach to detecting diabetic retinopathy based on integrated shallow convolutional neural networks. IEEE Access. 8, 178552–178562 (2020). https://doi.org/10.1109/ACCESS.2020.3027794
Pratt, H., Coenen, F., Broadbent, D.M., Harding, S.P., Zheng, Y.: Convolutional neural networks for diabetic retinopathy. Proc. Comput. Sci. 90, 200–205 (2016). https://doi.org/10.1016/J.PROCS.2016.07.014
Qummar, S., et al.: A deep learning ensemble approach for diabetic retinopathy detection. IEEE Access. 7, 150530–150539 (2019). https://doi.org/10.1109/ACCESS.2019.2947484
Wang, X., Lu, Y., Wang, Y., Chen, W.B.: Diabetic retinopathy stage classification using convolutional neural networks. In: Proceedings - 2018 IEEE 19th International Conference on Information Reuse and Integration for Data Science, IRI 2018, pp. 465–471 (2018). https://doi.org/10.1109/IRI.2018.00074
Khaled, O., El-Sahhar, M., El-Dine, M.A., Talaat, Y., Hassan, Y.M.I., Hamdy, A.: Cascaded architecture for classifying the preliminary stages of diabetic retinopathy. In: ACM International Conference Proceeding Series, pp. 108–112 (2020). https://doi.org/10.1145/3436829.3436854
Kumar, S.: Diabetic retinopathy diagnosis with ensemble deep-learning. ACM Inter. Conf. Proc. Ser. (2019). https://doi.org/10.1145/3387168.3387206
Ghosh, R., Ghosh, K., Maitra, S.: Automatic detection and classification of diabetic retinopathy stages using CNN. In: 2017 4th International Conference on Signal Processing and Integrated Networks, SPIN 2017, pp. 550–554 (2017). https://doi.org/10.1109/SPIN.2017.8050011
Xu, K., Feng, D., Mi, H.: Deep Convolutional Neural Network-Based Early Automated Detection of Diabetic Retinopathy Using Fundus Image. Molecules 2017, vol. 22, pp. 2054. 22, 2054 (2017). https://doi.org/10.3390/MOLECULES22122054
García, G., Gallardo, J., Mauricio, A., López, J., del Carpio, C.: Detection of Diabetic Retinopathy Based on a Convolutional Neural Network Using Retinal Fundus Images. LNCS(LNAI and LNBI), vol. 10614, pp. 635–642 (2017). https://doi.org/10.1007/978-3-319-68612-7_72
Kolla, M., Venugopal, T.: Efficient classification of diabetic retinopathy using binary CNN. In: Proceedings of 2nd IEEE International Conference on Computational Intelligence and Knowledge Economy, ICCIKE 2021, pp. 244–247 (2021). https://doi.org/10.1109/ICCIKE51210.2021.9410719
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Long, F., Xiong, H., Sang, J. (2024). A Classification Method for Diabetic Retinopathy Based on Self-supervised Learning. In: Huang, DS., Zhang, Q., Guo, J. (eds) Advanced Intelligent Computing in Bioinformatics. ICIC 2024. Lecture Notes in Computer Science(), vol 14881. Springer, Singapore. https://doi.org/10.1007/978-981-97-5689-6_30
Download citation
DOI: https://doi.org/10.1007/978-981-97-5689-6_30
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-97-5688-9
Online ISBN: 978-981-97-5689-6
eBook Packages: Computer ScienceComputer Science (R0)