Abstract
A decision tree (DT) is one of the most popular and efficient techniques in data mining. Specifically, in the clinical domain, DTs have been widely used thanks to their relatively easy explainable nature, efficient computation time, and relatively accurate predictions. However, some DT constriction algorithms may produce a large tree size structure which is difficult to understand and often leads to misclassification of data in the testing process due to poor generalization. Post pruning (PP) algorithms have been introduced to reduce the size of the tree structure with a minor (or not at all) decrease in the accuracy of classification while trying to improve the model's generalization. In this paper, we propose a new Boolean satisfiability (SAT) based PP algorithm called SAT-PP. Our algorithm reduces the tree size while preserving the accuracy of the unpruned tree. We implemented our algorithm on a medical-related classification data sets since in medical-related tasks we emphatically try to avoid decreasing the model's performance when better training is not an option. Namely, in the case of medical-related tasks, one may prefer an unpruned DT model to a pruned DT model with worse performance. Indeed, we empirically obtained that the SAT-PP algorithm produce the same accuracy and F1 score as the DT model without PP while statistically significantly reducing the model size and as a result computation time (6.8%). In addition, we compared the proposed algorithm with other PP algorithms and found similar generalization capabilities.
Original language | English |
---|---|
Article number | 102173 |
Journal | Data and Knowledge Engineering |
Volume | 145 |
DOIs | |
State | Published - May 2023 |
Externally published | Yes |
Bibliographical note
Funding Information:The authors wish to thank Galina Fudim for her help in analyzing the experiment's results and Sicco Verwer for his valuable insights. No funds, grants, or other support was received.
Publisher Copyright:
© 2023 Elsevier B.V.
Keywords
- Decision tree clinical data
- Random forest pruning
- SAT based pruning