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 条
  • [31] A stacking ensemble machine learning model for improving monthly runoff prediction
    Wang, Wen-chuan
    Gu, Miao
    Li, Zong
    Hong, Yang-hao
    Zang, Hong-fei
    Xu, Dong-mei
    EARTH SCIENCE INFORMATICS, 2025, 18 (01)
  • [32] Model Ensemble for Click Prediction in Bing Search Ads
    Ling, Xiaoliang
    Deng, Weiwei
    Gu, Chen
    Zhou, Hucheng
    Li, Cui
    Sun, Feng
    WWW'17 COMPANION: PROCEEDINGS OF THE 26TH INTERNATIONAL CONFERENCE ON WORLD WIDE WEB, 2017, : 689 - 698
  • [33] A novel voting ensemble model for spatial prediction of landslides using GIS
    Sachdeva, Shruti
    Bhatia, Tarunpreet
    Verma, Anil K.
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2020, 41 (03) : 929 - 952
  • [34] A Fruit Tree Disease Diagnosis Model Based on Stacking Ensemble Learning
    Li, Honglei
    Jin, Ying
    Zhong, Jiliang
    Zhao, Ruixue
    COMPLEXITY, 2021, 2021
  • [35] A novel multistage ensemble approach for prediction and classification of diabetes
    Simaiya, Sarita
    Kaur, Rajwinder
    Sandhu, Jasminder Kaur
    Alsafyani, Majed
    Alroobaea, Roobaea
    Alsekait, Deema mohammed
    Margala, Martin
    Chakrabarti, Prasun
    FRONTIERS IN PHYSIOLOGY, 2022, 13
  • [36] Diabetes Prediction Using Ensemble Perceptron Algorithm
    Mirshahvalad, Roxana
    Zanjani, Nastaran Asadi
    2017 9TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL INTELLIGENCE AND COMMUNICATION NETWORKS (CICN), 2017, : 190 - 194
  • [37] Research and application of a novel weight-based evolutionary ensemble model using principal component analysis for wind power prediction
    Zhang, Chu
    Tao, Zihan
    Xiong, Jinlin
    Qian, Shijie
    Fu, Yongyan
    Ji, Jie
    Nazir, Muhammad Shahzad
    Peng, Tian
    RENEWABLE ENERGY, 2024, 232
  • [38] A novel stacking-based ensemble learning model for drilling efficiency prediction in earth-rock excavation
    Lv, Fei
    Yu, Jia
    Zhang, Jun
    Yu, Peng
    Tong, Da-Wei
    Wu, Bin-Ping
    JOURNAL OF ZHEJIANG UNIVERSITY-SCIENCE A, 2022, 23 (12): : 1027 - 1046
  • [39] Research and application of a novel selective stacking ensemble model based on error compensation and parameter optimization for AQI prediction
    Peng, Tian
    Xiong, Jinlin
    Sun, Kai
    Qian, Shijie
    Tao, Zihan
    Nazir, Muhammad Shahzad
    Zhang, Chu
    ENVIRONMENTAL RESEARCH, 2024, 247
  • [40] An evolutionary ensemble model based on GA for epidemic transmission prediction
    Li, Xiaoning
    Yu, Qiancheng
    Yang, Yufan
    Tang, Chen
    Wang, Jinyun
    JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2023, 44 (05) : 7469 - 7481