Vol. 50, Issue 4, pp. 567-577

Vol. 50 Issue 4 pp. 567-577

Optical coherence tomography image for automatic classification of diabetic macular edema

Ping Wang, Jia-Li Li, Hao Ding

Keywords

diabetic macular edema, optical coherence tomography, transfer learning, fine-tuning

Abstract

Diabetic macular edema (DME) is the dominant reason of diabetic visual loss, so early detection and treatment of DME is of great significance for the treatment of diabetes. Based on transfer learning, an automatic classification method is proposed to distinguish DME images from normal images in optical coherence tomography (OCT) retinal fundus images. Features of the DME are automatically identified and extracted by the pre-trained convolutional neural network (CNN), which only involves fine-tuning the VGGNet-16 network without any user intervention. An accuracy of 97.9% and a sensitivity of 98.0% are acquired with the OCT images in the Duke data set from experimental results. The proposed method, a core part of an automated diagnosis system of the DME, revealed the ability of fine-tuning models to train non-medical images, allowing them can be classified with limited training data. Moreover, it can be developed to assist early diagnosis of the disease, effectively delaying (or avoiding) the progression of the disease, consequently.

Vol. 50
Issue 4
Article No: 05
pp. 567-577

0.62 MB
OPTICA APPLICATA - a quarterly of the Wrocław University of Science and Technology, Faculty of Fundamental Problems of Technology