A novel evolutionary ensemble prediction model using harmony search and stacking for diabetes diagnosis

被引:4
|
作者
Zhang, Zaiheng [1 ]
Lu, Yanjie [2 ]
Ye, Mingtao [2 ]
Huang, Wanyu [1 ]
Jin, Lixu [3 ]
Zhang, Guodao [2 ,4 ,5 ]
Ge, Yisu [6 ]
Baghban, Alireza [7 ]
Zhang, Qiwen [2 ]
Wang, Haiou [3 ]
Zhu, Wenzong [8 ,9 ]
机构
[1] Zhejiang Chinese Med Univ, Sch Clin Med 3, Hangzhou 310053, Peoples R China
[2] Hangzhou Dianzi Univ, Inst Intelligent Media Comp, Hangzhou 310018, Peoples R China
[3] Wenzhou Med Univ, Affiliated Hosp 1, Dept Obstet, Wenzhou 325000, Peoples R China
[4] Tianjin Univ Technol, Minist Educ, Key Lab Comp Vis & Syst, Tianjin 300384, Peoples R China
[5] Ocean Univ China, Coll Engn, Qingdao 266100, Peoples R China
[6] Wenzhou Univ, Zhejiang Key Lab Intelligent Informat Safety & Eme, Wenzhou 325035, Peoples R China
[7] NISOC Co, Proc Engn Dept, Ahvaz, Iran
[8] Wenzhou Hosp Integrated Tradit Chinese & Western M, Wenzhou 325000, Peoples R China
[9] Zhejiang Chinese Med Univ, Wenzhou TCM Hosp, Wenzhou Hosp Tradit Chinese Med, Wenzhou 325000, Peoples R China
关键词
Diabetes diagnosis; Ensemble learning; Stacking; Feature selection; Harmony search; Combination optimization; MACHINE LEARNING ALGORITHMS; ANT COLONY OPTIMIZATION; FUTURE-DEVELOPMENT; TYPE-2; OBESITY; RISK; COMPLICATIONS; ASSOCIATION; POPULATION; FITNESS;
D O I
10.1016/j.jksuci.2023.101873
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Diabetes is a dreaded disease that can be identified by elevated blood glucose levels in the blood, and undiagnosed diabetes can cause a host of related complications, such as retinopathy and nephropathy. In terms of type, the main categories are type 1 diabetes (T1DM), type 2 diabetes (T2DM) and gestational diabetes mellitus (GDM). Machine learning models and metaheuristic optimization algorithms can play an important role in the early detection, diagnosis and treatment of this disease. To this end, we propose AHDHSStacking, an ensemble learning framework for diabetes mellitus classification and diagnosis that is based on the harmony search (HS) algorithm and stacking and includes two stages of feature selection and optimization of base-learner combinations. To improve the model's overall performance, the average performance of all base learners is used as the feature selection target, and an adaptive hyperparameter strategy is used to accelerate the iterative process. HS is then used to optimize to find the best combination of base learners, which improves model performance while reducing complexity. Following that, we conducted experiments on the Pima Indians Diabetes (PID) dataset and the Chinese and Western Medicine Diabetes (CWMD) dataset, achieving accuracy of 93.09%, precision of 93.22%, recall of 91.60% , F-measure of 92.25%, and MCC of 84.79% on PID dataset, which is better than all benchmark models and validated the model's validity. CWMD dataset experimental results showed that AHDHS-Stacking screened for key features such as age, gender, urinary glucose, fasting glucose, BMI and cholesterol, and can be used as a practical and accurate method for early diabetes prediction.
引用
收藏
页数:20
相关论文
共 50 条
  • [21] Rumen Fermentation Parameters Prediction Model for Dairy Cows Using a Stacking Ensemble Learning Method
    Wang, Yuxuan
    Zhou, Jianzhao
    Wang, Xinjie
    Yu, Qingyuan
    Sun, Yukun
    Li, Yang
    Zhang, Yonggen
    Shen, Weizheng
    Wei, Xiaoli
    ANIMALS, 2023, 13 (04):
  • [22] Blended Ensemble Learning Prediction Model for Strengthening Diagnosis and Treatment of Chronic Diabetes Disease
    Mahesh, T. R.
    Kumar, Dhilip
    Kumar, V. Vinoth
    Asghar, Junaid
    Bazezew, Banchigize Mekcha
    Natarajan, Rajesh
    Vivek, V.
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
  • [23] Blended Ensemble Learning Prediction Model for Strengthening Diagnosis and Treatment of Chronic Diabetes Disease
    Mahesh, T. R.
    Kumar, Dhilip
    Kumar, V. Vinoth
    Asghar, Junaid
    Bazezew, Banchigize Mekcha
    Natarajan, Rajesh
    Vivek, V.
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
  • [24] Explainable stacking ensemble with feature tokenizer transformers for men's diabetes prediction
    Tran, Vinh Quang
    Choi, Younsung
    Byeon, Haewon
    JOURNAL OF MENS HEALTH, 2024, 20 (11) : 38 - 56
  • [25] Optimizing urban traffic light scheduling problem using harmony search with ensemble of local search
    Gao, Kaizhou
    Zhang, Yicheng
    Sadollah, Ali
    Su, Rong
    APPLIED SOFT COMPUTING, 2016, 48 : 359 - 372
  • [26] Fake news detection using an ensemble learning model based on Self-Adaptive Harmony Search algorithms
    Huang, Yin-Fu
    Chen, Po-Hong
    EXPERT SYSTEMS WITH APPLICATIONS, 2020, 159
  • [27] AIOPS Prediction for Hard Drive Failures Based on Stacking Ensemble Model
    Wang, Haifeng
    Zhang, Haili
    2020 10TH ANNUAL COMPUTING AND COMMUNICATION WORKSHOP AND CONFERENCE (CCWC), 2020, : 417 - 423
  • [28] Brain stroke prediction model based on boosting and stacking ensemble approach
    Mondal S.
    Ghosh S.
    Nag A.
    International Journal of Information Technology, 2024, 16 (1) : 437 - 446
  • [29] Multi-model Stacking Ensemble Learning for Student Achievement Prediction
    Fang, Tao
    Huang, Sirui
    Zhou, Ya
    Zhang, Huibing
    PAAP 2021: 2021 12TH INTERNATIONAL SYMPOSIUM ON PARALLEL ARCHITECTURES, ALGORITHMS AND PROGRAMMING, 2021, : 136 - 140
  • [30] Prediction of gaseous nitrous acid based on Stacking ensemble learning model
    Tang, Ke
    Qin, Min
    Zhao, Xing
    Duan, Jun
    Fang, Wu
    Liang, Shuai-Xi
    Meng, Fan-Hao
    Ye, Kai-Di
    Zhang, He-Lu
    Xie, Pin-Hua
    Zhongguo Huanjing Kexue/China Environmental Science, 2020, 40 (02): : 582 - 590