arXiv:2607.07935cs.LGcs.AI2026-07

用路径梯度提升法,让图数据预测结果变得可解释

path_boost: A Python Package for Interpretable Graph-Level Prediction using Path-Based Gradient Boosting

论文配图:path_boost: A Python Package for Interpretable Graph-Level Prediction using Path-Based Gradient Boosting
图 1 · 摘自论文原文
  • 通过迭代选择有预测力的路径构建模型
  • 在6个分子数据集上优于图神经网络和图核方法
  • 适合需要解释性的分子性质预测场景

我们提出 path_boost,一个用于图结构数据可解释监督学习的 Python 工具包。该工具包实现了 PathBoost 算法,一种在学习过程中自动发现图中具有预测能力的带标签路径的梯度提升方法。与通常难以解释的图神经网络不同,PathBoost 基于路径特征构建可加性预测模型,明确揭示哪些子结构驱动了预测结果。为避免穷举所有可能路径,算法根据预测能力迭代选择并扩展路径,利用提升法将弱学习器组合成强集成模型。该工具包支持回归和二分类任务,具备与 scikit-learn 兼容、支持自定义基学习器与选择器、自动起始节点选择、跨锚点并行训练及内置变量重要性计算等特性。我们在过渡金属化合物分子性质预测任务中验证了 PathBoost,其中原子为节点,键为边,并在六个分子数据集上与主流图神经网络和图核方法进行了基准对比。该工具包已开源,发布于 PyPI 与 GitHub。

原文摘要 · Abstract (English)

We present path_boost, a Python package for interpretable supervised learning on graph-structured input data. The package implements PathBoost, a gradient boosting algorithm that automatically discovers predictive labeled paths within graphs during the learning process. Unlike graph neural networks, which are generally difficult to interpret, PathBoost produces an additive prediction model over path-based features that explicitly reveals which substructures drive predictions. To avoid an exhaustive enumeration of all possible paths, the algorithm iteratively selects and extends paths during learning based on their predictive power, using boosting to combine weak learners into a strong ensemble. The package supports both regression and binary classification. Key features include compatibility with scikit-learn workflows, support for custom base learners and selectors, automatic starting node selection, parallel training across anchor nodes, and built-in variable importance computation. We demonstrate PathBoost on molecular property prediction of transition metal compounds, where atoms serve as nodes and bonds as edges, and further benchmark PathBoost against an established graph neural network and a graph kernel method across six molecular datasets. The package is available on PyPI and GitHub under an open-source license.

图学习可解释性路径挖掘

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