Beyond Sparsity: Tree Regularization of Deep Models for Interpretability
Not on the map. Why?
Abstract (arXiv)
The lack of interpretability remains a key barrier to the adoption of deep models in many applications. In this work, we explicitly regularize deep models so human users might step through the process behind their predictions in little time. Specifically, we train deep time-series models so their class-probability predictions have high accuracy while being closely modeled by decision trees with few nodes. Using intuitive toy examples as well as medical tasks for treating sepsis and HIV, we demonstrate that this new tree regularization yields models that are easier for humans to simulate than simpler L1 or L2 penalties without sacrificing predictive power.
Key facts. Cited by 68 papers in the map (data through 7 Oct 2026). First posted on arXiv 16 Nov 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 0; 2020: 0; 2021: 19; 2022: 18; 2023: 14; 2024: 13; 2025: 3.
68 papers.
Sort:
By area:
- 2022From Anecdotal Evidence to Quantitative Evaluation Methods: A Systematic Review on Evaluating Explainable AIcited by 299
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
- 2023Generative Time Series Forecasting with Diffusion, Denoise, and Disentanglementcited by 89
- 2021Unbox the Black-box for the Medical Explainable AI via Multi-modal and Multi-centre Data Fusion: A Mini-Review, Two Showcases and Beyondcited by 75
- 2021Improving Deep Learning Interpretability by Saliency Guided Trainingcited by 68
- 2022Global Explainability of GNNs via Logic Combination of Learned Conceptscited by 64
- 2022Leveraging Explanations in Interactive Machine Learning: An Overviewcited by 52
- 2021A Review on Explainability in Multimodal Deep Neural Netscited by 46
- 2021Defining and Quantifying the Emergence of Sparse Concepts in DNNscited by 36
- 2021Explainable AI for Robot Failures: Generating Explanations that Improve User Assistance in Fault Recoverycited by 32
- 2023Does a Neural Network Really Encode Symbolic Concepts?cited by 27
- 2022Explainable AI for clinical and remote health applications: a survey on tabular and time series datacited by 24
38 more
- 2023A Review on Explainable Artificial Intelligence for Healthcare: Why, How, and When?cited by 24
- 2022Computing Abductive Explanations for Boosted Treescited by 23
- 2021Taxonomy of Machine Learning Safety: A Survey and Primercited by 22
- 2021DeepAID: Interpreting and Improving Deep Learning-based Anomaly Detection in Security Applicationscited by 18
- 2024CausalFormer: An Interpretable Transformer for Temporal Causal Discoverycited by 17
- 2022Neural Networks are Decision Treescited by 16
- 2021Show or Suppress? Managing Input Uncertainty in Machine Learning Model Explanationscited by 15
- 2021Explainable Activity Recognition for Smart Home Systemscited by 14
- 2024A Survey of Explainable Artificial Intelligence (XAI) in Financial Time Series Forecastingcited by 14
- 2018Granger-causal Attentive Mixtures of Experts: Learning Important Features with Neural Networkscited by 13
- 2022Attributable Visual Similarity Learningcited by 12
- 2022Generating Synthetic Clinical Data that Capture Class Imbalanced Distributions with Generative Adversarial Networks: Example using Antiretroviral Therapy for HIVcited by 12
- 2022Learning Interpretable, High-Performing Policies for Autonomous Drivingcited by 11
- 2021Learning Optimal Predictive Checklistscited by 10
- 2023Concept Distillation: Leveraging Human-Centered Explanations for Model Improvementcited by 9
- 2023Interpretability-Aware Vision Transformercited by 8
- 2024Tree-Based Leakage Inspection and Control in Concept Bottleneck Modelscited by 8
- 2021Feature Importance Explanations for Temporal Black-Box Modelscited by 7
- 2022A Survey of Neural Treescited by 7
- 2023A Survey on Trustworthy Edge Intelligence: From Security and Reliability To Transparency and Sustainabilitycited by 7
- 2024A Survey of Explainable Knowledge Tracingcited by 7
- 2021Explainable Diabetic Retinopathy Detection and Retinal Image Generationcited by 6
- 2022DT+GNN: A Fully Explainable Graph Neural Network using Decision Treescited by 6
- 2024Learning Electromagnetic Metamaterial Physics With ChatGPTcited by 6
- 2024Logical Distillation of Graph Neural Networkscited by 6
- 2023Diagrammatization and Abduction to Improve AI Interpretability With Domain-Aligned Explanations for Medical Diagnosiscited by 4
- 2021Improving Compositionality of Neural Networks by Decoding Representations to Inputscited by 2
- 2021Learning Conditional Invariance through Cycle Consistencycited by 2
- 2021Tree-based Focused Web Crawling with Reinforcement Learningcited by 2
- 2023SCAAT: Improving Neural Network Interpretability via Saliency Constrained Adaptive Adversarial Trainingcited by 2
- 2024CA-Stream: Attention-based pooling for interpretable image recognitioncited by 2
- 2024Understanding with toy surrogate models in machine learningcited by 2
- 2022Explainable multi-class anomaly detection on functional datacited by 1
- 2022NN2Rules: Extracting Rule List from Neural Networkscited by 1
- 2022EiHi Net: Out-of-Distribution Generalization Paradigmcited by 1
- 2023MED-VT++: Unifying Multimodal Learning with a Multiscale Encoder-Decoder Video Transformercited by 1
- 2023Rule-Extraction Methods From Feedforward Neural Networks: A Systematic Literature Reviewcited by 1
- 2024Axe the X in XAI: A Plea for Understandable AIcited by 1
50 most cited of 68 All 68 (JSON)
- 2025Towards Provably Unlearnable Examples via Bayes Error Optimisation11 Nov 2025
- 2025NIMO: a Nonlinear Interpretable MOdel5 Jun 2025
- 2025Evaluating Scenario-based Decision-making for Interactive Autonomous Driving Using Rational Criteria: A Survey3 Jan 2025
- 2024Tree-Based Leakage Inspection and Control in Concept Bottleneck Models8 Oct 2024
- 2024Understanding with toy surrogate models in machine learning8 Oct 2024
- 2024AOTree: Aspect Order Tree-based Model for Explainable Recommendation29 Jul 2024
- 2024A Survey of Explainable Artificial Intelligence (XAI) in Financial Time Series Forecasting22 Jul 2024
- 2024Semantically Rich Local Dataset Generation for Explainable AI in Genomics3 Jul 2024
- 2024CausalFormer: An Interpretable Transformer for Temporal Causal Discovery24 Jun 2024
- 2024TREE: Tree Regularization for Efficient Execution18 Jun 2024
- 2024Logical Distillation of Graph Neural Networks11 Jun 2024
- 2024Learning Electromagnetic Metamaterial Physics With ChatGPT23 Apr 2024
38 more
- 2024CA-Stream: Attention-based pooling for interpretable image recognition23 Apr 2024
- 2024An Interpretable Power System Transient Stability Assessment Method with Expert Guiding Neural-Regression-Tree3 Apr 2024
- 2024A Survey of Explainable Knowledge Tracing12 Mar 2024
- 2024Axe the X in XAI: A Plea for Understandable AI1 Mar 2024
- 2023Rule-Extraction Methods From Feedforward Neural Networks: A Systematic Literature Review20 Dec 2023
- 2023Concept Distillation: Leveraging Human-Centered Explanations for Model Improvement26 Nov 2023
- 2023SCAAT: Improving Neural Network Interpretability via Saliency Constrained Adaptive Adversarial Training9 Nov 2023
- 2023A Survey on Trustworthy Edge Intelligence: From Security and Reliability To Transparency and Sustainability27 Oct 2023
- 2023Interpretable time series neural representation for classification purposes25 Oct 2023
- 2023Interpretability-Aware Vision Transformer14 Sep 2023
- 2023Reinforcement Logic Rule Learning for Temporal Point Processes11 Aug 2023
- 2023Does Saliency-Based Training bring Robustness for Deep Neural Networks in Image Classification?28 Jun 2023
- 2023MED-VT++: Unifying Multimodal Learning with a Multiscale Encoder-Decoder Video Transformer12 Apr 2023
- 2023A Review on Explainable Artificial Intelligence for Healthcare: Why, How, and When?10 Apr 2023
- 2023LEURN: Learning Explainable Univariate Rules with Neural Networks27 Mar 2023
- 2023Does a Neural Network Really Encode Symbolic Concepts?25 Feb 2023
- 2023Diagrammatization and Abduction to Improve AI Interpretability With Domain-Aligned Explanations for Medical Diagnosis2 Feb 2023
- 2023Generative Time Series Forecasting with Diffusion, Denoise, and Disentanglement8 Jan 2023
- 2022Global Explainability of GNNs via Logic Combination of Learned Concepts13 Oct 2022
- 2022Neural Networks are Decision Trees11 Oct 2022
- 2022EiHi Net: Out-of-Distribution Generalization Paradigm29 Sep 2022
- 2022Computing Abductive Explanations for Boosted Trees16 Sep 2022
- 2022Explainable AI for clinical and remote health applications: a survey on tabular and time series data14 Sep 2022
- 2022Arithmetical Binary Decision Tree Traversals11 Sep 2022
- 2022A Survey of Neural Trees7 Sep 2022
- 2022Generating Synthetic Clinical Data that Capture Class Imbalanced Distributions with Generative Adversarial Networks: Example using Antiretroviral Therapy for HIV18 Aug 2022
- 2022Leveraging Explanations in Interactive Machine Learning: An Overview29 Jul 2022
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networks27 Jul 2022
- 2022NN2Rules: Extracting Rule List from Neural Networks4 Jul 2022
- 2022PhilaeX: Explaining the Failure and Success of AI Models in Malware Detection2 Jul 2022
- 2022DT+GNN: A Fully Explainable Graph Neural Network using Decision Trees26 May 2022
- 2022Explainable multi-class anomaly detection on functional data3 May 2022
- 2022Semantic interpretation for convolutional neural networks: What makes a cat a cat?16 Apr 2022
- 2022Attributable Visual Similarity Learning28 Mar 2022
- 2022Learning Interpretable, High-Performing Policies for Autonomous Driving4 Feb 2022
- 2022From Anecdotal Evidence to Quantitative Evaluation Methods: A Systematic Review on Evaluating Explainable AI20 Jan 2022
- 2021Tree-based Focused Web Crawling with Reinforcement Learning12 Dec 2021
- 2021Learning Optimal Predictive Checklists2 Dec 2021
50 newest of 68 All 68 (JSON)
- 2022From Anecdotal Evidence to Quantitative Evaluation Methods: A Systematic Review on Evaluating Explainable AIcited by 299
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
- 2023Generative Time Series Forecasting with Diffusion, Denoise, and Disentanglementcited by 89
- 2021Unbox the Black-box for the Medical Explainable AI via Multi-modal and Multi-centre Data Fusion: A Mini-Review, Two Showcases and Beyondcited by 75
- 2021Improving Deep Learning Interpretability by Saliency Guided Trainingcited by 68
- 2022Global Explainability of GNNs via Logic Combination of Learned Conceptscited by 64
- 2022Leveraging Explanations in Interactive Machine Learning: An Overviewcited by 52
- 2021A Review on Explainability in Multimodal Deep Neural Netscited by 46
- 2021Defining and Quantifying the Emergence of Sparse Concepts in DNNscited by 36
- 2021Explainable AI for Robot Failures: Generating Explanations that Improve User Assistance in Fault Recoverycited by 32
- 2023Does a Neural Network Really Encode Symbolic Concepts?cited by 27
- 2022Explainable AI for clinical and remote health applications: a survey on tabular and time series datacited by 24
12 most cited of 65 All 65 (JSON)
- 2025Towards Provably Unlearnable Examples via Bayes Error Optimisation11 Nov 2025
- 2025NIMO: a Nonlinear Interpretable MOdel5 Jun 2025
- 2024Tree-Based Leakage Inspection and Control in Concept Bottleneck Models8 Oct 2024
- 2024Understanding with toy surrogate models in machine learning8 Oct 2024
- 2024AOTree: Aspect Order Tree-based Model for Explainable Recommendation29 Jul 2024
- 2024A Survey of Explainable Artificial Intelligence (XAI) in Financial Time Series Forecasting22 Jul 2024
- 2024Semantically Rich Local Dataset Generation for Explainable AI in Genomics3 Jul 2024
- 2024CausalFormer: An Interpretable Transformer for Temporal Causal Discovery24 Jun 2024
- 2024TREE: Tree Regularization for Efficient Execution18 Jun 2024
- 2024Logical Distillation of Graph Neural Networks11 Jun 2024
- 2024Learning Electromagnetic Metamaterial Physics With ChatGPT23 Apr 2024
- 2024CA-Stream: Attention-based pooling for interpretable image recognition23 Apr 2024
12 newest of 65 All 65 (JSON)