Adaptive Multi-Branching for Shallow Decision Tree Induction
This work proposes the Multi-Branch Neural Decision Tree with Adaptive Pruning (MBNDT), a single axis-aligned tree trained end-to-end with differentiable multi-way splits that achieves the best average rank and mean balanced accuracy among depth-constrained single-tree baselines.
H. Park, Jeonghoon Choi, Juseong Kim et al.
· 0 citations