Using k-dependence causal forest to mine the most significant dependency relationships among clinical variables for thyroid disease diagnosis

PLoS One. 2017 Aug 17;12(8):e0182070. doi: 10.1371/journal.pone.0182070. eCollection 2017.

Abstract

Numerous data mining models have been proposed to construct computer-aided medical expert systems. Bayesian network classifiers (BNCs) are more distinct and understandable than other models. To graphically describe the dependency relationships among clinical variables for thyroid disease diagnosis and ensure the rationality of the diagnosis results, the proposed k-dependence causal forest (KCF) model generates a series of submodels in the framework of maximum spanning tree (MST) and demonstrates stronger dependence representation. Friedman test on 12 UCI datasets shows that KCF has classification accuracy advantage over the other state-of-the-art BNCs, such as Naive Bayes, tree augmented Naive Bayes, and k-dependence Bayesian classifier. Our extensive experimental comparison on 4 medical datasets also proves the feasibility and effectiveness of KCF in terms of sensitivity and specificity.

MeSH terms

  • Algorithms
  • Bayes Theorem
  • Data Mining*
  • Datasets as Topic
  • Diagnosis, Computer-Assisted / methods*
  • Humans
  • Markov Chains
  • Thyroid Diseases / diagnosis*

Grants and funding

This work was supported by the National Science Foundation of China (Grant No. 61272209) and the Agreement of Science & Technology Development Project, Jilin Province (No. 20150101014JC). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.