DOI QR코드

DOI QR Code

HiGANCNN: A Hybrid Generative Adversarial Network and Convolutional Neural Network for Glaucoma Detection

  • Alsulami, Fairouz (Computer Science Department, Jamoum University College, Umm Al-Qura University) ;
  • Alseleahbi, Hind (Computer Science Department, Jamoum University College, Umm Al-Qura University) ;
  • Alsaedi, Rawan (Computer Science Department, Jamoum University College, Umm Al-Qura University) ;
  • Almaghdawi, Rasha (Computer Science Department, Jamoum University College, Umm Al-Qura University) ;
  • Alafif, Tarik (Computer Science Department, Jamoum University College, Umm Al-Qura University) ;
  • Ikram, Mohammad (Computer Science Department, Jamoum University College, Umm Al-Qura University) ;
  • Zong, Weiwei (WeCare.WeTeach) ;
  • Alzahrani, Yahya (Academic Affairs and Training, Secuirty Forces Hospital) ;
  • Bawazeer, Ahmed (Department of Ophthalmology, Faculty of medicine, King Abdulaziz University)
  • 투고 : 2022.09.05
  • 발행 : 2022.09.30

초록

Glaucoma is a chronic neuropathy that affects the optic nerve which can lead to blindness. The detection and prediction of glaucoma become possible using deep neural networks. However, the detection performance relies on the availability of a large number of data. Therefore, we propose different frameworks, including a hybrid of a generative adversarial network and a convolutional neural network to automate and increase the performance of glaucoma detection. The proposed frameworks are evaluated using five public glaucoma datasets. The framework which uses a Deconvolutional Generative Adversarial Network (DCGAN) and a DenseNet pre-trained model achieves 99.6%, 99.08%, 99.4%, 98.69%, and 92.95% of classification accuracy on RIMONE, Drishti-GS, ACRIMA, ORIGA-light, and HRF datasets respectively. Based on the experimental results and evaluation, the proposed framework closely competes with the state-of-the-art methods using the five public glaucoma datasets without requiring any manually preprocessing step.

키워드

