VisRuler: Visual analytics for extracting decision rules from bagged and boosted decision trees

被引:6
作者
Chatzimparmpas, Angelos [1 ,3 ]
Martins, Rafael M. [1 ]
Kerren, Andreas [1 ,2 ]
机构
[1] Linnaeus Univ, Dept Comp Sci & Media Technol, Vaxjo, Sweden
[2] Linkoping Univ, Dept Sci & Technol, Norrkoping, Sweden
[3] Linnaeus Univ, Dept Comp Sci & Media Technol, Vejdes Plats 7, S-35195 Vaxjo, Kronoberg, Sweden
关键词
Decisions evaluation; rules interpretation; ensemble learning; visual analytics; visualization; MACHINE LEARNING-MODELS; CLASSIFICATION; EXPLORATION; VISUALIZATION; ALGORITHMS; SELECTION; ADABOOST; SEARCH; SPACE; USERS;
D O I
10.1177/14738716221142005
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
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.
引用
收藏
页码:115 / 139
页数:25
相关论文
共 126 条
[41]   Greedy function approximation: A gradient boosting machine [J].
Friedman, JH .
ANNALS OF STATISTICS, 2001, 29 (05) :1189-1232
[42]   Variable selection using random forests [J].
Genuer, Robin ;
Poggi, Jean-Michel ;
Tuleau-Malot, Christine .
PATTERN RECOGNITION LETTERS, 2010, 31 (14) :2225-2236
[43]   Extremely randomized trees [J].
Geurts, P ;
Ernst, D ;
Wehenkel, L .
MACHINE LEARNING, 2006, 63 (01) :3-42
[44]   Boxer: Interactive Comparison of Classifier Results [J].
Gleicher, Michael ;
Barve, Aditya ;
Yu, Xinyi ;
Heimerl, Florian .
COMPUTER GRAPHICS FORUM, 2020, 39 (03) :181-193
[45]   Visualizing Change over Time Using Dynamic Hierarchies: TreeVersity2 and the StemView [J].
Guerra-Gomez, John Alexis ;
Pack, Michael L. ;
Plaisant, Catherine ;
Shneiderman, Ben .
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS, 2013, 19 (12) :2566-2575
[46]   Colorful Trees: Visualizing Random Forests for Analysis and Interpretation [J].
Haensch, Ronny ;
Wiesner, Philipp ;
Wendler, Sophie ;
Hellwich, Olaf .
2019 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2019, :294-302
[47]  
Hastie T., 2009, The elements of statistical learning: data mining, inference, and prediction: with 200 full-color illustrations (Springer series in statistics)
[48]  
Hilasaca GM., 2019, ARXIV
[49]   GBRTVis: online analysis of gradient boosting regression tree [J].
Huang, Yifei ;
Liu, Yuhua ;
Li, Chenhui ;
Wang, Changbo .
JOURNAL OF VISUALIZATION, 2019, 22 (01) :125-140
[50]   Deep Sequencing of the Vaginal Microbiota of Women with HIV [J].
Hummelen, Ruben ;
Fernandes, Andrew D. ;
Macklaim, Jean M. ;
Dickson, Russell J. ;
Changalucha, John ;
Gloor, Gregory B. ;
Reid, Gregor .
PLOS ONE, 2010, 5 (08)