Deep Feature Analysis in a Transfer Learning-based Approach for the Automatic Identification of Diabetic Macular Edema

Diabetic Macular Edema (DME) is one of the most common causes of vision impairment and blindness in individuals with diabetes. Among the different imaging modalities, Optical Coherence Tomography (OCT) is a non-invasive ophthalmological imaging technique that is commonly used for the diagnosis, monitoring and treatment of DME. In this context, this paper proposes a new methodology for the automatic classification of DME using OCT images. Firstly, the method extracts a set of deep features from the target OCT images using a transfer learning-based approach. Then, the most relevant subset of deep features is selected using different feature selection strategies. Finally, a machine learning approach is applied to test the potential of the implemented method. The proposed methodology was validated using an OCT image dataset retrieved from 400 different patients, being 200 with DME and 200 normal cases. The proposed system achieved satisfactory results, reaching a best accuracy of 97.50%, using only 14.65% of the deep features in the classification of this ocular pathology, demonstrating also its competitive performance with respect to others approaches of the state-of-the-art.

[1]  David S. Johnson,et al.  Support Vector Machines for Classification , 2022 .

[2]  Kunio Doi,et al.  Computer-aided diagnosis in medical imaging: Historical review, current status and future potential , 2007, Comput. Medical Imaging Graph..

[3]  Muhammad Awais,et al.  Classification of SD-OCT images using a Deep learning approach , 2017, 2017 IEEE International Conference on Signal and Image Processing Applications (ICSIPA).

[4]  F. Zhou,et al.  Machine learning based detection of age-related macular degeneration (AMD) and diabetic macular edema (DME) from optical coherence tomography (OCT) images. , 2016, Biomedical optics express.

[5]  Fabrice Mériaudeau,et al.  Transfer learning for Diabetic Macular Edema (DME) detection on Optical Coherence Tomography (OCT) images , 2017, 2017 IEEE International Conference on Signal and Image Processing Applications (ICSIPA).

[6]  Fei-Fei Li,et al.  ImageNet: A large-scale hierarchical image database , 2009, 2009 IEEE Conference on Computer Vision and Pattern Recognition.

[7]  Gadi Wollstein,et al.  OCT for glaucoma diagnosis, screening and detection of glaucoma progression , 2013, British Journal of Ophthalmology.

[8]  Khaled Alsaih,et al.  Machine learning techniques for diabetic macular edema (DME) classification on SD-OCT images , 2017, BioMedical Engineering OnLine.

[9]  Mark A. Hall,et al.  Correlation-based Feature Selection for Machine Learning , 2003 .

[10]  Sina Farsiu,et al.  Validated automatic segmentation of AMD pathology including drusen and geographic atrophy in SD-OCT images. , 2012, Investigative ophthalmology & visual science.

[11]  Marcos Ortega,et al.  Automatic macular edema identification and characterization using OCT images , 2018, Comput. Methods Programs Biomed..

[12]  Marko Robnik-Sikonja,et al.  Theoretical and Empirical Analysis of ReliefF and RReliefF , 2003, Machine Learning.

[13]  Kotagiri Ramamohanarao,et al.  Classification of healthy and diseased retina using SD-OCT imaging and Random Forest algorithm , 2018, PloS one.

[14]  Yurong Liu,et al.  A survey of deep neural network architectures and their applications , 2017, Neurocomputing.

[15]  Gretchen A. Stevens,et al.  Magnitude, temporal trends, and projections of the global prevalence of blindness and distance and near vision impairment: a systematic review and meta-analysis. , 2017, The Lancet. Global health.

[16]  Yiu-ming Cheung,et al.  Feature Selection and Kernel Learning for Local Learning-Based Clustering , 2011, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[17]  Marcos Ortega,et al.  Automatic Detection of Epiretinal Membrane in OCT Images by Means of Local Luminosity Patterns , 2017, IWANN.

[18]  Marcos Ortega,et al.  Enhanced visualization of the retinal vasculature using depth information in OCT , 2017, Medical & Biological Engineering & Computing.

[19]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[20]  Debjani Chakraborty,et al.  Transfer learning based classification of optical coherence tomography images with diabetic macular edema and dry age-related macular degeneration. , 2017, Biomedical optics express.

[21]  Raoof Gholami,et al.  Support Vector Machine: Principles, Parameters, and Applications , 2017 .

[22]  Sergey Ioffe,et al.  Rethinking the Inception Architecture for Computer Vision , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[23]  Manuel G. Penedo,et al.  Automatic Identification of Intraretinal Cystoid Regions in Optical Coherence Tomography , 2017, AIME.

[24]  Larry A. Rendell,et al.  A Practical Approach to Feature Selection , 1992, ML.

[25]  Nikola Bogunovic,et al.  A review of feature selection methods with applications , 2015, 2015 38th International Convention on Information and Communication Technology, Electronics and Microelectronics (MIPRO).

[26]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[27]  Sina Farsiu,et al.  Fully automated detection of diabetic macular edema and dry age-related macular degeneration from optical coherence tomography images. , 2014, Biomedical optics express.