{"title":"An Ensemble Learning System Based on Stacking Strategy for Survival Risk Prediction of Patients with Esophageal Cancer","authors":"","doi":"10.1016/j.irbm.2024.100860","DOIUrl":null,"url":null,"abstract":"<div><div><em>Background</em>: Predicting the prognosis of esophageal cancer (EC) patients is crucial for optimizing the treatment plan and allocating medical resources effectively.</div><div><em>Methods</em>: This study proposes a novel ensemble learning-based EC survival prediction model. Firstly, recursive feature elimination (RFE) is used to determine the key feature subsets from the dataset. Based on the determined key features, the improved clustering by fast search and find of density peaks (IDPC) is proposed to construct a novel indicator related to EC survival risk. The cosine distance is introduced in IDPC to cluster samples with similar characteristics. Then, the adaptive synthetic (ADASYN) technique is used to generate more high-risk samples to balance high-risk and low-risk samples. Finally, the hyperparameters of the three models, including extreme gradient boosting (XGBoost), adaptive boosting (AdaBoost), and random forest (RF), are optimized by whale optimization algorithm (WOA) and a new stacking model is constructed to evaluate the survival risk of patients.</div><div><em>Results</em>: The proposed stacking model achieved an area under the receiver operating characteristic curve (AUC) of 0.952 and Accuracy of 0.899, on the dataset from the First Affiliated Hospital of Zhengzhou University.</div><div><em>Conclusions</em>: The survival prediction model the proposed ensemble learning system is much more accurate and convenient, providing a basis clinical judgment and decision making and improving the survival status of esophageal cancer patients.</div></div>","PeriodicalId":14605,"journal":{"name":"Irbm","volume":null,"pages":null},"PeriodicalIF":5.6000,"publicationDate":"2024-09-11","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"Irbm","FirstCategoryId":"5","ListUrlMain":"https://www.sciencedirect.com/science/article/pii/S1959031824000411","RegionNum":4,"RegionCategory":"医学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"ENGINEERING, BIOMEDICAL","Score":null,"Total":0}
引用次数: 0
Abstract
Background: Predicting the prognosis of esophageal cancer (EC) patients is crucial for optimizing the treatment plan and allocating medical resources effectively.
Methods: This study proposes a novel ensemble learning-based EC survival prediction model. Firstly, recursive feature elimination (RFE) is used to determine the key feature subsets from the dataset. Based on the determined key features, the improved clustering by fast search and find of density peaks (IDPC) is proposed to construct a novel indicator related to EC survival risk. The cosine distance is introduced in IDPC to cluster samples with similar characteristics. Then, the adaptive synthetic (ADASYN) technique is used to generate more high-risk samples to balance high-risk and low-risk samples. Finally, the hyperparameters of the three models, including extreme gradient boosting (XGBoost), adaptive boosting (AdaBoost), and random forest (RF), are optimized by whale optimization algorithm (WOA) and a new stacking model is constructed to evaluate the survival risk of patients.
Results: The proposed stacking model achieved an area under the receiver operating characteristic curve (AUC) of 0.952 and Accuracy of 0.899, on the dataset from the First Affiliated Hospital of Zhengzhou University.
Conclusions: The survival prediction model the proposed ensemble learning system is much more accurate and convenient, providing a basis clinical judgment and decision making and improving the survival status of esophageal cancer patients.
期刊介绍:
IRBM is the journal of the AGBM (Alliance for engineering in Biology an Medicine / Alliance pour le génie biologique et médical) and the SFGBM (BioMedical Engineering French Society / Société française de génie biologique médical) and the AFIB (French Association of Biomedical Engineers / Association française des ingénieurs biomédicaux).
As a vehicle of information and knowledge in the field of biomedical technologies, IRBM is devoted to fundamental as well as clinical research. Biomedical engineering and use of new technologies are the cornerstones of IRBM, providing authors and users with the latest information. Its six issues per year propose reviews (state-of-the-art and current knowledge), original articles directed at fundamental research and articles focusing on biomedical engineering. All articles are submitted to peer reviewers acting as guarantors for IRBM''s scientific and medical content. The field covered by IRBM includes all the discipline of Biomedical engineering. Thereby, the type of papers published include those that cover the technological and methodological development in:
-Physiological and Biological Signal processing (EEG, MEG, ECG…)-
Medical Image processing-
Biomechanics-
Biomaterials-
Medical Physics-
Biophysics-
Physiological and Biological Sensors-
Information technologies in healthcare-
Disability research-
Computational physiology-
…