Improved Support Vector Machine based on CNN-SVD for vision-threatening diabetic retinopathy detection and classification

PLoS One. 2024 Jan 2;19(1):e0295951. doi: 10.1371/journal.pone.0295951. eCollection 2024.

Abstract

The integration of artificial intelligence (AI) in diagnosing diabetic retinopathy, a major contributor to global vision impairment, is becoming increasingly pronounced. Notably, the detection of vision-threatening diabetic retinopathy (VTDR) has been significantly fortified through automated techniques. Traditionally, the reliance on manual analysis of retinal images, albeit slow and error-prone, constituted the conventional approach. Addressing this, our study introduces a novel methodology that amplifies the robustness and precision of the detection process. This is complemented by the groundbreaking Hierarchical Block Attention (HBA) and HBA-U-Net architecture, which notably propel attention mechanisms in image segmentation. This innovative model refines image processing without imposing excessive computational demands by honing in on individual pixel intricacies, spatial relationships, and channel-specific attention. Building upon this innovation, our proposed method employs a multi-stage strategy encompassing data pre-processing, feature extraction via a hybrid CNN-SVD model, and classification employing an amalgamation of Improved Support Vector Machine-Radial Basis Function (ISVM-RBF), DT, and KNN techniques. Rigorously tested on the IDRiD dataset classified into five severity tiers, the hybrid model yields remarkable performance, achieving a 99.18% accuracy, 98.15% sensitivity, and 100% specificity in VTDR detection, thus surpassing existing methods. These results underscore a more potent avenue for diagnosing and addressing this crucial ocular condition while underscoring AI's transformative potential in medical care, particularly in ophthalmology.

MeSH terms

  • Artificial Intelligence
  • Diabetes Mellitus*
  • Diabetic Retinopathy* / diagnostic imaging
  • Diagnosis, Computer-Assisted / methods
  • Humans
  • Image Interpretation, Computer-Assisted / methods
  • Support Vector Machine

Grants and funding

This research was funded by the Foreign Young Talents Program of the State Bureau of Foreign Experts Ministry of Science and Technology China (No. QN2023034001), Hainan Provincial Natural Science Foundation of China (No. 823RC488, No. 621RC1059), National Natural Science Foundation of China (No.62262019) and Haikou Science and Technology Plan Project of China (No.2022-016).