Classification system from Optical Coherence Tomography using transfer learning

Abstract

This research aims to create a decision support system to identify retinal diseases using a four-class classification problem. To achieve this, the proposed system uses deep learning architecture to automatically recognize CNV, DME, and drusen from OCT images. The model employs two transfer learning architectures with several additional layers to classify retinal diseases. The purpose of model training, validation, and testing, the experiment uses 6,000 grayscale images labeled into four classes from the OCT data set. The Inception V3 model’s proposed additional layer exhibits an increase in accuracy of 3.08% and a reduction in the loss by 0.3767. The experiment’s results indicate that the Inception V3 model achieved an accuracy rate of 99.31%, and the VGG-16 model reached 98.83%, which outperformed other deep learning models using the OCT data set.

Keywords

• deep learning
• transfer learning
• optical coherence tomography
• inception v3
• vgg-16