arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2205.04035cs.LG
收藏

支持可解释模型的通用线坐标决策树可视化

Visualization of Decision Trees based on General Line Coordinates to Support Explainable Models

Central Washington University(中央华盛顿大学)

Alex Worland, Sridevi Wagle, Boris Kovalerchuk

摘要:

机器学习(ML)模型的可视化是ML过程的重要组成部分,以增强ML模型的可解释性和预测精度。提出了一种新的决策树可视化方法SPC-DT。这些方法使用一种名为“移位配对坐标”(SPC)的“常规线坐标”版本。在SPC中,每个n-D点在一组移位的2-D笛卡尔坐标对中可视化为有向图。新方法扩展并补充了现有方法的功能,以可视化DT模型。它显示:(1)属性之间的关系,(2)相对于DT结构的各个情况,(3)DT中的数据流,(4)每个分裂与DT节点中的阈值的紧密程度,以及(5)n-D空间的部分中的情况的密度。这些信息对于领域专家评估和改进DT模型非常重要,包括避免模型的过度泛化和过度拟合以及它们的性能。在案例研究中,使用三个真实的数据集的方法的好处。

英文摘要:

Visualization of Machine Learning (ML) models is an important part of the ML process to enhance the interpretability and prediction accuracy of the ML models. This paper proposes a new method SPC-DT to visualize the Decision Tree (DT) as interpretable models. These methods use a version of General Line Coordinates called Shifted Paired Coordinates (SPC). In SPC, each n-D point is visualized in a set of shifted pairs of 2-D Cartesian coordinates as a directed graph. The new method expands and complements the capabilities of existing methods, to visualize DT models. It shows: (1) relations between attributes, (2) individual cases relative to the DT structure, (3) data flow in the DT, (4) how tight each split is to thresholds in the DT nodes, and (5) the density of cases in parts of the n-D space. This information is important for domain experts for evaluating and improving the DT models, including avoiding overgeneralization and overfitting of models, along with their performance. The benefits of the methods are demonstrated in the case studies, using three real datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