Improving the Intrinsic Interpretability of Differentiable Decision Trees through Euclidean Feature-Concentration Regularisation
Abstract
Differentiable decision trees (DDTs) support end-to-end optimisation and explicit decision paths, but dense oblique splits can obscure the role of individual input features. We propose the Euclidean-Concentrated Differentiable Decision Tree (EU-DDT), which regularises the relative absolute feature weights at each internal node using their normalised squared Euclidean distance from the uniform distribution. This encourages dominance by a small number of features while retaining probabilistic soft routing. Across five matched random seeds, EU-DDT reduces the mean effective number of features from 8.94 to 1.54 on Wine and from 45.28 to 9.06 on Digits. Mean test accuracy changes from 100.0% to 93.9% on Wine and from 92.3% to 94.6% on Digits, illustrating the dataset-dependent trade-off between concentration and predictive performance. We also use EU-DDT to parameterise a PPO actor with an MLP critic. Two random-seed runs per environment show a consistent qualitative increase in concentration while preserving effective task performance. In representative CartPole-v1 and MiniPacman runs, EU-DDT reaches 100% final evaluation success, while mean node concentration increases from 0.1272 to 0.7179 and from 0.0437 to 0.7237, respectively. These results support EU regularisation as a mechanism for reducing node-level feature complexity; the reinforcement-learning evidence remains preliminary because only two seeds are evaluated.
Figures & tables
| Parameter | Setting |
|---|---|
| Comparison models | DDT / L1-DDT / L2-DDT / EU-DDT / Neural Network |
| Input scaling | Wine: standardisation; Digits: MinMax scaling to |
| Tree depth | Wine: 3; Digits: 6 |
| Internal nodes / leaves | Wine: 7 / 8; Digits: 63 / 64 |
| Optimizer | Adam |
| Batching | Full batch |
| Dataset | Model | Maximum weight share | Top-1/Top-2 margin | Effective features | Relative sparsity | Mean active-node |
|---|---|---|---|---|---|---|
| Wine | DDT | 0.1797 | 0.0296 | 8.94 | 0.0681 | 0.0403 |
| Wine | DDT+L1 | 0.3272 | 0.0985 | 5.23 | 0.4374 | 0.1668 |
| Wine | DDT+L2 | 0.1894 | 0.0334 | 8.49 | 0.0835 | 0.0468 |
| Wine | EU-DDT | 0.7804 | 0.5906 | 1.54 | 0.8659 | 0.6962 |
| Digits | DDT | 0.0381 | 0.0022 | 45.28 | 0.0726 | 0.0069 |
| Digits | DDT+L1 | 0.1175 | 0.0360 | 30.13 | 0.3386 | 0.0554 |
| Dataset | DDT | DDT+L1 | DDT+L2 | EU-DDT | Neural Network |
|---|---|---|---|---|---|
| Wine five-seed mean accuracy | 1.000 | 1.000 | 1.000 | 0.939 | 1.000 |
| Digits five-seed mean accuracy | 0.923 | 0.949 | 0.960 | 0.946 | 0.969 |
| Parameter | CartPole-v1 | MiniPacman |
|---|---|---|
| Representative seed | 3 | 1 |
| State dimension | 4 | 14 |
| Action dimension | 2 | 4 |
| Tree depth | 3 | 4 |
| Internal nodes / leaves | 7 / 8 | 15 / 16 |
| PPO updates | 60 | 60 |
| CartPole-v1 | ||||
|---|---|---|---|---|
| Metric | DDT | L1 | L2 | EU |
| Final success rate | 100% | 90% | 46.7% | 100% |
| Concentration | 0.1272 | 0.3153 | 0.0798 | 0.7179 |
| Maximum share | 0.4661 | 0.6031 | 0.4114 | 0.8220 |
| Top-1/Top-2 margin | 0.1648 | 0.3562 | 0.1272 | 0.7236 |
| k90 | 3.29 | 2.86 | 3.71 | 1.71 |
Interpretable ML
Interpretable machine learning develops models and explanations that make predictions, evidence, and decision processes understandable to people.
See all 310 papers