A comparative study of machine learning models for automated detection and classification of retinal diseases in Ghana.

introductionRetinal diseases, a significant global health concern, often lead to severe vision impairment and blindness, resulting in substantial functional and social limitations. This study explored a novel goal of developing and comparing the performance of multiple state-of-the-art convolutional neural network (CNN) models for the automated detection and classification of retinal diseases using optical coherence tomography (OCT) images.

methodThe study utilized several models, including DenseNet121, ResNet50, Inception V3, MobileNet, and OCT images obtained from the WATBORG Eye Clinic, to detect and classify multiple retinal diseases such as glaucoma, macular edema, posterior vitreous detachment (PVD), and normal eye cases. The preprocessing techniques employed included data augmentation, resizing, and one-hot encoding. We also used the Gaussian Process-based Bayesian Optimization (GPBBO) approach to fine-tune the hyperparameters. Model performance was evaluated using the F1-Score, precision, recall, and area under the curve (AUC).

resultAll the CNN models evaluated in this study demonstrated a strong capability to detect and classify various retinal diseases with high accuracy. MobileNet achieved the highest accuracy at 96% and AUC of 0.975, closely followed by DenseNet121, which had 95% accuracy and an AUC of 0.963. Inception V3 and ResNet50, while not as high in accuracy, showed potential in specific contexts, with 83% and 79% accuracy, respectively.

conclusionThese results underscore the potential of advanced CNN models for diagnosing retinal diseases. With the exception of ResNet50, the other CNN models displayed accuracy levels that are comparable to other state-of-the-art deep learning models. Notably, MobileNet and DenseNet121 showed considerable promise for use in clinical settings, enabling healthcare practitioners to make rapid and accurate diagnoses of retinal diseases. Future research should focus on expanding datasets, integrating multi-modal data, exploring hybrid models, and validating these models in clinical environments to further enhance their performance and real-world applicability.