摘要
arXiv:2405.11968v3 宣布类型: replace-cross
摘要:图神经网络(GNNs)已经成为了预测图结构化数据中结果的强大工具。尽管它们非常有效,但GNNs的一个显著缺点在于它们有限的能力提供稳健的不确定性估计,这在错误可能带来重大后果的背景下给它们的可靠性带来了挑战。此外,GNNs通常在同分布设置下表现优异,即训练数据和测试数据遵循相同分布,这一条件在实际的图数据场景中往往不满足。本文通过对预测模型输出进行转换以生成预测集,利用一种广泛认可的统计技术——一致性预测,来解决图基半监督学习(SSL)中条件迁移下的不确定性量化问题。此外,我们提出了一个新颖的损失函数,旨在通过最小化潜变量阶段的条件迁移来改进模型预测。我们称这种方法为适用于GNN的一致性预测条件迁移鲁棒(CondSR)。我们的方法是模型无关的,并能够适应各种分类模型。我们在标准图基准数据集上验证了方法的有效性,并将其与最先进的GNNs结合用于节点分类任务中。综合评估表明,我们的方法可以一致地达到任何预定义的目标边际覆盖率,条件迁移下将最先进的GNN模型的准确性提高最多12%,并将预测集大小减少最多48%。该代码实现已公开,供进一步探索和实验使用。