DeforestVis: Behavior Analysis of Machine Learning Models with Surrogate Decision Stumps

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chatzimparmpas, Angelos, Martins, Rafael M., Telea, Alexandru C., Kerren, Andreas
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913320578056192
author Chatzimparmpas, Angelos
Martins, Rafael M.
Telea, Alexandru C.
Kerren, Andreas
author_facet Chatzimparmpas, Angelos
Martins, Rafael M.
Telea, Alexandru C.
Kerren, Andreas
contents As the complexity of machine learning (ML) models increases and their application in different (and critical) domains grows, there is a strong demand for more interpretable and trustworthy ML. A direct, model-agnostic, way to interpret such models is to train surrogate models-such as rule sets and decision trees-that sufficiently approximate the original ones while being simpler and easier-to-explain. Yet, rule sets can become very lengthy, with many if-else statements, and decision tree depth grows rapidly when accurately emulating complex ML models. In such cases, both approaches can fail to meet their core goal-providing users with model interpretability. To tackle this, we propose DeforestVis, a visual analytics tool that offers summarization of the behaviour of complex ML models by providing surrogate decision stumps (one-level decision trees) generated with the Adaptive Boosting (AdaBoost) technique. DeforestVis helps users to explore the complexity versus fidelity trade-off by incrementally generating more stumps, creating attribute-based explanations with weighted stumps to justify decision making, and analysing the impact of rule overriding on training instance allocation between one or more stumps. An independent test set allows users to monitor the effectiveness of manual rule changes and form hypotheses based on case-by-case analyses. We show the applicability and usefulness of DeforestVis with two use cases and expert interviews with data analysts and model developers.
format Preprint
id arxiv_https___arxiv_org_abs_2304_00133
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle DeforestVis: Behavior Analysis of Machine Learning Models with Surrogate Decision Stumps
Chatzimparmpas, Angelos
Martins, Rafael M.
Telea, Alexandru C.
Kerren, Andreas
Machine Learning
Human-Computer Interaction
As the complexity of machine learning (ML) models increases and their application in different (and critical) domains grows, there is a strong demand for more interpretable and trustworthy ML. A direct, model-agnostic, way to interpret such models is to train surrogate models-such as rule sets and decision trees-that sufficiently approximate the original ones while being simpler and easier-to-explain. Yet, rule sets can become very lengthy, with many if-else statements, and decision tree depth grows rapidly when accurately emulating complex ML models. In such cases, both approaches can fail to meet their core goal-providing users with model interpretability. To tackle this, we propose DeforestVis, a visual analytics tool that offers summarization of the behaviour of complex ML models by providing surrogate decision stumps (one-level decision trees) generated with the Adaptive Boosting (AdaBoost) technique. DeforestVis helps users to explore the complexity versus fidelity trade-off by incrementally generating more stumps, creating attribute-based explanations with weighted stumps to justify decision making, and analysing the impact of rule overriding on training instance allocation between one or more stumps. An independent test set allows users to monitor the effectiveness of manual rule changes and form hypotheses based on case-by-case analyses. We show the applicability and usefulness of DeforestVis with two use cases and expert interviews with data analysts and model developers.
title DeforestVis: Behavior Analysis of Machine Learning Models with Surrogate Decision Stumps
topic Machine Learning
Human-Computer Interaction
url https://arxiv.org/abs/2304.00133