참고문헌

  1. S. Pascual, A. Bonafonte, and J. Serra, "Segan: Speech enhancement generative adversarial network," arXiv preprint arXiv:1703.09452, 2017.
  2. Y.-C. Tham, X. Li, T. Y. Wong, H. A. Quigley, T. Aung, and C.-Y. Cheng, "Global prevalence of glaucoma and projections of glaucoma burden through 2040: a systematic review and meta-analysis," Ophthalmology, vol. 121, no. 11, pp. 2081-2090, 2014. https://doi.org/10.1016/j.ophtha.2014.05.013
  3. M. Aouf, S. Almotatiri, A. Bajahzar, and G. Kareem, "Glaucoma detection from fundus camera image," in 2019 Ninth International Conference on Intelligent Computing and Information Systems (ICICIS). IEEE, 2019, pp. 312-316.
  4. T. Khalil, S. Khalid, and A. M. Syed, "Review of machine learning techniques for glaucoma detection and prediction," in 2014 Science and Information Conference. IEEE, 2014, pp. 438-442.
  5. B. N. Kumar, R. Chauhan, and N. Dahiya, "Detection of glaucoma using image processing techniques: A review," in 2016 International Conference on Microelectronics, Computing and Communications (MicroCom). IEEE, 2016, pp. 1-6.
  6. M. Al Ghamdi, M. Li, M. Abdel-Mottaleb, and M. Abou Shousha, "Semi-supervised transfer learning for convolutional neural networks for glaucoma detection," in ICASSP 2019-2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 2019, pp. 3812-3816.
  7. S. Albawi, T. A. Mohammed, and S. Al-Zawi, "Understanding of a convolutional neural network," in 2017 International Conference on Engineering and Technology (ICET), 2017, pp. 1-6.
  8. L. Racic, T. Popovic, S. cakic, and S. Sandi, "Pneumonia detection using deep learning based on convolutional neural network," in 2021 25th International Conference on Information Technology (IT), 2021, pp. 1-4.
  9. H. Maryam Tavakoli, T. Xie, J. Shi, M. Hadzikadic, and Y. Ge, "Predicting neural deterioration in patients with alzheimer's disease using a convolutional neural network," in 2020 IEEE International Conference on Bioinformatics and Biomedicine (BIBM), 2020, pp. 1951-1958.
  10. N. George and C. V. Jiji, "Convolutional neural network for automatic segmentation of edi oct images," in 2019 10th International Conference on Computing, Communication and Networking Technologies (ICCCNT), 2019, pp. 1-5.
  11. E. Chen, X. Wu, C. Wang, and Y. Du, "Application of improved convolutional neural network in image classification," in 2019 International Conference on Machine Learning, Big Data and Business Intelligence (MLBDBI), 2019, pp. 109-113.
  12. Z. Zhang, X. Pan, S. Jiang, and P. Zhao, "High-quality face image generation based on generative adversarial networks," Journal of Visual Communication and Image Representation, vol. 71, p. 102719, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S1047320319303402 1047320319303402
  13. W. Chen, T. Zhang, and X. Zhao, "Semantic segmentation using generative adversarial network," in 2021 40th Chinese Control Conference (CCC), 2021, pp. 8492-8495.
  14. H. Zhang, T. Xu, H. Li, S. Zhang, X. Wang, X. Huang, and D. Metaxas, "Stackgan: Text to photo-realistic image synthesis with stacked generative adversarial networks," in 2017 IEEE International Conference on Computer Vision (ICCV), 2017, pp. 5908-5916.
  15. I. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. WardeFarley, S. Ozair, A. Courville, and Y. Bengio, "Generative adversarial nets," Advances in neural information processing systems, vol. 27, 2014.
  16. T. Alkhodidi, L. Aljoudi, A. Fallatah, A. Bashy, N. Ali, N. Alqahtani, N. Almajnooni, A. Allhabi, T. Albarakati, T. Alafif, and S. Sasi, "Geac: Generating and evaluating handwritten Arabic characters using generative adversarial networks," in 2021 International Conference on Computational Intelligence and Knowledge Economy (ICCIKE), 2021, pp. 228-233.
  17. S. Chintala, E. Denton, M. Arjovsky, and M. Mathieu, "How to train a gan? tips and tricks to make gans work," GitHub, Dec, 2016.
  18. D. D. Patil, R. R. Manza, G. C. Bedke, and D. D. Rathod, "Development of primary glaucoma classification technique using optic cup disc ratio," in 2015 International Conference on Pervasive Computing (ICPC), 2015, pp. 1-5.
  19. K. A. Thakoor, X. Li, E. Tsamis, P. Sajda, and D. C. Hood, "Enhancing the accuracy of glaucoma detection from oct probability maps using convolutional neural networks," in 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC). IEEE, 2019, pp. 2036-2040.
  20. A. Diaz-Pinto, A. Colomer, V. Naranjo, S. Morales, Y. Xu, and A. F. Frangi, "Retinal image synthesis and semisupervised learning for glaucoma assessment," IEEE transactions on medical imaging, vol. 38, no. 9, pp. 2211-2218, 2019. https://doi.org/10.1109/tmi.2019.2903434
  21. S. Ovreiu, I. Cristescu, F. Balta, and E. Ovreiu, "An exploratory study for glaucoma detection using densely connected neural networks," in 2020 International Conference on e-Health and Bioengineering (EHB). IEEE, 2020, pp. 1-4.
  22. T. R. V. Bisneto, A. O. de Carvalho Filho, and D. M. V. Magalhaes, "Generative adversarial network and texture features applied to automatic glaucoma detection," Applied Soft Computing, vol. 90, p. 106165, 2020.
  23. F. Fumero, S. Alayon, J. L. Sanchez, J. Sigut, and M. Gonzalez-Hernandez, "Rim-one: An open retinal image database for optic nerve evaluation," in 2011 24th International Symposium on Computer-Based Medical Systems (CBMS), 2011, pp. 1-6.
  24. J. Sivaswamy, S. R. Krishnadas, G. Datt Joshi, M. Jain, and A. U. Syed Tabish, "Drishti-gs: Retinal image dataset for optic nerve head(onh) segmentation," in 2014 IEEE 11th International Symposium on Biomedical Imaging (ISBI), 2014, pp. 53-56.
  25. Z. Zhang, F. S. Yin, J. Liu, W. K. Wong, N. M. Tan, B. H. Lee, J. Cheng, and T. Y. Wong, "Origa-light: An online retinal fundus image database for glaucoma analysis and research," in 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology, 2010, pp. 3065-3068.
  26. "High-resolution fundus (hrf) image database," https://www5.cs.fau.de/research/data/fundus-images/.
  27. A. Diaz-Pinto, S. Morales, V. Naranjo, T. Kohler, J. M. Mossi, and A. Navea, "Cnns for automatic glaucoma assessment using fundus images: an extensive validation," Biomedical engineering online, vol. 18, no. 1, pp. 1-19, 2019. https://doi.org/10.1186/s12938-018-0620-3
  28. O. Rochmawanti and F. Utaminingrum, "Chest x-ray image to classify lung diseases in different resolution size using densenet-121 architectures," in 6th International Conference on Sustainable Information Engineering and Technology 2021, 2021, pp. 327-331.
  29. L. G. Falconi, M. P' erez, and W. G. Aguilar, "Transfer learning in breast' mammogram abnormalities classification with mobilenet and nasnet," in 2019 International Conference on Systems, Signals and Image Processing (IWSSIP). IEEE, 2019, pp. 109-114.
  30. B. Al-Bander, W. Al-Nuaimy, M. A. Al-Taee, and Y. Zheng, "Automated glaucoma diagnosis using deep learning approach," in 2017 14th International Multi-Conference on Systems, Signals & Devices (SSD). IEEE, 2017, pp. 207-210.
  31. M. Abadi, A. Agarwal, P. Barham, E. Brevdo, Z. Chen, C. Citro, G. S. Corrado, A. Davis, J. Dean, M. Devin, S. Ghemawat, I. Goodfellow, A. Harp, G. Irving, M. Isard, Y. Jia, R. Jozefowicz, L. Kaiser, M. Kudlur, J. Levenberg, D. Mane, R. Monga, S. Moore, D. Murray, C. Olah, M. Schuster, J. Shlens, B. Steiner, I. Sutskever, K. Talwar, P. Tucker, V. Vanhoucke, V. Vasudevan, F. Viegas, O. Vinyals, P. Warden, M. Wattenberg, M. Wicke, Y. Yu, and X. Zheng, "Tensorflow: Large-scale machine learning on heterogeneous distributed systems," arXiv preprint arXiv:1603.04467, 2016.
  32. M. Heusel, H. Ramsauer, T. Unterthiner, B. Nessler, and S. Hochreiter, "Gans trained by a two time-scale update rule converge to a local nash equilibrium," in Advances in neural information processing systems, 2017, pp. 6626-6637.