arrow
Return

RuleExplorer: A Scalable Matrix Visualization for Understanding Tree Ensemble Classifiers

delete2025-09-01
delete0
PRE
AI
Z
Zhen Li
W
Weikai Yang
J
Jun Yuan
吴婧 (Jing Wu)
C
Changjian Chen
Y
Yao Ming
F
Fan Yang
H
Hui Zhang
S
Shixia Liu
DOI:10.1109/TVCG.2024.3514115delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
The high performance of tree ensemble classifiers benefits from a large set of rules, which, in turn, makes the models hard to understand. To improve interpretability, existing methods extract a subset of rules for approximation using model reduction techniques. However, by focusing on the reduced rule set, these methods often lose fidelity and ignore anomalous rules that, despite their infrequency, play crucial roles in real-world applications. This paper introduces a scalable visual analysis method to explain tree ensemble classifiers that contain tens of thousands of rules. The key idea is to address the issue of losing fidelity by adaptively organizing the rules as a hierarchy rather than reducing them. To ensure the inclusion of anomalous rules, we develop an anomaly-biased model reduction method to prioritize these rules at each hierarchical level. Synergized with this hierarchical organization of rules, we develop a matrix-based hierarchical visualization to support exploration at different levels of detail. Our quantitative experiments and case studies demonstrate how our method fosters a deeper understanding of both common and anomalous rules, thereby enhancing interpretability without sacrificing comprehensiveness.
Keywords:
Hierarchical visualization
model reduction
tree ensemble classifier

Journal

IEEE Transactions on Visualization and Computer Graphics cover
IEEE Transactions on Visualization and Computer Graphics
IF:
6.5
Papers:
294
Citations:
2.2W

Organization

H
Hong Kong University of Science and Technology
Scholars:
2.0K
Papers: 1.2K
Citations: 3.9W
T
tsinghua university
Scholars:
11.6W
Papers: 9.9W
Citations: 137
C
Cardiff University
Scholars:
2.7W
Papers: 2.5W
Citations: 3.5W
H
hunan university
Scholars:
4.4W
Papers: 3.3W
Citations: 70
K
kuaishou technology, beijing, china
Scholars:
6
Papers: 4
Citations: 0
C
citadel securities, new york, usa
Scholars:
1
Papers: 1
Citations: 0
researcher View more organizations