arXiv:2606.10770stat.MEcs.AI2026-06

修正随机森林变量重要性,解决相关变量干扰问题

Correcting Variable Importance Scored by Random Forests

论文配图:Correcting Variable Importance Scored by Random Forests
图 1 · 摘自论文原文
  • 按响应变量条件相关性分组变量,降低相关性干扰
  • 两种方法均有效提升重要变量的得分准确性
  • 适合需要精准特征筛选的建模与解释场景

随机森林生成的变量重要性广泛应用于统计数据分析,在模型解释、选择与诊断、成本受限学习等任务中发挥重要作用。然而,其变量重要性计算未考虑变量间的相关性,导致与多个变量相关的变量常被低估或完全掩盖(重要性接近零)。为消除无关相关变量的影响,我们提出基于响应变量条件相关性的变量分组方法。探索了两种计算高效方案:一是逐个变量分组并将其与相关变量分离;二是利用聚类根据成对条件相关性分组。实验表明,两种方法均能合理修正变量重要性。

原文摘要 · Abstract (English)

Variable importance produced by Random Forests (RF) is used widely in statistical data analysis, and has played an important role in a variety of tasks such as assisting model interpretation, model selection and diagnosis, and cost-bounded learning etc. However, the calculation of variable importance in RF does not take into account of the correlations among variables, and variables that are correlated to many other variables tend to receive a lower importance index or being completely masked (i.e., with an importance index near zero) by other strongly correlated variables. To prevent influence from unwanted correlated variables in calculating variable importance, we propose to group variables by their conditional correlations (conditional on the response variable). We explore two computationally efficient options, with one grouping variables individually, and then separates the variable of interest from all correlated variables, while the other uses clustering to group variables according to their pair-wise conditional correlations. Our experiments show that both lead to sensible corrections to the importance of variables.

随机森林变量重要性相关性修正

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