Expected-depth selection for interpretable decision trees
摘要
Interpretable decision trees are usually selected by worst-case structural measures such as maximum depth or node count, although many deployments incur cost along the particular root-to-leaf path followed by each case. We study expected depth, the test-sample average number of internal decision tests, as a direct measure of average inference burden. The proposed soft-depth (Soft-D) selector chooses, from a cost-complexity pruning candidate path, the tree with minimum validation expected depth among candidates whose validation accuracy remains within a user-specified tolerance of the best candidate. Across eleven benchmark datasets and 25 outer test folds per dataset, Soft-D reduces expected depth by 47.4% relative to unpruned CART and by 12.1% relative to accuracy-first CCP at