Distilling a Neural Network Into a Soft Decision Tree
read the original abstract
Deep neural networks have proved to be a very effective way to perform classification tasks. They excel when the input data is high dimensional, the relationship between the input and the output is complicated, and the number of labeled training examples is large. But it is hard to explain why a learned network makes a particular classification decision on a particular test case. This is due to their reliance on distributed hierarchical representations. If we could take the knowledge acquired by the neural net and express the same knowledge in a model that relies on hierarchical decisions instead, explaining a particular decision would be much easier. We describe a way of using a trained neural net to create a type of soft decision tree that generalizes better than one learned directly from the training data.
This paper has not been read by Pith yet.
Forward citations
Cited by 14 Pith papers
-
TILT: Target-induced loss tilting under covariate shift
TILT adds a target-data penalty on an auxiliary predictor component to induce effective importance weighting for unsupervised domain adaptation under covariate shift.
-
Minimax Rates and Spectral Distillation for Tree Ensembles
Spectral analysis of tree ensembles produces minimax rates for random forests governed by kernel eigenvalue decay and enables distillation of RFs and GBMs into compact models via leading eigenfunctions and singular vectors.
-
Approximation-Free Differentiable Oblique Decision Trees
DTSemNet gives an exact, invertible neural-network encoding of hard oblique decision trees that supports direct gradient training for both classification and regression without probabilistic softening or quantized estimators.
-
Ternary Decision Trees with Locally-Adaptive Uncertainty Zones
Ternary decision trees with locally-adaptive uncertainty zones estimated from CART statistics improve decided accuracy over standard trees by blending boundary predictions and flagging uncertain cases.
-
Ternary Decision Trees with Locally-Adaptive Uncertainty Zones
Ternary decision trees add locally adaptive uncertainty zones around splits and estimate zone width from existing CART statistics, yielding higher decided accuracy than standard trees on 71 OpenML datasets.
-
Critic-Driven Voronoi-Quantization for Distilling Deep RL Policies to Explainable Models
Critic-Driven Voronoi State Partitioning distills deep RL policies into piecewise-linear models by iteratively adding linear subpolicies in high-value-error regions identified by the critic.
-
Prophecy: Inferring Formal Properties from Neuron Activations
Prophecy infers formal properties of feed-forward neural networks by extracting rules from neuron activation patterns that imply desirable output behaviors.
-
The Ratchet Effect in Silico: How Interaction Drives Cumulative Intelligence in Large Language Models
Populations of 1-4B parameter LLMs using peer verification and shared cultural memory achieve 8.8-18.9 point gains on mathematical reasoning tasks and close much of the gap to 70B+ single models.
-
SkillTree: Explainable Skill-Based Deep Reinforcement Learning for Long-Horizon Control Tasks
SkillTree reduces continuous action spaces to discrete skills via a differentiable decision tree in a hierarchical policy, achieving comparable performance to neural skill methods with added skill-level explainability...
-
A 65 nm Trustworthy Hypoglycemia Forecasting Engine Achieving 11.3 nJ per Inference
A fabricated 65 nm chip uses a hybrid probabilistic decision tree architecture for hypoglycemia forecasting, achieving 11.3 nJ/inference, F1 score 0.825, and 4.1x-16.1x better noise robustness than conventional models.
-
SaliencyDecor: Enhancing Neural Network Interpretability through Feature Decorrelation
Enforcing feature decorrelation during training produces sharper saliency maps and higher accuracy on image classification benchmarks.
-
Explainable AI for Next-Generation Wireless Physical Layer: Basics, State-of-the-Art, and Open Challenges
A survey formalizing responsibility-oriented goals for wireless XAI, developing a taxonomy of explainability approaches, reviewing PHY layer applications, and discussing open challenges including performance tradeoffs...
-
Cross-Paradigm Knowledge Distillation: A Comprehensive Study of Bidirectional Transfer Between Random Forests and Deep Neural Networks for Big Data Applications
A study of bidirectional knowledge transfer between Random Forests and Deep Neural Networks using proposed distillation methods, evaluated on classification and regression tasks across six datasets.
-
What does it mean to understand a neural network?
Simple training code produces complex neural networks, suggesting that brain learning rules may be easier to understand than mature brain properties and that neuroscience should shift focus accordingly.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.