arXiv:2507.22665cs.LGcs.HC2025-07

通过聚类提升随机森林可解释性,让复杂模型像树状图一样看懂。

Cluster-Based Random Forest Visualization and Interpretation

  • 按决策规则与预测结果聚类相似决策树,减少分析负担。
  • 在Glass数据集上验证有效,用户研究显示理解效率提升。
  • 提供特征位置和决策规则双可视化,适合模型调试者使用。

随机森林由大量决策树构成,虽分类性能优越且泛化能力强,但难以解释。本文提出一种可视化方法与系统,通过聚类相似决策树,使用户无需逐个分析每棵树,即可理解模型整体行为。为此,我们设计了一种新距离度量,综合考虑决策规则与预测结果。同时提出两种可视化方式:特征图(Feature Plot)展示特征在树中的拓扑位置,规则图(Rule Plot)呈现决策规则。通过在相对复杂的Glass标准数据集上的案例研究及小规模用户实验,验证了该方法的有效性。

原文摘要 · Abstract (English)

Random forests are a machine learning method used to automatically classify datasets and consist of a multitude of decision trees. While these random forests often have higher performance and generalize better than a single decision tree, they are also harder to interpret. This paper presents a visualization method and system to increase interpretability of random forests. We cluster similar trees which enables users to interpret how the model performs in general without needing to analyze each individual decision tree in detail, or interpret an oversimplified summary of the full forest. To meaningfully cluster the decision trees, we introduce a new distance metric that takes into account both the decision rules as well as the predictions of a pair of decision trees. We also propose two new visualization methods that visualize both clustered and individual decision trees: (1) The Feature Plot, which visualizes the topological position of features in the decision trees, and (2) the Rule Plot, which visualizes the decision rules of the decision trees. We demonstrate the efficacy of our approach through a case study on the "Glass" dataset, which is a relatively complex standard machine learning dataset, as well as a small user study.

可解释性随机森林可视化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。