Multi-stage glaucoma classification using pre-trained convolutional neural networks and voting-based classifier fusion

Vijaya Kumar Velpula; Lakhan Dev Sharma

doi:10.3389/fphys.2023.1175881

Multi-stage glaucoma classification using pre-trained convolutional neural networks and voting-based classifier fusion

Front Physiol. 2023 Jun 13:14:1175881. doi: 10.3389/fphys.2023.1175881. eCollection 2023.

Authors

Vijaya Kumar Velpula¹, Lakhan Dev Sharma¹

Affiliation

¹ School of Electronics Engineering, VIT-AP University, Amaravati, Andhra Pradesh, India.

Abstract

Aim: To design an automated glaucoma detection system for early detection of glaucoma using fundus images. Background: Glaucoma is a serious eye problem that can cause vision loss and even permanent blindness. Early detection and prevention are crucial for effective treatment. Traditional diagnostic approaches are time consuming, manual, and often inaccurate, thus making automated glaucoma diagnosis necessary. Objective: To propose an automated glaucoma stage classification model using pre-trained deep convolutional neural network (CNN) models and classifier fusion. Methods: The proposed model utilized five pre-trained CNN models: ResNet50, AlexNet, VGG19, DenseNet-201, and Inception-ResNet-v2. The model was tested using four public datasets: ACRIMA, RIM-ONE, Harvard Dataverse (HVD), and Drishti. Classifier fusion was created to merge the decisions of all CNN models using the maximum voting-based approach. Results: The proposed model achieved an area under the curve of 1 and an accuracy of 99.57% for the ACRIMA dataset. The HVD dataset had an area under the curve of 0.97 and an accuracy of 85.43%. The accuracy rates for Drishti and RIM-ONE were 90.55 and 94.95%, respectively. The experimental results showed that the proposed model performed better than the state-of-the-art methods in classifying glaucoma in its early stages. Understanding the model output includes both attribution-based methods such as activations and gradient class activation map and perturbation-based methods such as locally interpretable model-agnostic explanations and occlusion sensitivity, which generate heatmaps of various sections of an image for model prediction. Conclusion: The proposed automated glaucoma stage classification model using pre-trained CNN models and classifier fusion is an effective method for the early detection of glaucoma. The results indicate high accuracy rates and superior performance compared to the existing methods.

Keywords: classifier fusion; convolutional neural network; deep learning; fundus image; hybrid model; transfer learning.