VisRuler: Visual Analytics for Extracting Decision Rules from Bagged and Boosted Decision Trees

Abstract

Bagging and boosting are two popular ensemble methods in machine learning (ML) that produce many individual decision trees. Due to the inherent ensemble characteristic of these methods, they typically outperform single decision trees or other ML models in predictive performance. However, numerous decision paths are generated for each decision tree, increasing the overall complexity of the model and hindering its use in domains that require trustworthy and explainable decisions, such as finance, social care, and health care. Thus, the interpretability of bagging and boosting algorithms, such as random forest and adaptive boosting, reduces as the number of decisions rises. In this paper, we propose a visual analytics tool that aims to assist users in extracting decisions from such ML models via a thorough visual inspection workflow that includes selecting a set of robust and diverse models (originating from different ensemble learning algorithms), choosing important features according to their global contribution, and deciding which decisions are essential for global explanation (or locally, for specific cases). The outcome is a final decision based on the class agreement of several models and the explored manual decisions exported by users. We evaluated the applicability and effectiveness of VisRuler via a use case, a usage scenario, and a user study. The evaluation revealed that most users managed to successfully use our system to explore decision rules visually, performing the proposed tasks and answering the given questions in a satisfying way.

This manuscript is accepted for publication in the Information Visualization (IV) - SAGE Journals

References in corpus (12)

• Scikit-learn: Machine Learning in Python
• XGBoost: A Scalable Tree Boosting System
• UMAP: Uniform Manifold Approximation and Projection for Dimension Reduction
• Popular Ensemble Methods: An Empirical Study
• Social Influence and the Collective Dynamics of Opinion Formation
• Manifold: A Model-Agnostic Framework for Interpretation and Diagnosis of Machine Learning Models
• The State of the Art in Enhancing Trust in Machine Learning Models with the Use of Visualizations
• StackGenVis: Alignment of Data, Algorithms, and Models for Stacking Ensemble Learning Using Performance Metrics
• Explainable Matrix -- Visualization for Global and Local Interpretability of Random Forest Classification Ensembles
• VisEvol: Visual Analytics to Support Hyperparameter Search through Evolutionary Optimization
• An Interactive Machine Learning Framework
• ExplorerTree: a focus+context exploration approach for 2D embeddings

Cited by in corpus (3)

• TimberTrek: Exploring and Curating Sparse Decision Trees with Interactive Visualization
• Visual Analytics for Explainable and Trustworthy Artificial Intelligence
• DeforestVis: Behavior Analysis of Machine Learning Models with Surrogate Decision Stumps