LLM2D
RoCP-GNN:用于节点分类的图神经网络鲁棒一致性预测
RoCP-GNN: Robust Conformal Prediction for Graph Neural Networks in Node-Classification
作者: S. Akansha
发布日期: 10/10/2024
arXiv ID: oai:arXiv.org:2408.13825v2

摘要

图神经网络 (GNN) 已成为预测图结构数据结果的有力工具。然而,GNN 的一个显著局限性在于其无法提供稳健的不确定性估计,这削弱了其在错误代价高昂的环境中的可靠性。解决这个问题的一种方法是提供包含真实标签且具有预定义概率裕度的预测集。我们的方法基于一致性预测 (CP),这是一个有望构建统计稳健预测集或区间的方法框架。主要有两个挑战:首先,对于像图这样的相关数据,当应用于节点分类时,CP 中的关键假设——可交换性——是否仍然成立尚不清楚。其次,即使一致性预测的假设对于一致性链接预测有效,我们也需要确保高效率,即生成的预测集或区间的长度足够小以提供有用的信息。在本文中,我们提出了一种名为鲁棒一致性预测图神经网络 (RoCP-GNN) 的新方法,该方法将一致性预测 (CP) 直接集成到 GNN 训练过程中。该方法生成预测集,而不仅仅是点预测,在仅假设可交换性的前提下,这些预测集在用户定义的置信水平下是有效的。我们的方法使用任何预测性 GNN 模型稳健地预测结果,同时量化基于图的半监督学习 (SSL) 领域内预测的不确定性。实验结果表明,具有大小损失的 GNN 模型在性能上具有统计显著的提高。我们通过将其与各种最先进的 GNN 结合用于节点分类,在标准图基准数据集上验证了我们的方法。代码将在发表后提供。