A hybrid model for diabetic retinopathy and diabetic macular edema severity grade classification
摘要
Diabetic retinopathy (DR) is a common reason for permanent vision loss, particularly among the elderly population worldwide; diabetic macular edema (DME) is characterized by the accumulation of fluid or swelling in the macula, which can occur in any phase of DR progression. New studies have deepened our knowledge of the needs in eye care to improve methods for recognizing, addressing, diagnosing, and treating retinal conditions.
ObjectiveThe work intends to present a scheme for disease severity classification following the processes: pre-processing, segmentation, and disease severity classification. Firstly, pre-processing is done by employing a median filter, followed by GD-STFA-based SwinUNet for segmentation of lesions into soft and hard exudates, hemorrhages, and microaneurysms.
MethodsThen, DR and DME severity classification is employed by AlexNet-DQN, which is proposed by combining AlexNet and DQN models. Moreover, the proposed AlexNet-DQN is trained using the Exponential Gradient Descent-Sea Turtle Foraging Algorithm (ExpGD-STFA), which is developed by including the exponential weighted moving average (EWMA) concept in GD-STFA.
ResultThe presented mechanism attained better results when compared to traditional methods.
ConclusionFor DR severity classification, the proposed mechanism attained an accuracy of 98.3%, sensitivity of 98.9%, specificity of 97.8%, and precision of 97.6%. Meanwhile, for DME severity classification, the presented mechanism attained an accuracy of 98.1%, sensitivity of 98.8%, specificity of 97.5%, and precision of 97.3%.