Stacked ensemble deep learning for pancreas cancer classification using extreme gradient boosting

被引:2
|
作者
Bakasa, Wilson [1 ]
Viriri, Serestina [1 ]
机构
[1] Univ KwaZulu Natal, Coll Agr Engn & Sci, Sch Math Stat & Comp Sci, Durban, South Africa
来源
关键词
stacking ensemble; deep learning; XGBoost; hyperparameters; classification; pancreas segmentation; XGBOOST; IMBALANCE;
D O I
10.3389/frai.2023.1232640
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Ensemble learning aims to improve prediction performance by combining several models or forecasts. However, how much and which ensemble learning techniques are useful in deep learning-based pipelines for pancreas computed tomography (CT) image classification is a challenge. Ensemble approaches are the most advanced solution to many machine learning problems. These techniques entail training multiple models and combining their predictions to improve the predictive performance of a single model. This article introduces the idea of Stacked Ensemble Deep Learning (SEDL), a pipeline for classifying pancreas CT medical images. The weak learners are Inception V3, VGG16, and ResNet34, and we employed a stacking ensemble. By combining the first-level predictions, an input train set for XGBoost, the ensemble model at the second level of prediction, is created. Extreme Gradient Boosting (XGBoost), employed as a strong learner, will make the final classification. Our findings showed that SEDL performed better, with a 98.8% ensemble accuracy, after some adjustments to the hyperparameters. The Cancer Imaging Archive (TCIA) public access dataset consists of 80 pancreas CT scans with a resolution of 512 * 512 pixels, from 53 male and 27 female subjects. A sample of two hundred and twenty-two images was used for training and testing data. We concluded that implementing the SEDL technique is an effective way to strengthen the robustness and increase the performance of the pipeline for classifying pancreas CT medical images. Interestingly, grouping like-minded or talented learners does not make a difference.
引用
收藏
页数:17
相关论文
共 50 条
  • [1] Accelerating the Performance of Sequence Classification Using GPU Based Ensemble Learning with Extreme Gradient Boosting
    Kaur, Karamjeet
    Sagar, Anil Kumar
    Chakraborty, Sudeshna
    Gupta, Manoj Kumar
    [J]. ADVANCES IN COMPUTING AND DATA SCIENCES (ICACDS 2022), PT I, 2022, 1613 : 257 - 268
  • [2] Mammographic Classification Using Stacked Ensemble Learning with Bagging and Boosting Techniques
    Abubacker, Nirase Fathima
    Hashem, Ibrahim Abaker Targio
    Hui, Lim Kun
    [J]. JOURNAL OF MEDICAL AND BIOLOGICAL ENGINEERING, 2020, 40 (06) : 908 - 916
  • [3] Mammographic Classification Using Stacked Ensemble Learning with Bagging and Boosting Techniques
    Nirase Fathima Abubacker
    Ibrahim Abaker Targio Hashem
    Lim Kun Hui
    [J]. Journal of Medical and Biological Engineering, 2020, 40 : 908 - 916
  • [4] Stacked ensemble learning for facial gender classification using deep learning based features extraction
    Waris, Fazal
    Da, Feipeng
    Liu, Shanghuan
    [J]. CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2024, 27 (08): : 11491 - 11513
  • [5] A boosting ensemble learning based hybrid light gradient boosting machine and extreme gradient boosting model for predicting house prices
    Sibindi, Racheal
    Mwangi, Ronald Waweru
    Waititu, Anthony Gichuhi
    [J]. ENGINEERING REPORTS, 2023, 5 (04)
  • [6] Extreme gradient boosting and deep neural network based ensemble learning approach to forecast hourly solar irradiance
    Kumari, Pratima
    Toshniwal, Durga
    [J]. JOURNAL OF CLEANER PRODUCTION, 2021, 279
  • [7] A hybrid model for detecting intrusions using stacked autoencoders and extreme gradient boosting
    M.V., Hari Vinayak
    T., Jarin
    [J]. Computers and Security, 2025, 150
  • [8] Extreme Gradient Boosting for yield estimation compared with Deep Learning approaches
    Huber, Florian
    Yushchenko, Artem
    Stratmann, Benedikt
    Steinhage, Volker
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 202
  • [9] Extreme Gradient Boosting for yield estimation compared with Deep Learning approaches
    Huber, Florian
    Yushchenko, Artem
    Stratmann, Benedikt
    Steinhage, Volker
    [J]. Computers and Electronics in Agriculture, 2022, 202
  • [10] On using eXtreme Gradient Boosting (XGBoost) Machine Learning algorithm for Home Network Traffic Classification
    Cherif, Iyad Lahsen
    Kortebi, Abdesselem
    [J]. 2019 WIRELESS DAYS (WD), 2019,