Detecting Macular Disease Based on Optical Coherence Tomography Using a Deep Convolutional Network

초록

Neovascular age-related macular degeneration (nAMD) and central serous chorioretinopathy (CSC) are two of the most common macular diseases. This study proposes a convolutional neural network (CNN)-based deep learning model for classifying the subtypes of nAMD (polypoidal choroidal vasculopathy, retinal angiomatous proliferation, and typical nAMD) and CSC (chronic CSC and acute CSC) and healthy individuals using single spectral-domain optical coherence tomography (SD-OCT) images. The proposed model was trained and tested using 6063 SD-OCT images from 521 patients and 47 healthy participants. We used three well-known CNN architectures (VGG-16, VGG-19, and ResNet) and two customized classification layers. Additionally, transfer learning and mix-up-based data augmentation were applied to improve robustness and accuracy. Our model demonstrated high accuracies of 99.7% and 91.1% in the nAMD and CSC classification and retinopathy (nAMD and CSC) subtype classification, including normal participants, respectively. Furthermore, we performed an external test to compare the classification accuracy with that of eight ophthalmologists, and our model showed the highest accuracy. The region determined to be important for classification by the model was confirmed using gradient-weighted class activation mapping. The model's clinical criteria were similar to that of the ophthalmologists.

키워드