XGBoost Model for Chronic Kidney Disease Diagnosis

By Adeola Ogunleye, Qing-Guo Wang. IEEE/ACM Transactions on Computational Biology & Bioinformatics, 2020

615 citations recorded by the source corpus.

Chronic Kidney Disease (CKD) is a menace that is affecting 10 percent of the world population and 15 percent of the South African population. The early and cheap diagnosis of this disease with accuracy and reliability will save 20,000 lives in South Africa per year. Scientists are developing smart solutions with Artificial Intelligence (AI). In this paper, several typical and recent AI algorithms are studied in the context of CKD and the extreme gradient boosting (XGBoost) is chosen as our base model for its high performance. Then, the model is optimized and the optimal full model trained on all the features achieves a testing accuracy, sensitivity, and specificity of 1.000, 1.000, and 1.000, respectively. Note that, to cover the widest range of people, the time and monetary costs of CKD diagnosis have to be minimized with fewest patient tests. Thus, the reduced model using fewer features is desirable while it should still maintain high performance. To this end, the set-theory based rule is presented which combines a few feature selection methods with their collective strengths. The reduced model using about a half of the original full features performs better than the models based on individual feature selection methods and achieves accuracy, sensitivity and specificity, of 1.000, 1.000, and 1.000, respectively.

Subfields: Artificial Intelligence, Health Information Management

View Paper: https://www.semanticscholar.org/paper/6178542e6905055c5ac7a3e9f2a3d0a3641e4124

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Rohitash Chandra
  2. Azal Ahmad Khan
  3. Omkar Chaudhari

Citations in the corpus