Deep Learning Classification on Optical Coherence Tomography Retina Images
Document Type
Article
Publication Date
7-1-2020
Abstract
This paper presents a novel deep learning classification technique applied on optical coherence tomography (OCT) retinal images. We propose the deep neural networks based on Vgg16 pre-trained network model. The OCT retinal image dataset consists of four classes, including three most common retina diseases and one normal retina scan. Because the scale of training data is not sufficiently large, we use the transfer learning technique. Since the convolutional neural networks are sensitive to a little data change, we use data augmentation to analyze the classified results on retinal images. The input grayscale OCT scan images are converted to RGB images using colormaps. We have evaluated different types of classifiers with variant parameters in training the network architecture. Experimental results show that testing accuracy of 99.48% can be obtained as combined on all the classes.
Identifier
85074088077 (Scopus)
Publication Title
International Journal of Pattern Recognition and Artificial Intelligence
External Full Text Location
https://doi.org/10.1142/S0218001420520023
ISSN
02180014
Issue
8
Volume
34
Fund Ref
National Science Foundation
Recommended Citation
Shih, Frank Y. and Patel, Himanshu, "Deep Learning Classification on Optical Coherence Tomography Retina Images" (2020). Faculty Publications. 5174.
https://digitalcommons.njit.edu/fac_pubs/5174
