Enhancing the Early Detection of Chronic Kidney Disease: A Robust Machine Learning Model

被引:27
作者
Arif, Muhammad Shoaib [1 ,2 ]
Mukheimer, Aiman [1 ]
Asif, Daniyal [3 ]
机构
[1] Prince Sultan Univ, Coll Humanities & Sci, Dept Math & Sci, Riyadh 11586, Saudi Arabia
[2] Air Univ, Dept Math, PAF Complex E9, Islamabad 44000, Pakistan
[3] COMSATS Univ Islamabad, Dept Math, Pk Rd, Islamabad 45550, Pakistan
关键词
chronic kidney disease; machine learning; artificial intelligence; data science; healthcare; bioinformatics; big data; PREDICTION;
D O I
10.3390/bdcc7030144
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Clinical decision-making in chronic disorder prognosis is often hampered by high variance, leading to uncertainty and negative outcomes, especially in cases such as chronic kidney disease (CKD). Machine learning (ML) techniques have emerged as valuable tools for reducing randomness and enhancing clinical decision-making. However, conventional methods for CKD detection often lack accuracy due to their reliance on limited sets of biological attributes. This research proposes a novel ML model for predicting CKD, incorporating various preprocessing steps, feature selection, a hyperparameter optimization technique, and ML algorithms. To address challenges in medical datasets, we employ iterative imputation for missing values and a novel sequential approach for data scaling, combining robust scaling, z-standardization, and min-max scaling. Feature selection is performed using the Boruta algorithm, and the model is developed using ML algorithms. The proposed model was validated on the UCI CKD dataset, achieving outstanding performance with 100% accuracy. Our approach, combining innovative preprocessing steps, the Boruta feature selection, and the k-nearest neighbors algorithm, along with a hyperparameter optimization using grid-search cross-validation (CV), demonstrates its effectiveness in enhancing the early detection of CKD. This research highlights the potential of ML techniques in improving clinical support systems and reducing the impact of uncertainty in chronic disorder prognosis.
引用
收藏
页数:16
相关论文
共 46 条
[1]  
Agrawal T., 2021, Hyperparameter optimization in machine learning: make your machine learning and deep learning models more efficient: Place of publication not identified, DOI [DOI 10.1007/978-1-4842-6579-6_2, 10.1007/978-1-4842-6579-6, DOI 10.1007/978-1-4842-6579-62]
[2]   Enhanced Credit Card Fraud Detection Model Using Machine Learning [J].
Alfaiz, Noor Saleh ;
Fati, Suliman Mohamed .
ELECTRONICS, 2022, 11 (04)
[3]  
Almustafa Khaled Mohamad, 2021, Informatics in Medicine Unlocked, V24, DOI 10.1016/j.imu.2021.100631
[4]   Enhancing Heart Disease Prediction through Ensemble Learning Techniques with Hyperparameter Optimization [J].
Asif, Daniyal ;
Bibi, Mairaj ;
Arif, Muhammad Shoaib ;
Mukheimer, Aiman .
ALGORITHMS, 2023, 16 (06)
[5]   Chronic Kidney Disease Diagnosis and Management: A Review [J].
Chen, Teresa K. ;
Knicely, Daphne H. ;
Grams, Morgan E. .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2019, 322 (13) :1294-1304
[6]   Prediction of Chronic Kidney Disease-A Machine Learning Perspective [J].
Chittora, Pankaj ;
Chaurasia, Sandeep ;
Chakrabarti, Prasun ;
Kumawat, Gaurav ;
Chakrabarti, Tulika ;
Leonowicz, Zbigniew ;
Jasinski, Michal ;
Jasinski, Lukasz ;
Gono, Radomir ;
Jasinska, Elzbieta ;
Bolshev, Vadim .
IEEE ACCESS, 2021, 9 :17312-17334
[7]   k-Nearest Neighbour Classifiers - A Tutorial [J].
Cunningham, Padraig ;
Delany, Sarah Jane .
ACM COMPUTING SURVEYS, 2021, 54 (06)
[8]   Principled missing data methods for researchers [J].
Dong, Yiran ;
Peng, Chao-Ying Joanne .
SPRINGERPLUS, 2013, 2 :1-17
[9]  
Ekanayake IU, 2020, MERCON 2020: 6TH INTERNATIONAL MULTIDISCIPLINARY MORATUWA ENGINEERING RESEARCH CONFERENCE (MERCON), P260, DOI [10.1109/mercon50084.2020.9185249, 10.1109/MERCon50084.2020.9185249]
[10]   Predicting Chronic Kidney Disease Using Machine Learning Algorithms [J].
Farjana, Afia ;
Liza, Fatema Tabassum ;
Pandit, Parth Pratim ;
Das, Madhab Chandra ;
Hasan, Mahadi ;
Tabassum, Fariha ;
Hossen, Md. Helal .
2023 IEEE 13TH ANNUAL COMPUTING AND COMMUNICATION WORKSHOP AND CONFERENCE, CCWC, 2023, :1267-1271