arXiv:2507.02151cs.LG2025-07KDD被引 6

为动态图神经网络设计了能准确评估不确定性的新方法。

Non-exchangeable Conformal Prediction for Temporal Graph Neural Networks

  • 用扩散模型构建捕捉时空不确定性的非交换性评分
  • 在真实动态图上实现保证覆盖率,最大减少31%预测集大小
  • 适合需要可靠预测的金融反洗钱等高风险场景

图神经网络(GNN)的置信预测提供了一种量化不确定性、提升高风险应用中可靠性的重要框架。然而,现有方法主要针对静态图,忽视了现实图结构随时间演化的特性。图结构、节点属性及真实标签中的时序依赖性违反了标准置信预测的可交换性假设,限制了其适用性。本文提出NCPNET,一种专为时序图设计的端到端置信预测框架。该方法将置信预测扩展至动态场景,缓解由时序依赖引起的统计覆盖率失效问题。我们提出基于扩散的非符合性评分,同时捕捉拓扑与时间不确定性;并设计高效优化算法,提升计算效率、降低覆盖率偏差。在WIKI、REDDIT、DBLP和IBM反洗钱数据集上的大量实验表明,NCPNET能在时序图中确保覆盖保证,在WIKI数据集上预测集大小最多减少31%,显著优于当前最优方法。代码与数据已公开于https://github.com/ODYSSEYWT/NCPNET。

原文摘要 · Abstract (English)

Conformal prediction for graph neural networks (GNNs) offers a promising framework for quantifying uncertainty, enhancing GNN reliability in high-stakes applications. However, existing methods predominantly focus on static graphs, neglecting the evolving nature of real-world graphs. Temporal dependencies in graph structure, node attributes, and ground truth labels violate the fundamental exchangeability assumption of standard conformal prediction methods, limiting their applicability. To address these challenges, in this paper, we introduce NCPNET, a novel end-to-end conformal prediction framework tailored for temporal graphs. Our approach extends conformal prediction to dynamic settings, mitigating statistical coverage violations induced by temporal dependencies. To achieve this, we propose a diffusion-based non-conformity score that captures both topological and temporal uncertainties within evolving networks. Additionally, we develop an efficiency-aware optimization algorithm that improves the conformal prediction process, enhancing computational efficiency and reducing coverage violations. Extensive experiments on diverse real-world temporal graphs, including WIKI, REDDIT, DBLP, and IBM Anti-Money Laundering dataset, demonstrate NCPNET's capability to ensure guaranteed coverage in temporal graphs, achieving up to a 31% reduction in prediction set size on the WIKI dataset, significantly improving efficiency compared to state-of-the-art methods. Our data and code are available at https://github.com/ODYSSEYWT/NCPNET.

图神经网络置信预测动态图不确定性

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