提升随机森林对无主效应交互项的建模与可解释性
Unity Forests: Improving Interaction Modelling and Interpretability in Random Forests
- 通过联合优化树根分裂捕捉无主效应的交互关系
- 新变量重要性度量能准确识别纯交互效应变量
- 可解释的树根代表提供变量作用条件的直观洞察
随机森林(RFs)广泛用于预测和变量重要性分析,常被认为可通过递归分割捕获各类交互。但由于分割是局部选择的,只有当至少一个相关协变量具有边际效应时,交互才被可靠捕捉。本文提出统一森林(UFOs),一种改进的随机森林变体,旨在更好利用无边际效应的交互。在UFOs中,每棵树的前几层分裂在随机协变量子集上联合优化,形成“树根”以捕捉此类交互;其余部分按常规生长。我们进一步提出统一变量重要性度量(VIM),基于树根的袋外分裂准则值。仅考虑每协变量中袋内准则值最高的少数树根分裂,反映仅当交互协变量提前分裂时,纯交互效应才具区分性。最后引入协变量代表性树根(CRTRs),为每个协变量选取代表性树根,揭示其最强效应发生的边际或交互条件。模拟研究表明,统一VIM能可靠识别无边际效应的交互变量,而传统方法无法做到。大规模真实数据比较显示,UFOs在判别力和预测精度上优于标准随机森林,校准性能相当。CRTRs在模拟数据中准确还原真实效应类型,并在真实数据分析中提供有意义见解。
原文摘要 · Abstract (English)
Random forests (RFs) are widely used for prediction and variable importance analysis and are often believed to capture any types of interactions via recursive splitting. However, since the splits are chosen locally, interactions are only reliably captured when at least one involved covariate has a marginal effect. We introduce unity forests (UFOs), an RF variant designed to better exploit interactions involving covariates without marginal effects. In UFOs, the first few splits of each tree are optimized jointly across a random covariate subset to form a "tree root" capturing such interactions; the remainder is grown conventionally. We further propose the unity variable importance measure (VIM), which is based on out-of-bag split criterion values from the tree roots. Here, only a small fraction of tree root splits with the highest in-bag criterion values are considered per covariate, reflecting that covariates with purely interaction-based effects are discriminative only if a split in an interacting covariate occurred earlier in the tree. Finally, we introduce covariate-representative tree roots (CRTRs), which select representative tree roots per covariate and provide interpretable insight into the conditions - marginal or interactive - under which each covariate has its strongest effects. In a simulation study, the unity VIM reliably identified interacting covariates without marginal effects, unlike conventional RF-based VIMs. In a large-scale real-data comparison, UFOs achieved higher discrimination and predictive accuracy than standard RFs, with comparable calibration. The CRTRs reproduced the covariates' true effect types reliably in simulated data and provided interesting insights in a real data analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。