Discrimination of diabetic retinopathy from optical coherence tomography angiography images using machine learning methods

Zhiping Liu, Chen Wang, Xiaodong Cai, Hong Jiang, Jianhua Wang

Research output: Contribution to journalArticlepeer-review

Abstract

Purpose: The goal was to discriminate between diabetic retinopathy (DR) and healthy controls (HC) by evaluating Optical coherence tomography angiography (OCTA) images from 3×3 mm scans with the assistance of different machine learning models. Methods: The OCTA angiography dataset of superficial vascular plexus (SVP), deep vascular plexus (DVP), and retinal vascular network (RVN) were acquired from 19 DR (38 eyes) patients and 25 HC (44 eyes). A discrete wavelet transform was applied to extract texture features from each image. Four machine learning models, including logistic regression (LR), logistic regression regularized with the elastic net penalty (LR-EN), support vector machine (SVM), and the gradient boosting tree named XGBoost, were used to classify wavelet features between groups. The area under the receiver operating characteristics curve (AUC), sensitivity, specificity, and diagnostic accuracy of the classifiers were obtained. Results: The OCTA image dataset included 114 and 132 images from DR and HC subjects, respectively. LR-EN and LR using all three images, SVP, DVP, and RVN, provided the highest sensitivity of 0.84 and specificity of 0.80, the best diagnostic accuracy of 0.82, and an AUC of 0.83 and 0.84, respectively, which were slightly lower than that of LR using one image SVP (0.85) or two images DVP and SVP (0.85). Conclusion: The LR-EN and LR classification algorithms had the high sensitivity, specificity, and diagnostic accuracy in identifying DR, which may be promising in facilitating the early diagnosis of DR.

Original languageEnglish (US)
JournalIEEE Access
DOIs
StateAccepted/In press - 2021

Keywords

  • Discrete wavelet transforms
  • Feature extraction
  • Logistics
  • Retina
  • Retinopathy
  • Sensitivity
  • Support vector machines
  • diabetic retinopathy
  • logistic regression
  • logistic regression regularized with the elastic net penalty
  • machine learning
  • support vector machine

ASJC Scopus subject areas

  • Computer Science(all)
  • Materials Science(all)
  • Engineering(all)

Fingerprint Dive into the research topics of 'Discrimination of diabetic retinopathy from optical coherence tomography angiography images using machine learning methods'. Together they form a unique fingerprint.

Cite this