Computer-aided diagnosis of early-stage Retinopathy of Prematurity in neonatal fundus images using artificial intelligence

Abstract

Retinopathy of Prematurity (ROP) is a retinal disorder affecting preterm babies, which can lead to permanent blindness without treatment. Early-stage ROP diagnosis is vital in providing optimal therapy for the neonates. The proposed study predicts early-stage ROP from neonatal fundus images using Machine Learning (ML) classifiers and Convolution Neural Networks (CNN) based pre-trained networks. The characteristic demarcation lines and ridges in early stage ROP are segmented utilising a novel Swin U-Net. 2000 Scale Invariant Feature Transform (SIFT) descriptors were extracted from the segmented ridges and are dimensionally reduced to 50 features using Principal Component Analysis (PCA). Seven ROP-specific features, including six Gray Level Co-occurrence Matrix (GLCM) and ridge length features, are extracted from the segmented image and are fused with the PCA reduced 50 SIFT features. Finally, three ML classifiers, such as Support Vector Machine (SVM), Random Forest (RF), and k- Nearest Neighbor (k-NN), are used to classify the 50 features to predict the early-stage ROP from Normal images. On the other hand, the raw retinal images are classified directly into normal and early-stage ROP using six pre-trained classifiers, namely ResNet50, ShuffleNet V2, EfficientNet, MobileNet, VGG16, and DarkNet19. It is seen that the ResNet50 network outperformed all other networks in predicting early-stage ROP with 89.5% accuracy, 87.5% sensitivity, 91.5% specificity, 91.1% precision, 88% NPV and an Area Under the Curve (AUC) of 0.92. Swin U-Net Convolutional Neural Networks (CNN) segmented the ridges and demarcation lines with an accuracy of 89.7% with 80.5% precision, 92.6% recall, 75.76% IoU, and 0.86 as the Dice coefficient. The SVM classifier using the 57 features from the segmented images achieved a classification accuracy of 88.75%, sensitivity of 90%, specificity of 87.5%, and an AUC of 0.91. The system can be utilised as a point-of-care diagnostic tool for ROP diagnosis of neonates in remote areas.