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 条
  • [1] Alam Md Zahangir, 2019, Informatics in Medicine Unlocked, V15, P93, DOI 10.1016/j.imu.2019.100180
  • [2] Multiple imputation by predictive mean matching in cluster-randomized trials
    Bailey, Brittney E.
    Andridge, Rebecca
    Shoben, Abigail B.
    [J]. BMC MEDICAL RESEARCH METHODOLOGY, 2020, 20 (01)
  • [3] Global, regional, and national burden of chronic kidney disease, 1990-2017: a systematic analysis for the Global Burden of Disease Study 2017
    Bikbov, Boris
    Purcell, Carrie
    Levey, Andrew S.
    Smith, Mari
    Abdoli, Amir
    Abebe, Molla
    Adebayo, Oladimeji M.
    Afarideh, Mohsen
    Agarwal, Sanjay Kumar
    Agudelo-Botero, Marcela
    Ahmadian, Elham
    Al-Aly, Ziyad
    Alipour, Vahid
    Almasi-Hashiani, Amir
    Al-Raddadi, Rajaa M.
    Alvis-Guzman, Nelson
    Amini, Saeed
    Andrei, Tudorel
    Andrei, Catalina Liliana
    Andualem, Zewudu
    Anjomshoa, Mina
    Arabloo, Jalal
    Ashagre, Alebachew Fasil
    Asmelash, Daniel
    Ataro, Zerihun
    Atout, Maha Moh'd Wahbi
    Ayanore, Martin Amogre
    Badawi, Alaa
    Bakhtiari, Ahad
    Ballew, Shoshana H.
    Balouchi, Abbas
    Banach, Maciej
    Barquera, Simon
    Basu, Sanjay
    Bayih, Mulat Tirfie
    Bedi, Neeraj
    Bello, Aminu K.
    Bensenor, Isabela M.
    Bijani, Ali
    Boloor, Archith
    Borzi, Antonio M.
    Camera, Luis Alberto
    Carrero, Juan J.
    Carvalho, Felix
    Castro, Franz
    Catala-Lopez, Ferran
    Chang, Alex R.
    Chin, Ken Lee
    Chung, Sheng-Chia
    Cirillo, Massimo
    [J]. LANCET, 2020, 395 (10225) : 709 - 733
  • [4] Benchmark for filter methods for feature selection in high-dimensional classification data
    Bommert, Andrea
    Sun, Xudong
    Bischl, Bernd
    Rahnenfuehrer, Joerg
    Lang, Michel
    [J]. COMPUTATIONAL STATISTICS & DATA ANALYSIS, 2020, 143
  • [5] Random forests
    Breiman, L
    [J]. MACHINE LEARNING, 2001, 45 (01) : 5 - 32
  • [6] A Machine-Learning-Based Prediction Method for Hypertension Outcomes Based on Medical Data
    Chang, Wenbing
    Liu, Yinglai
    Xiao, Yiyong
    Yuan, Xinglong
    Xu, Xingxing
    Zhang, Siyue
    Zhou, Shenghan
    [J]. DIAGNOSTICS, 2019, 9 (04)
  • [7] XGBoost: A Scalable Tree Boosting System
    Chen, Tianqi
    Guestrin, Carlos
    [J]. KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, : 785 - 794
  • [8] Chimwayi KB, 2017, International Journal of Multimedia and Ubiquitous Engineering, V12, P23, DOI [DOI 10.14257/IJMUE.2017.12.8.03, 10.14257/ijmue.2017.12.8.03]
  • [9] Chugh A., 2018, ML CHI SQUARE TEST F
  • [10] Intelligent Diagnostic Prediction and Classification System for Chronic Kidney Disease
    Elhoseny, Mohamed
    Shankar, K.
    Uthayakumar, J.
    [J]. SCIENTIFIC REPORTS, 2019, 9 (1)