Early-stage diagnosis of chronic kidney disease using majority vote - Grey Wolf optimization (MV-GWO)

被引:2
作者
Siddhartha, Manu [1 ]
Kumar, Vaibhav [2 ]
Nath, Rajendra [3 ]
机构
[1] Liverpool John Moores Univ, Comp Sci & Math, Liverpool, Merseyside, England
[2] REVA Acad Corp Excellence, Bangalore, Karnataka, India
[3] King Georges Med Univ, Dept Pharmacol, Lucknow, Uttar Pradesh, India
关键词
Chronic kidney disease; Machine learning; Feature selection; Healthcare analytics; Model interpretation; Grey wolf optimization; FEATURE-SELECTION;
D O I
10.1007/s12553-021-00617-8
中图分类号
R-058 [];
学科分类号
摘要
Chronic Kidney Disease (CKD) is a serious health issue that is growing at an alarming rate. A person having CKD can only be saved by providing kidney transplantation or dialysis but as it involves huge medical expenditure it remains unaffordable to low and middle-income countries. Therefore, early-stage diagnosis is critical to contain CKD at the initial stage, which can reduce the mortality rate and bring down the cost of treatment significantly. This study proposes a novel feature selection method namely Majority Vote-Grey Wolf Optimization (MV-GWO) with machine learning algorithms for the detection of CKD using benchmark CKD dataset. For validating the proposed method we have used different evaluation metrics. The proposed method achieved higher accuracy of 98.75%, sensitivity of 100%, F1-score of 99% and Mathew Correlation Coefficient of 0.9735 with only five optimal features. Furthermore, we have interpreted the machine learning model by analyzing the most critical features using SHapley Additive exPlanations (SHAP) feature importance plot, summary plot, dependence plot, and Partial dependence plots.
引用
收藏
页码:117 / 136
页数:20
相关论文
共 51 条
  • [41] Srivastava S., 2013, International Journal of Engineering Research and Development, P57
  • [42] Swinscow TDV, 1997, STAT SQUARE ONE
  • [43] Feature selection methods and genomic big data: a systematic review
    Tadist, Khawla
    Najah, Said
    Nikolov, Nikola S.
    Mrabti, Fatiha
    Zahi, Azeddine
    [J]. JOURNAL OF BIG DATA, 2019, 6 (01)
  • [44] Tu Chengsheng, 2017, MATEC Web of Conferences, V139, DOI 10.1051/matecconf/201713900222
  • [45] A Review of Feature Selection and Its Methods
    Venkatesh, B.
    Anuradha, J.
    [J]. CYBERNETICS AND INFORMATION TECHNOLOGIES, 2019, 19 (01) : 3 - 26
  • [46] An Improved Grey Wolf Optimizer Based on Differential Evolution and Elimination Mechanism
    Wang, Jie-Sheng
    Li, Shu-Xia
    [J]. SCIENTIFIC REPORTS, 2019, 9 (1)
  • [47] Prediction of Type 2 Diabetes Risk and Its Effect Evaluation Based on the XGBoost Model
    Wang, Liyang
    Wang, Xiaoya
    Chen, Angxuan
    Jin, Xian
    Che, Huilian
    [J]. HEALTHCARE, 2020, 8 (03)
  • [48] Clinical risk prediction with random forests for survival, longitudinal, and multivariate (RF-SLAM) data analysis
    Wongvibulsin, Shannon
    Wu, Katherine C.
    Zeger, Scott L.
    [J]. BMC MEDICAL RESEARCH METHODOLOGY, 2019, 20 (01)
  • [49] Integrating Correlation-Based Feature Selection and Clustering for Improved Cardiovascular Disease Diagnosis
    Wosiak, Agnieszka
    Zakrzewska, Danuta
    [J]. COMPLEXITY, 2018,
  • [50] Prediction of Hemodialysis Timing Based on LVW Feature Selection and Ensemble Learning
    Xiong, Chang-zhu
    Su, Minglian
    Jiang, Zitao
    Jiang, Wei
    [J]. JOURNAL OF MEDICAL SYSTEMS, 2019, 43 (01)