Multi-Scale Attention-Based Mechanism in Gradient Boosting Convolutional Neural Network for Diabetic Retinopathy Grade Classification.

Autor: Srinivasan, Valarmathi, Rajagopal, Vijayabhanu
Předmět:
Zdroj: International Journal of Intelligent Engineering & Systems; 2022, Vol. 15 Issue 4, p489-498, 10p
Abstrakt: Diabetic Retinopathy (DR) is a common complication of Diabetes Mellitus (DM) that produces retinal abnormalities and can lead to blindness if not diagnosed and treated on time. To address this concern, an adaptive Convolutional Neural Network (CNN) model with Gradient Boosting (GB) called ResNetGB has been used from the literature where a Principal Component Analysis (PCA) based Fully Connected (FC) layer is used to capture the discriminative characteristics from the Retinal Fundus (RF) samples. It is essential to extract more effective features to categorize the DR grades. Hence, in this article, the Multi-Scale Attention (MSA) strategy is incorporated into the ResNetGB model for effective DR grade classification. First, the encoder network is used to embed the RF image in a high-level interpretational space in which the mixture of mid and high-level characteristics is considered to enhance the representation. Then, a Multi-Scale Feature Pyramid (MSFP) is added to define the retinal pattern in various localities and the MSA strategy is applied to the high-level interpretation. Moreover, the entire MSAResNetGB framework is trained by the cross-entropy loss to categorize the patients with respective DR grades. Finally, the experimental analysis exhibit that the MSA-ResNetGB model achieves the 94.40% and 94.17% accuracy on two benchmark datasets: Kaggle-APTOS and IDRiD, respectively compared to the cutting-edge models. [ABSTRACT FROM AUTHOR]
Databáze: Complementary Index