Machine learning models for chronic kidney disease diagnosis and prediction

被引:5
作者
Rahman, Md. Mustafizur [1 ,2 ]
Al-Amin, Md. [1 ]
Hossain, Jahangir [2 ]
机构
[1] Jashore Univ Sci & Technol, Dept Elect & Elect Engn, Jashore 7408, Bangladesh
[2] Univ Technol Sydney, Sch Elect & Data Engn, 15 Broadway, Sydney, NSW 2007, Australia
关键词
Chronic kidney disease; Ensemble machine learning; Borderline-SMOTE; MICE; Prediction; SMOTE;
D O I
10.1016/j.bspc.2023.105368
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Background and objective: Chronic kidney disease is a severe health problem that affects people all over the world, particularly in South Asia. Therefore, proper diagnosis and treatment are required as early as possible. The main goal of this study is to detect the presence or absence of CKD in the human body utilizing a variety of features grasped from a few medical tests. Methods: This paper has focused on eight ensemble learning methods for diagnosing CKD on the UCI machine learning datasets. The datasets have been fixed by imputing the missing values using the MICE imputation method and handling the imbalance properties using the borderline SVMSMOTE method to improve the performance of classifiers. Moreover, recursive feature elimination and the boruta method have been used to find the most significant features and reduce the compilation time, while the hyperparameter tuning technique was used to raise the performance of classifiers and get optimal solutions. Results: Taking the most significant feature into consideration, RFE outperformed boruta and selected only 50% of the total features. Moreover, various performance matrices are used to find the best competent classifiers for detecting CKD. LightGBM outperformed state-of-the-art and other ensemble methods with the lowest compilation time and highest accuracy. Based on experimental findings, the proposed method achieved the highest average of 99.75% accuracy, 99.40% precision, 99.41% recall, 99.61% F-measure and 99.57% AUC-ROC. Moreover, our proposed method rises the average detection rate by 5.64%, 1%, 2.04%, 8.63%,1.99%, 2.84%, 2.42% and 4.76%, respectively, in comparison with different approaches performing on the same dataset. Conclusion: Experiments show that our suggested method can identify CKD more precisely than the most recent methods.
引用
收藏
页数:17
相关论文
共 61 条
  • [1] Adamczyk Agnieszka K, 2024, AJOB Neurosci, V15, pW4, DOI 10.1080/21507740.2021.1938292
  • [3] Comprehensive Performance Assessment of Deep Learning Models in Early Prediction and Risk Identification of Chronic Kidney Disease
    Akter, Shamima
    Habib, Ahsan
    Islam, Md Ashiqul
    Hossen, Md Sagar
    Fahim, Wasik Ahmmed
    Sarkar, Puza Rani
    Ahmed, Manik
    [J]. IEEE ACCESS, 2021, 9 : 165184 - 165206
  • [4] Alasker H, 2017, 2017 3RD INTERNATIONAL CONFERENCE ON SCIENCE IN INFORMATION TECHNOLOGY (ICSITECH), P681, DOI 10.1109/ICSITech.2017.8257199
  • [5] Ensemble Feature Ranking for Cost-Based Non-Overlapping Groups: A Case Study of Chronic Kidney Disease Diagnosis in Developing Countries
    Ali, Syed Imran
    Bilal, Hafiz Syed Muhammad
    Hussain, Musarrat
    Hussain, Jamil
    Satti, Fahad Ahmed
    Hussain, Maqbool
    Park, Gwang Hoon
    Chung, Taechoong
    Lee, Sungyoung
    [J]. IEEE ACCESS, 2020, 8 (08): : 215623 - 215648
  • [6] Neural network and support vector machine for the prediction of chronic kidney disease: A comparative study
    Almansour, Njoud Abdullah
    Syed, Hajra Fahim
    Khayat, Nuha Radwan
    Altheeb, Rawan Kanaan
    Juri, Renad Emad
    Alhiyafi, Jamal
    Alrashed, Saleh
    Olatunji, Sunday O.
    [J]. COMPUTERS IN BIOLOGY AND MEDICINE, 2019, 109 : 101 - 111
  • [7] Almasoud M, 2019, INT J ADV COMPUT SC, V10, P89
  • [8] Ensemble of Deep Learning Based Clinical Decision Support System for Chronic Kidney Disease Diagnosis in Medical Internet of Things Environment
    Alsuhibany, Suliman A.
    Abdel-Khalek, Sayed
    Algarni, Ali
    Fayomi, Aisha
    Gupta, Deepak
    Kumar, Vinay
    Mansour, Romany F.
    [J]. COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2021, 2021
  • [9] Avci E, 2018, 2018 6TH INTERNATIONAL SYMPOSIUM ON DIGITAL FORENSIC AND SECURITY (ISDFS), P412
  • [10] Bergstra J, 2012, J MACH LEARN RES, V13, P281