Deep hybrid architecture with stacked ensemble learning for binary classification of retinal disease

被引:4
|
作者
Priyadharsini, C. [1 ]
Phamila, Asnath Victy Y. [1 ]
机构
[1] Vellore Inst Technol, Sch Comp Sci Engn, Chennai 600127, India
关键词
Convolutional neural network; Machine Learning; Ensemble strategies; Stacking; Boosting models; Bagging models; DIABETIC-RETINOPATHY;
D O I
10.1016/j.rineng.2024.103219
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Purpose: Early retinal disease identification is vital since symptoms are passive at initial stages but lead to irreversible vision loss at advanced stages. Globally, a substantial population is at risk of vision impairment, prompting researchers to investigate methods for efficient classification. Methods: This work experimented one hundred and forty-four different hybrid architectures amalgamating each of the eight convolutional neural architectures (VGG, EfficientNet, Inception, ResNet, NasNet, DenseNet, InceptionResNet, Xception) with seven classifiers (Logistic regression, K-Nearest Neighbours, Support Vector Classifier, Decision Tree, Bagging classifier, Random Forest, Adaptive Boosting, Light Gradient Boost and Extra tree classifier). The top performing n (n=3,4,5) classifiers were ensembled with meta-learner using stacking strategy. The performance of the pipeline is evaluated with two distinct meta-learners, three different image sizes, and feature counts on the RFMiD dataset. Results: The architectures were assessed using (1) the performance metrics - accuracy, precision, recall, and F1score, (2) statistical graphics to understand the prevalence of classifiers, Borda count voting method to identify the best CNN model, (3) Tukey's honestly significance difference test to identify best-performing architecture. Two architectures achieved a high accuracy of 92.34 percent and F1 scores of 95.11 and 95.19. Conclusion: This is the first work to experiment with 144 combinations to identify suitable deep architecture for binary retinal disease classification. The study recommends Xception for feature extraction ensembled with ExtraTreeClassifier, Light gradient boosting machine, Random Forest, AdaBoost classifiers, and meta-learner as Logistic Regression. Another best-performing architecture is similar with a variation in DenseNet121 for feature extraction and support vector as a meta-learner.
引用
收藏
页数:20
相关论文
共 50 条
  • [1] Deep Ensemble Learning for Retinal Image Classification
    Ho, Edward
    Wang, Edward
    Youn, Saerom
    Sivajohan, Asaanth
    Lane, Kevin
    Chun, Jin
    Hutnik, Cindy M. L.
    TRANSLATIONAL VISION SCIENCE & TECHNOLOGY, 2022, 11 (10):
  • [2] Retinal diseases classification based on hybrid ensemble deep learning and optical coherence tomography images
    Pin, Kuntha
    Han, Jung Woo
    Nam, Yunyoung
    ELECTRONIC RESEARCH ARCHIVE, 2023, 31 (08): : 4843 - 4861
  • [3] Rice Disease Classification Using a Stacked Ensemble of Deep Convolutional Neural Networks
    Wang, Zhibin
    Wei, Yana
    Mu, Cuixia
    Zhang, Yunhe
    Qiao, Xiaojun
    SUSTAINABILITY, 2025, 17 (01)
  • [4] Stacked ensemble learning for facial gender classification using deep learning based features extraction
    Waris, Fazal
    Da, Feipeng
    Liu, Shanghuan
    CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2024, 27 (08): : 11491 - 11513
  • [5] Deep ensemble learning for Alzheimer's disease classification
    An, Ning
    Ding, Huitong
    Yang, Jiaoyun
    Au, Rhoda
    Ang, Ting F. A.
    JOURNAL OF BIOMEDICAL INFORMATICS, 2020, 105
  • [6] An Efficient Stacked Ensemble Model for Heart Disease Detection and Classification
    Abbas, Sidra
    Sampedro, Gabriel Avelino
    Alsubai, Shtwai
    Almadhor, Ahmad
    Kim, Tai-hoon
    CMC-COMPUTERS MATERIALS & CONTINUA, 2023, 77 (01): : 665 - 680
  • [7] Design ensemble deep learning model for pneumonia disease classification
    El Asnaoui, Khalid
    INTERNATIONAL JOURNAL OF MULTIMEDIA INFORMATION RETRIEVAL, 2021, 10 (01) : 55 - 68
  • [8] Classification of Retinal Images with Deep Learning for Early Detection of Diabetic Retinopathy Disease
    Yalcin, Nursel
    Alver, Seyfullah
    Uluhatun, Necla
    2018 26TH SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS CONFERENCE (SIU), 2018,
  • [9] Deep Learning Architectures for OCT Images Retinal Disease Classification
    Ranjitha Rajan
    S. N. Kumar
    SN Computer Science, 6 (2)
  • [10] An Interpretable Ensemble Deep Learning Model for Diabetic Retinopathy Disease Classification
    Jiang, Hongyang
    Yang, Kang
    Gao, Mengdi
    Zhang, Dongdong
    Ma, He
    Qian, Wei
    2019 41ST ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2019, : 2045 - 2048