Deep Learning for Ocular Disease Recognition: An Inner-Class Balance.

Journal: Computational intelligence and neuroscience
Published Date:

Abstract

It can be challenging for doctors to identify eye disorders early enough using fundus pictures. Diagnosing ocular illnesses by hand is time-consuming, error-prone, and complicated. Therefore, an automated ocular disease detection system with computer-aided tools is necessary to detect various eye disorders using fundus pictures. Such a system is now possible as a consequence of deep learning algorithms that have improved image classification capabilities. A deep-learning-based approach to targeted ocular detection is presented in this study. For this study, we used state-of-the-art image classification algorithms, such as VGG-19, to classify the ODIR dataset, which contains 5000 images of eight different classes of the fundus. These classes represent different ocular diseases. However, the dataset within these classes is highly unbalanced. To resolve this issue, the work suggested converting this multiclass classification problem into a binary classification problem and taking the same number of images for both classifications. Then, the binary classifications were trained with VGG-19. The accuracy of the VGG-19 model was 98.13% for the normal (N) versus pathological myopia (M) class; the model reached an accuracy of 94.03% for normal (N) versus cataract (C), and the model provided an accuracy of 90.94% for normal (N) versus glaucoma (G). All of the other models also improve the accuracy when the data is balanced.

Authors

  • Md Shakib Khan
    Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh.
  • Nafisa Tafshir
    Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh.
  • Kazi Nabiul Alam
    Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh.
  • Abdur Rab Dhruba
    Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh.
  • Mohammad Monirujjaman Khan
    Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh.
  • Amani Abdulrahman Albraikan
    Department of Computer Sciences, College of Computer and Information Sciences, Princess Nourah Bint Abdulrahman University, P.O. Box 84428, Riyadh 11671, Saudi Arabia.
  • Faris A Almalki
    Department of Computer Engineering, College of Computers and Information Technology, Taif University, P.O. Box 11099, Taif 21944, Saudi Arabia.