AOCT-NET: a convolutional network automated classification of multiclass retinal diseases using spectral-domain optical coherence tomography images
Since introducing optical coherence tomography (OCT) technology for 2D eye imaging, it has become one of the most important and widely used imaging modalities for the noninvasive assessment of retinal eye diseases. Age-related macular degeneration (AMD) and diabetic macular edema eye disease are the leading causes of blindness being diagnosed using OCT. Recently, by developing machine learning and deep learning techniques, the classification of eye retina diseases using OCT images has become quite a challenge. In this paper, a novel automated convolutional neural network (CNN) architecture for a multiclass classification system based on spectral-domain optical coherence tomography (SD-OCT) has been proposed. The system used to classify five types of retinal diseases (age-related macular degeneration (AMD), choroidal neovascularization (CNV), diabetic macular edema (DME), and drusen) in addition to normal cases. The proposed CNN architecture with a softmax classifier overall correctly identified 100% of cases with AMD, 98.86% of cases with CNV, 99.17% cases with DME, 98.97% cases with drusen, and 99.15% cases of normal with an overall accuracy of 95.30%. This architecture is a potentially impactful tool for the diagnosis of retinal diseases using SD-OCT images.
KeywordsRetina Optical coherence tomography Spectral domain Classification Deep learning
The author would like to thank the Nvidia Learning Center at Yarmouk University (YU), represented by Dr. Ahmad Alomari, for providing access to the GPU Unit for training and testing the CNN architecture on the whole image dataset and getting the results. Also, the author would like to thank Zain Innovation Campus (ZINC) at Yarmouk University (YU) represented by Eng. Nour Al-Ajlouni, for providing access to the GPU Unit for initial results and testing of the CNN architecture on part of the dataset.
Availability of data and materials
The image dataset that used to test and support the findings of this research is available from different hospitals in Jordan, but restrictions apply to the availability of these data, which were used under license for the current research only, and so are not publicly available. Data are however may be available from the author upon reasonable request and with permission of all hospitals.
Compliance with ethical standards
Conflict of interest
The author declares that he has no conflict of interest.
Human and animal rights
This article does not contain any studies with human participants or animals performed by the author.
Informed consent was obtained from all individual participants included in the study.
- 1.Kermany DS, Goldbaum M, Cai W, Valentim CCS, Liang H, Baxter SL, McKeown A, Yang G, Wu X, Yan F, Dong J, Prasadha MK, Pei J, Ting MYL, Zhu J, Li C, Hewett S, Dong J, Ziyar I, Shi A, Zhang R, Zheng L, Hou R, Shi W, Fu X, Duan Y, Huu VAN, Wen C, Zhang ED, Zhang CL, Li O, Wang X, Singer MA, Sun X, Xu J, Tafreshi A, Lewis MA, Xia H, Zhang K (2018) Identifying medical diagnoses and treatable diseases by image-based deep learning. Cell 172(5):1122–1131 e9CrossRefGoogle Scholar
- 2.Farsiu S, Chiu SJ, O'Connell RV, Folgar FA, Yuan E, Izatt JA, Toth CA, Age-Related Eye Disease Study 2 Ancillary Spectral Domain Optical Coherence Tomography Study Group (2014) Quantitative classification of eyes with and without intermediate age-related macular degeneration using optical coherence tomography. Ophthalmology 121(1):162–172CrossRefGoogle Scholar
- 9.Ting DSW, Wu WC, Toth C (2018) Deep learning for retinopathy of prematurity screening. Br J OphthalmolGoogle Scholar
- 14.Awais M, Müller H, Tang TB, Meriaudeau F (2017) Classification of sd-oct images using a deep learning approach. In: 2017 IEEE International Conference on Signal and Image Processing Applications (ICSIPA), IEEE, pp 489–492Google Scholar
- 22.Mehta P, Lee AY, Lee C, Balazinska M, Rokem A (2018) Multilabel multiclass classification of OCT images augmented with age, gender and visual acuity data. bioRxiv:316349Google Scholar
- 25.Perdomo O, Otálora S, González FA, Meriaudeau F, Müller H (2018) Oct-net: a convolutional network for automatic classification of normal and diabetic macular edema using SD-OCT volumes. In: 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018), IEEE, pp 1423-1426Google Scholar
- 27.Nugroho KA (2018) A comparison of handcrafted and deep neural network feature extraction for classifying optical coherence tomography (OCT) images. In: 2018 2nd International Conference on Informatics and Computational Sciences (ICICoS), IEEE, pp 1-6Google Scholar
- 29.Kingma DP, Ba J (2014) Adam: a method for stochastic optimization, arXiv preprint arXiv:1412.6980Google Scholar