Visual diagnosis of tree boosting methods

Liu, Shixia, Xiao, Jiannan, Liu, Junlin, Wang, Xiting, Wu, Jing ORCID: https://orcid.org/0000-0001-5123-9861 and Zhu, Jun 2018. Visual diagnosis of tree boosting methods. IEEE Transactions on Visualization and Computer Graphics 24 (1) , pp. 163-173. 10.1109/TVCG.2017.2744378

PDF - Published Version Available under License Creative Commons Attribution. Download (5MB)

Abstract

Tree boosting, which combines weak learners (typically decision trees) to generate a strong learner, is a highly effective and widely used machine learning method. However, the development of a high performance tree boosting model is a time-consuming process that requires numerous trial-and-error experiments. To tackle this issue, we have developed a visual diagnosis tool, BOOSTVis, to help experts quickly analyze and diagnose the training process of tree boosting. In particular, we have designed a temporal confusion matrix visualization, and combined it with a t-SNE projection and a tree visualization. These visualization components work together to provide a comprehensive overview of a tree boosting model, and enable an effective diagnosis of an unsatisfactory training process. Two case studies that were conducted on the Otto Group Product Classification Challenge dataset demonstrate that BOOSTVis can provide informative feedback and guidance to improve understanding and diagnosis of tree boosting algorithms.

Item Type:	Article
Date Type:	Publication
Status:	Published
Schools:	Computer Science & Informatics
Additional Information:	This work is licensed under a Creative Commons Attribution 3.0 License
Publisher:	IEEE
ISSN:	1077-2626
Date of First Compliant Deposit:	1 September 2017
Date of Acceptance:	1 August 2017
Last Modified:	05 May 2023 19:31
URI:	https://orca.cardiff.ac.uk/id/eprint/104233

Citation Data

Cited 53 times in Scopus. View in Scopus. Powered By Scopus® Data

Actions (repository staff only)

Edit Item