Cost-sensitive ensemble feature ranking and automatic threshold selection for chronic kidney disease diagnosis

Syed Imran Ali, Bilal Ali, Jamil Hussain, Musarrat Hussain, Fahad Ahmed Satti, Gwang Hoon Park, Sungyoung Lee

Research output: Contribution to journalArticlepeer-review

5 Citations (Scopus)

Abstract

Automated medical diagnosis is one of the important machine learning applications in the domain of healthcare. In this regard, most of the approaches primarily focus on optimizing the accuracy of classification models. In this research, we argue that, unlike general-purpose classification problems, medical applications, such as chronic kidney disease (CKD) diagnosis, require special treatment. In the case of CKD, apart from model performance, other factors such as the cost of data acquisition may also be taken into account to enhance the applicability of the automated diagnosis system. In this research, we proposed two techniques for cost-sensitive feature ranking. An ensemble of decision tree models is employed in both the techniques for computing the worth of a feature in the CKD dataset. An automatic threshold selection heuristic is also introduced which is based on the intersection of features' worth and their accumulated cost. A set of experiments are conducted to evaluate the efficacy of the proposed techniques on both tree-based and non tree-based classification models. The proposed approaches were also evaluated against several comparative techniques. Furthermore, it is demonstrated that the proposed techniques select around 1/4th of the original CKD features while reducing the cost by a factor of 7.42 of the original feature set. Based on the extensive experimentation, it is concluded that the proposed techniques employing feature-cost interaction heuristic tend to select feature subsets that are both useful and cost-effective.

Original languageEnglish
Article number5663
JournalApplied Sciences (Switzerland)
Volume10
Issue number16
DOIs
Publication statusPublished - Aug 2020

Keywords

  • Chronic kidney disease
  • Cost-sensitive feature selection
  • Decision tree classifiers
  • Ensemble models
  • Gradient boosted trees
  • Random forest

Fingerprint

Dive into the research topics of 'Cost-sensitive ensemble feature ranking and automatic threshold selection for chronic kidney disease diagnosis'. Together they form a unique fingerprint.

Cite this