An ensemble method of the machine learning to prognosticate the gastric cancer

被引:30
|
作者
Rezaei, Hirad Baradaran [1 ]
Amjadian, Alireza [2 ]
Sebt, Mohammad Vahid [2 ]
Askari, Reza [2 ]
Gharaei, Abolfazl [2 ]
机构
[1] Amirkabir Univ Technol, Fac Ind Engn, Dept Ind Engn, Tehran, Iran
[2] Kharazmi Univ, Fac Engn, Dept Ind Engn, Tehran, Iran
关键词
Gastric cancer; Ensemble learning; Machine learning; Classification; Mutual information; Stacking; ARTIFICIAL-INTELLIGENCE; PREDICTION; SURVIVAL; SELECTION;
D O I
10.1007/s10479-022-04964-1
中图分类号
C93 [管理学]; O22 [运筹学];
学科分类号
070105 ; 12 ; 1201 ; 1202 ; 120202 ;
摘要
Gastric Cancer is the most common malignancy of the digestive tract, which is the third leading cause of cancer-related mortality worldwide. The early prognosis methods, especially Machine Learning (ML)-based approaches are one of the main strategies against GC, which have become a necessity to identify and prognosticate the factors that affect the GC. They enable the specialists to accelerate the subsequent clinical management of patients, who suffer from the GC. This paper aims at creating an Ensemble Method inspired from ML to predict the most significant factors of the GC occurrence. The main objective of this research is to predict the probabilities of the GC occurrence and its associated deaths. To achieve this goal, the created EM benefits from some ML-based methods, including Least Absolute Shrinkage and Selection Operator (LASSO)/Ridge Regression, Elastic Net, Logistic Regression (LR), Random Forest (RF), Gradient Boosting Decision Trees (GBDTs), and Deep Neural Network (DNN). The purpose of the provided EM is to lessen the prediction errors for the large number of the patients' features. The main novelties of this research include: (i) A sequential EM created by a Stacking method to predict the probability of the GC and associated deaths; (ii) Benefiting from the significance level to make an accurate prediction; (iii) Employing two Chi-square tests to select the influent features; (iv) Tuning the parameters of the applied ML models to avoid over-fitting and intensifying the errors; (v) Applying different kinds of the regression methods to treat hyper-dimension cases; (vi) A new model for weighting the applied ML models. The outcomes of the implementation of the created EM in seven pioneer hospitals in the field of GC show that the designed EM generates more precise predictions with an accuracy of 97.9% and 76.3% to predict the GC and its associated deaths, respectively. Moreover, the obtained results from the Area Under Curve (AUC) validates and confirms the capability of the created EM to predict the probability of the GC and its related deaths with an accuracy of 98% and 90% to predict the GC and its associated deaths, respectively.
引用
收藏
页码:151 / 192
页数:42
相关论文
共 50 条
  • [1] An ensemble method of the machine learning to prognosticate the gastric cancer
    Hirad Baradaran Rezaei
    Alireza Amjadian
    Mohammad Vahid Sebt
    Reza Askari
    Abolfazl Gharaei
    Annals of Operations Research, 2023, 328 : 151 - 192
  • [2] Non-endoscopic Applications of Machine Learning in Gastric Cancer: A Systematic Review
    Sy-Janairo, Marianne Linley L.
    Janairo, Jose Isagani B.
    JOURNAL OF GASTROINTESTINAL CANCER, 2024, 55 (01) : 47 - 64
  • [3] Study on a confidence machine learning method based on ensemble learning
    Jiang, Fang Chun
    CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2017, 20 (04): : 3357 - 3368
  • [4] Optimized Stacking Ensemble Learning Model for Breast Cancer Detection and Classification Using Machine Learning
    Kumar, Mukesh
    Singhal, Saurabh
    Shekhar, Shashi
    Sharma, Bhisham
    Srivastava, Gautam
    SUSTAINABILITY, 2022, 14 (21)
  • [5] Machine Learning Meets Cancer
    Varlamova, Elena V.
    Butakova, Maria A.
    Semyonova, Vlada V.
    Soldatov, Sergey A.
    Poltavskiy, Artem V.
    Kit, Oleg I.
    Soldatov, Alexander V.
    CANCERS, 2024, 16 (06)
  • [6] Diagnosis of Gastric Cancer Using Machine Learning Techniques in Healthcare Sector: A Survey
    Jamil, Danish
    Palaniappan, Sellappan
    Lokman, Asiah
    Jamil, Danish
    Naseem, Muhammad
    Zia, Syed Saood
    INFORMATICA-AN INTERNATIONAL JOURNAL OF COMPUTING AND INFORMATICS, 2021, 45 (07): : 147 - 166
  • [7] Identification of Prognostic Biomarkers for Gastric Cancer Using a Machine Learning Method
    Li, Chunguang
    Gong, Cheng
    Chen, Wenhao
    Li, Daojiang
    Xie, Youli
    Tao, Wenhui
    JOURNAL OF BIOLOGICAL REGULATORS AND HOMEOSTATIC AGENTS, 2023, 37 (01) : 259 - 270
  • [8] Study on a confidence machine learning method based on ensemble learning
    Fang Chun Jiang
    Cluster Computing, 2017, 20 : 3357 - 3368
  • [9] A practical framework for early detection of diabetes using ensemble machine learning models
    Saihood, Qusay
    Sonuc, Emrullah
    TURKISH JOURNAL OF ELECTRICAL ENGINEERING AND COMPUTER SCIENCES, 2023, 31 (04) : 722 - 738
  • [10] Item Response Theory Based Ensemble in Machine Learning
    Chen, Ziheng
    Ahn, Hongshik
    INTERNATIONAL JOURNAL OF AUTOMATION AND COMPUTING, 2020, 17 (05) : 621 - 636