| Literature DB >> 32103888 |
Miguel Angel Zapata1, Dídac Royo-Fibla1, Octavi Font1, José Ignacio Vela2,3, Ivanna Marcantonio2,3, Eduardo Ulises Moya-Sánchez4,5, Abraham Sánchez-Pérez5, Darío Garcia-Gasulla4, Ulises Cortés4,6, Eduard Ayguadé4,6, Jesus Labarta4,6.
Abstract
PURPOSE: To assess the performance of deep learning algorithms for different tasks in retinal fundus images: (1) detection of retinal fundus images versus optical coherence tomography (OCT) or other images, (2) evaluation of good quality retinal fundus images, (3) distinction between right eye (OD) and left eye (OS) retinal fundus images,(4) detection of age-related macular degeneration (AMD) and (5) detection of referable glaucomatous optic neuropathy (GON). PATIENTS AND METHODS: Five algorithms were designed. Retrospective study from a database of 306,302 images, Optretina's tagged dataset. Three different ophthalmologists, all retinal specialists, classified all images. The dataset was split per patient in a training (80%) and testing (20%) splits. Three different CNN architectures were employed, two of which were custom designed to minimize the number of parameters with minimal impact on its accuracy. Main outcome measure was area under the curve (AUC) with accuracy, sensitivity and specificity.Entities:
Keywords: artificial intelligence; retinal diseases; retinal fundus image; screening
Year: 2020 PMID: 32103888 PMCID: PMC7025650 DOI: 10.2147/OPTH.S235751
Source DB: PubMed Journal: Clin Ophthalmol ISSN: 1177-5467
Figure 1Distribution of nonmydriatic cameras (NMCs) and optical coherence tomography (OCT) used for the dataset. Only central retinal images were included.
Number of Images Used for Training the CNNs. Images Were Divided into Two Groups: One for Training the Model and Another Naïve Group for Validating the Training Procedure. Color Fundus photography (CFP), Optical coherence tomography (OCT), Retinal nerve fiber layer (RNFL), Right eye (OD), Left eye (OS), Age related macular degeneration (AMD), referable glaucomatous optic neuropathy (GON).
| Model | Classification Task | Images for Training | Data Augmentation from Keras | Number of Classes | Number of Ophthalmologists for Label the Images |
|---|---|---|---|---|---|
| 1 | Image type selection | 56,396 | Shear_range=0.2, | 4 | 1 |
| 2 | Data Collection Good quality images | 150,075 | Shear_range=0.2, | 2 | 3 |
| 3 | Data Display | 30,119 | Shear_range=0.2, | 2 | 1 |
| 4 | AMD | 8832 | Zoom_range 0.1 | 2 | 3 |
| 5 | GON | 3776 | Zoom_range 0.1 | 2 | 3 |
CNN Architecture for Algorithms: 1. Color Fundus Photography (CFP) versus Macular Optical Coherence Tomography (OCT), Retinal Nerve Fiber Layer in the OCT and Other images. 2. Good Quality Retinal Fundus image. 3. Right Eye versus Left Eye (OD/OS). 4. AMD. 5. Glaucomatous Optic Neuropathy (GON) Classification. (Notations are Based on Keras22)
| CNN-1 | AMD-Net | GON-Net |
|---|---|---|
| Input (128,128,3) | Input (512,512,3) | Input (224,224,3) |
| Conv_2D_1(ReLu,3,3,32) | Conv_2D_1(ReLu,5,5,32) | RESNET50 |
| Max_Pooling_1(2,2) | Max_Pooling_1(2,2) | Flatten |
| Conv_2D_2(ReLu,3,3,32) | Conv_2D_2(ReLu,3,3,32) | Dropout_1(0.5) |
| Max_Pooling_2(2,2) | Conv_2D_3(ReLu,3,3,32) | Dense_1(ReLu,64) |
| Conv_2D_3(ReLu,3,3,64) | Max_Pooling_2(2,2) | Dropout_2(0.2) |
| Max_Pooling_3(2,2) | Conv_2D_4(ReLu,3,3,32) | Dense_2 (Softmax, 2) |
| Flatten | Conv_2D_5(ReLu,3,3,32) | |
| Dense_1 (ReLu,64) | Max_Pooling_3(2,2) | |
| Dropout_1 (0,5) | Conv_2D_6(ReLu,3,3,32) | |
| Dense_2 (Sigmoid, 5*/2) | Conv_2D_7(ReLu,3,3,32) | |
| Max_Pooling_4(2,2) | ||
| We trained only for 50 epochs, batch size of 32 and learning rate of 0.001 for each model. | Conv_2D_8(ReLu,3,3,32) | |
| Conv_2D_9(ReLu,3,3,32) | ||
| Max_Pooling_5(2,2) | ||
| Conv_2D_10(ReLu,3,3,64) | ||
| Conv_2D_11(ReLu,3,3,64) | ||
| Max_Pooling_6(2,2) | ||
| Flatten | ||
| Dense_1(ReLu64) | ||
| Dropout_1(0.5) | ||
| Dense_2(ReLu,64) | ||
| Dropout_2(0.2) | ||
| Dense_2 (Softmax, 2) |
Figure 2(A) Examples of good quality images. Retinal fundus images are centered on the macula with correct focus, good visualization of the parafoveal vessels and over two disc diameters around the fovea, and correct observation of the optic disc (at least three-quarters) and the vascular arcades. (B) Examples of poor quality images.
Outcomes Calculated Using the Confusion Matrix in Different Algorithms
| Model | AUC | Accuracy | Sensitivity | Specificity |
|---|---|---|---|---|
| Image type | 0.979 (0.978–0.981) | 0.960 (0.957–0.962) | 0.977 (0.976–0.979) | 0.924 (0.920–0.929) |
| Eye | 0.989 (0.988–0.991) | 0.974 (0.973–0.977) | 0.983 (0.981–0.985) | 0.966 (0.963–0.970) |
| Image quality | 0.947 (0.945–0.948) | 0.918 (0.916–0.919) | 0.969 (0.968–0.970) | 0.818 (0.814–0.821) |
| AMD classification | 0.936 (0.922–0.946) | 0.863 (0.845–0.877) | 0.902 (0.880–0.916) | 0.825 (0.799–0.849) |
| GON classification | 0.863 (0.827–0.894) | 0.803 (0.760–0.836) | 0.768 (0.710–0.824) | 0.838 (0.784–0.888) |
Validation Confusion Matrix from the Five CNNs: 1. Color Fundus Photography (CFP) versus Other Images. 2. Good Quality Retinal Fundus Images. 3. Right Eye versus Left Eye (OD/OS). 4. Age Related Macular Degeneration (AMD). 5. Glaucomatous Optic Neuropathy (GON)
| Filetype | ||
|---|---|---|
| Prediction | False | True |
| False | 31% | 2% |
| True | 3% | 64% |
| Laterality | ||
| Prediction | False | True |
| False | 48% | 1% |
| True | 2% | 49% |
| Quality assessment | ||
| Prediction | False | True |
| False | 28% | 2% |
| True | 6% | 64% |
| AMD | ||
| Prediction | False | True |
| False | 42% | 5% |
| True | 9% | 44% |
| GON | ||
| Prediction | False | True |
| False | 42% | 11% |
| True | 8% | 39% |
Figure 3Receiver operating characteristic (ROC) curves from the different CNNs. (A) Image type. (B) Quality assessment. (C) Laterality. (D) AMD. (E) GON.