Automatic Identification of Diabetic Macular Edema Using a Transfer Learning-Based Approach

Abstract

[Abstract] This paper presents a complete system for the automatic identification of pathological Diabetic Macular Edema (DME) cases using Optical Coherence Tomography (OCT) images as source of information. To do so, the system extracts a set of deep features using a transfer learning-based approach from different fully-connected layers and different pre-trained Convolutional Neural Network (CNN) models. Next, the most relevant subset of deep features is identified using representative feature selection methods. Finally, a machine learning strategy is applied to train and test the potential of the identified deep features in the pathological classification process. Satisfactory results were obtained, demonstrating the suitability of the presented system to filter those pathological DME cases, helping the specialist to optimize their diagnostic procedures.