跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

基于路径的梯度提升用于图级别预测

文章摘要

PathBoost是一种新的梯度树提升方法,用于图级别的分类和回归。它直接从输入图结构中学习基于路径的判别特征,扩展了先前的工作,支持二分类、多节点和边属性,并自动选择锚点节点。在多个基准数据集上,PathBoost在半数测试中优于图神经网络和图核方法,其余结果相当,尤其在大图上表现更佳。

来源arXiv Machine Learning作者: Claudio Meggio, Johan Pensar, Riccardo De Bin
基于路径的梯度提升用于图级别预测
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

研究人员Claudio Meggio及其合作者提出了一种名为PathBoost的新型梯度树提升方法,专门用于图级别的分类与回归任务。与传统的黑盒模型不同,PathBoost能够直接从输入图结构中学习具有判别能力的路径特征,从而显著提升模型的可解释性。该方法建立在先前针对特定化学应用的研究基础上,但引入了三个关键扩展:首先,通过逻辑损失函数的梯度提升来适应二分类问题;其次,采用基于前缀的分解技术,将多种节点和边属性整合到路径特征空间中;第三,基于类别属性的多样性自动选择锚点节点,无需用户手动指定路径起点。

为了评估PathBoost的性能,研究团队在多个基准数据集上将其与图神经网络(GNN)和图核方法进行了对比。实验结果显示,PathBoost在半数数据集上取得了更优的结果,在其余数据集上则实现了相当的性能。特别值得注意的是,PathBoost在平均节点数较大的图上表现尤为出色,这表明该方法具有良好的可扩展性。总体而言,这些结果证明基于路径的梯度提升方法能够与更为复杂的黑盒方法相竞争,为可解释的图级别预测提供了一条有前景的路径。

展开要点与分析

文章情报

研究者进阶

要点

  • PathBoost是一种基于路径特征的梯度提升方法,适用于图级别预测。
  • 它通过前缀分解整合多种节点和边属性,并自动选择锚点节点。
  • 在多个基准上,PathBoost与GNN和核方法相比,半数更优,其余持平。
  • 该方法在节点数较大的图上表现更好,证明了路径提升的竞争力。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。