arXiv:2412.03190cs.LG2024-12

让图神经网络在不确定时主动放弃预测,提升可靠性。

Node Classification With Integrated Reject Option

  • 在GNN中集成拒答机制,根据置信度决定是否预测
  • 在Cora/Citeseer/Pubmed上准确率提升1.5%~3.2%,拒答率可控
  • 可解释拒答原因,适合法律判决等高风险场景

节点分类是图学习中的关键任务。尽管图神经网络(GNN)已广泛应用于各类场景,但其对拒答选项设置的适应性尚未被探索。本文提出NCwR,一种在图神经网络中集成拒答选项的新方法,使模型在不确定性较高时可主动放弃预测。我们提出了基于成本和覆盖率的两种拒答策略,并在三个标准引用网络数据集(Cora、Citeseer、Pubmed)上进行实验,与基线方法对比。此外,我们将ILDC数据集上的法律判决预测问题建模为节点分类任务,其中节点代表案件,边代表引用关系。通过可视化分析模型拒答的案例,揭示了影响决策的关键输入特征部分。

原文摘要 · Abstract (English)

One of the key tasks in graph learning is node classification. While Graph neural networks have been used for various applications, their adaptivity to reject option setting is not previously explored. In this paper, we propose NCwR, a novel approach to node classification in Graph Neural Networks (GNNs) with an integrated reject option, which allows the model to abstain from making predictions when uncertainty is high. We propose both cost-based and coverage-based methods for classification with abstention in node classification setting using GNNs. We perform experiments using our method on three standard citation network datasets Cora, Citeseer and Pubmed and compare with relevant baselines. We also model the Legal judgment prediction problem on ILDC dataset as a node classification problem where nodes represent legal cases and edges represent citations. We further interpret the model by analyzing the cases that the model abstains from predicting by visualizing which part of the input features influenced this decision.

图神经网络节点分类拒答机制可解释性

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