Abstract:Decision trees are popular classification models, providing high accuracy and intuitive explanations. However, as the tree size grows the model interpretability deteriorates. Traditional tree-induction algorithms, such as C4.5 and CART, rely on impurity-reduction functions that promote the discriminative power of each split. Thus, although these traditional methods are accurate in practice, there has been no theoretical guarantee that they will produce small trees. In this paper, we justify the use of a general family of impurity functions, including the popular functions of entropy and Gini-index, in scenarios where small trees are desirable, by showing that a simple enhancement can equip them with complexity guarantees. We consider a general setting, where objects to be classified are drawn from an arbitrary probability distribution, classification can be binary or multi-class, and splitting tests are associated with non-uniform costs. As a measure of tree complexity, we adopt the expected cost to classify an object drawn from the input distribution, which, in the uniform-cost case, is the expected number of tests. We propose a tree-induction algorithm that gives a logarithmic approximation guarantee on the tree complexity. This approximation factor is tight up to a constant factor under mild assumptions. The algorithm recursively selects a test that maximizes a greedy criterion defined as a weighted sum of three components. The first two components encourage the selection of tests that improve the balance and the cost-efficiency of the tree, respectively, while the third impurity-reduction component encourages the selection of more discriminative tests. As shown in our empirical evaluation, compared to the original heuristics, the enhanced algorithms strike an excellent balance between predictive accuracy and tree complexity.

An Improved Algorithm of Decision Trees by Using the Convex Function

Rough set based decision tree

Application of variable precision rough set model in building decision tree

Regularized impurity reduction: Accurate decision trees with complexity guarantees

Improved Information Gain Estimates for Decision Tree Induction

A Review: the Effects of Imperfect Data on Incremental Decision Tree

Improving Decision Trees by Tsallis Entropy Information Metric Method.

An improved decision tree algorithm based on boundary mixed attribute dependency

Efficient non-greedy optimization of decision trees

A Communication-Efficient Parallel Algorithm for Decision Tree

Heuristic Mode Research and Application of Decision Tree Algorithm

The research and realization of decision trees based on credibility

Taiga: Performance Optimization of the C4.5 Decision Tree Construction Algorithm

A New Method for Learning Decision Tree Classifier

Optimal decision trees for the algorithm selection problem: integer programming based approaches

On the Optimality of Trees Generated by ID3

The Description of Optimal Decision Tree Algorithm and Its Application in Customer Consumption Behavior

Mixed-Integer Convex Nonlinear Optimization with Gradient-Boosted Trees Embedded

Heuristic Model and Application of Decision Tree Based on Quasi-Linear Criterion

Induction of a Novel Hybrid Decision Forest Model Based on Information Theory.

Improved differentiation classification of variable precision artificial intelligence higher education management