arXiv:2509.15611stat.MLcs.LG2025-09被引 1

提出可解释的网络增强随机森林,兼顾高精度与透明决策。

Interpretable Network-assisted Random Forest+

  • 基于随机森林扩展,融合网络邻域信息提升预测。
  • 在多个数据集上达到接近甚至超过GNN的准确率。
  • 提供特征重要性、网络贡献度等多维可解释工具,适合医疗等高风险场景。

机器学习算法通常假设训练样本相互独立。当数据点通过网络连接时,样本间的依赖关系既降低了有效样本量,也为利用邻居信息提升预测提供了机会。现有方法如图神经网络虽能利用网络信息,但难以解释;而网络辅助线性回归虽可解释,但性能较差。本文提出一类基于随机森林(RF+)泛化的网络辅助模型,兼具高预测精度与可解释性。通过开发一套解释工具,可识别驱动预测的重要特征,并量化网络信息对预测的贡献。提供全局与局部重要性度量及样本影响分析,支持对单个观测的影响评估。该框架拓展了网络辅助机器学习在需透明决策的高影响力问题中的应用范围。

原文摘要 · Abstract (English)

Machine learning algorithms often assume that training samples are independent. When data points are connected by a network, the induced dependency between samples is both a challenge, reducing effective sample size, and an opportunity to improve prediction by leveraging information from network neighbors. Multiple methods taking advantage of this opportunity are now available, but many, including graph neural networks, are not easily interpretable, limiting their usefulness for understanding how a model makes its predictions. Others, such as network-assisted linear regression, are interpretable but often yield substantially worse prediction performance. We bridge this gap by proposing a family of flexible network-assisted models built upon a generalization of random forests (RF+), which achieves highly-competitive prediction accuracy and can be interpreted through feature importance measures. In particular, we develop a suite of interpretation tools that enable practitioners to not only identify important features that drive model predictions, but also quantify the importance of the network contribution to prediction. Importantly, we provide both global and local importance measures as well as sample influence measures to assess the impact of a given observation. This suite of tools broadens the scope and applicability of network-assisted machine learning for high-impact problems where interpretability and transparency are essential.

可解释性随机森林图神经网络网络建模

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