Abstract
Diabetic retinopathy (DR) is a complication of diabetes and one of the causes of blindness and visual impairment. Automated and accurate DR grading is of great significance for timely and effective treatment of fundus diseases. At present, traditional convolutional neural networks (CNNs) will pay too much atteneion to the DR level with a lot of samples because of the unbalanced data distribution. So unbalanced data distribution will affect the classification ability of the model. In This paper, Double-Generator-Efficientnet Network (DGENet) based on Efficientnet is proposed to effectively improve the grading ability of DR, which consists of the following: (1) The first network classifies No DR and DR to solve the problem that No DR occupies most of the data distribution. (2) The second network divides the DR in the first network into mild, moderate, severe non-proliferative DR (NPDR) and proliferative DR(PDR). The second add the generative network from Deep Convolutional Generative Adversarial Networks (DCGAN) for data enhancement to solve the problem of imbalanced data in the four levels of DR. (3) Add attention mechanism modules to the four-level network. Experimental results show that the proposed DGENet outperforms some new methods in the publicly available Kaggle and Messidor-2 fundus image datasets, especially on the four levels of DR1 to DR4.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Cho, N.H., Shaw, J.E., Karuranga, S., et al.: IDF Diabetes Atlas: Global estimates of diabetes prevalence for 2017 and projections for 2045. Diab. Res. Clin. Pract. 138, 271–281 (2018)
Ding, J., Wong, T.Y.: Current epidemiology of diabetic retinopathy and diabetic macular edema. Curr. Diab. Rep. 12(4), 346–354 (2012)
Usman Akram, M., Khalid, S., Tariq, A., Khan, S.A., Azam, F.: Detection and classification of retinal lesions for grading of diabetic retinopathy. Comput. Biol. Med. 45(1), 161–171 (2014)
Haneda, S., Yamashita, H.: International clinical diabetic retinopathy disease severity scale. Nippon rinsho. Jpn. J. Clin. Med. 68(9), 228–235 (2010)
Wilkinson, C.P., et al.: Proposed international clinical diabetic retinopathy and diabetic macular edema disease severity scales. Ophthalmology 110(9), 1677–1682 (2003)
Zhou, J., et al.: Improved softmax loss for deep learning based face and expression recognition. Cogn. Comput. Syst. 1(4), 97–102 (2019)
Long, D., Zhang, S., Zhang, Y.: Performance prediction based on neural architecture features. Cogn. Comput. Syst. 2(2), 80–83 (2020)
Xu, X., Li, G., **e, G., et al.: Weakly supervised deep semantic segmentation using CNN and ELM with semantic candidate regions. Complexity 2019, 1–12 (2019)
Zhou, J., Zhao, W., Guo, L., Xu, X., **e, G.: Real time detection of surface defects with inception-based MobileNet-SSD detection network. In: Ren, J., et al. (eds.) BICS 2019. LNCS, vol. 11691, pp. 510–519. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-39431-8_49
Wang, G.: A perspective on deep imaging. IEEE Access 4, 8914–8924 (2017)
Jeyaraj, P.R., Nadar, E.: Deep Boltzmann machine algorithm for accurate medical image analysis for classification of cancerous region. Cogn. Comput. Syst. 1(3), 85–90 (2019)
Gargeya, R., Leng, T.: Automated identification of diabetic retinopathy using deep learning. Ophthalmology 124(7), 962–969 (2017)
Zhang, W., et al.: Automated identification and grading system of diabetic retinopathy using deep neural networks. Know. Bas. Syst. 175, 12–25 (2019)
Jordi, D.L.T., Alls, A.V., Puig, D.: A deep learning interpretable classifier for diabetic retinopathy disease grading. Neurocomputing 396, 465–476 (2019)
Van Grinsven, M., Van Ginneken, B., Hoyng, C., Theelen, T., Sanchez, C.: Fast convolutional neural network training using selective data sampling: application to hemorrhage detection in color fundus images. IEEE Trans. Med. Imag. 35(5), 1273–1284 (2016)
Zhou, K., et al.: Multi-cell multi-task convolutional neural networks for diabetic retinopathy grading. In: Proceedings of the Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS (2018)
Gulshan, V., et al.: Development and validation of a deep learning algorithm for detection of diabetic retinopathy in retinal fundus photographs. JAMA 316(22), 2402–2410 (2016)
Goodfellow, I.J., Pouget-Abadie, J., Mirza, M., et al.: Generative adversarial nets. In: International Conference on Neural Information Processing Systems. MIT Press, pp. 2672–2680 (2014)
Radford, A., Metz, L., Chintala, S.: Unsupervised representation learning with deep convolutional generative adversarial networks. Computer Science. ar**v:1511.06434v2 (2015)
Rezaei, M., et al.: A conditional adversarial network for semantic segmentation of brain tumor. In: Crimi, A., Bakas, S., Kuijf, H., Menze, B., Reyes, M. (eds.) BrainLes 2017. LNCS, vol. 10670, pp. 241–252. Springer, Cham (2017). https://doi.org/10.1007/978-3-319-75238-9_21
Shankaranarayana, S.M., Ram, K., Mitra, K., Sivaprakasam, M.: Joint optic disc and cup segmentation using fully convolutional and adversarial networks. In: Jorge Cardoso, M., Arbel, T., Melbourne, A., Bogunovic, H., Moeskops, P., Chen, X., Schwartz, E., Garvin, M., Robinson, E., Trucco, E., Ebner, M., Yanwu, Xu., Makropoulos, A., Desjardin, A., Vercauteren, T. (eds.) FIFI/OMIA -2017. LNCS, vol. 10554, pp. 168–176. Springer, Cham (2017). https://doi.org/10.1007/978-3-319-67561-9_19
Lahiri, A., Ayush, K., Biswas, P.K., et al.: Generative adversarial learning for reducing manual annotation in semantic segmentation on large scale miscroscopy images: automated vessel segmentation in retinal fundus image as test case. In: 2017 IEEE Conf Comp Vision Pattern Recog Workshops (CVPRW) Workshops (CVPRW), pp. 794–800 (2017)
Tan, M., Le, Q.V.: EfficientNet: rethinking model scaling for convolutional neural networks. In: ICML (2019)
Gulshan, V., et al.: Development and validation of a deep learning algorithm for detection of diabetic retinopathy in retinal fundus photographs. JAMA - J. Am. Med. Assoc. 316(22), 2402–2410 (2016)
Sun, Y., Chen, Y., Wang, X., Tang, X.: Deep learning face representation by joint identification-verification. In: Advances in Neural Information Processing Systems, pp. 1988–1996 (2017)
Decenciere, E., et al.: Feedback on a publicly distributed image database: the messidor database. Image Anal. Stereol. 33(3), 231–234 (2014)
Kwasigroch, A., Jarzembinski, B., Grochowski, M.: Deep CNN based decision support system for detection and assessing the stage of diabetic retinopathy. In: 2018 International Interdisciplinary PhD Workshop (IIPhDW); IEEE, pp.111–116 (2018)
Qummar, S., Khan, F.G., Shah, S., Khan, A., Shamshirband, S., Rehman, Z.U., et al.: A deep learning ensemble approach for diabetic retinopathy detection. IEEE Access 7, 150530–150539 (2019)
Li, X., Song, Y., Squirrell, D., et al.: Towards implementation of AI in New Zealand national diabetic screening program: cloud-based, robust, and bespoke. PLoS ONE 15(4), e0225015 (2020)
He, A., Li, T., Li, N., et al.: CABNet: category attention block for imbalanced diabetic retinopathy grading. IEEE Trans. Med. Imag. 40(1), 143–153 (2020)
Wu, Z., Shi, G., Chen, Y., et al.: Coarse-to-fine classification for diabetic retinopathy grading using convolutional neural network. Artif. Intell. Med. 108, 101936 (2020)
Hacisoftaoglu, R.E., Karakaya, M., Sallam, A.B.: Deep learning frameworks for diabetic retinopathy detection with smartphone-based retinal imaging systems. Patt. Recog. Lett. 135, 409–417 (2020)
Acknowledgments
This work was supported by the Natural Science Foundation of Shanxi Province (201801D121144) and the Natural Science Foundation of Shanxi Province (201901D211079).
Author information
Authors and Affiliations
Corresponding author
Rights and permissions
Copyright information
© 2022 Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Zhao, K., Zhao, W., **e, J., Li, B., Zhang, Z., Xu, X. (2022). Generative Adversarial Networks and Improved Efficientnet for Imbalanced Diabetic Retinopathy Grading. In: Sun, F., Hu, D., Wermter, S., Yang, L., Liu, H., Fang, B. (eds) Cognitive Systems and Information Processing. ICCSIP 2021. Communications in Computer and Information Science, vol 1515. Springer, Singapore. https://doi.org/10.1007/978-981-16-9247-5_27
Download citation
DOI: https://doi.org/10.1007/978-981-16-9247-5_27
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-16-9246-8
Online ISBN: 978-981-16-9247-5
eBook Packages: Computer ScienceComputer Science (R0)