Online GNN Evaluation Under Test-time Graph Distribution Shifts

Xin Zheng, Dongjin Song, Qingsong Wen, Bo Du, Shirui Pan
3/1/2024

No official code found. Best implementation: Amanda-Zheng/LEBED (16 stars, MIT, last commit 2024-03-23).

Abstract

Evaluating the performance of a well-trained GNN model on real-world graphs is a pivotal step for reliable GNN online deployment and serving. Due to a lack of test node labels and unknown potential training-test graph data distribution shifts, conventional model evaluation encounters limitations in calculating performance metrics (e.g., test error) and measuring graph data-level discrepancies, particularly when the training graph used for developing GNNs remains unobserved during test time. In this paper, we study a new research problem, online GNN evaluation, which aims to provide valuable insights into the well-trained GNNs's ability to effectively generalize to real-world unlabeled graphs under the test-time graph distribution shifts. Concretely, we develop an effective learning behavior discrepancy score, dubbed LeBeD, to estimate the test-time generalization errors of well-trained GNN models. Through a novel GNN re-training strategy with a parameter-free optimality criterion, the proposed LeBeD comprehensively integrates learning behavior discrepancies from both node prediction and structure reconstruction perspectives. This enables the effective evaluation of the well-trained GNNs' ability to capture test node semantics and structural representations, making it an expressive metric for estimating the generalization error in online GNN evaluation. Extensive experiments on real-world test graphs under diverse graph distribution shifts could verify the effectiveness of the proposed method, revealing its strong correlation with ground-truth test errors on various well-trained GNN models.

View on arXivView PDF

Code Implementations(1)

Amanda-Zheng/LEBEDCommunity100%
MIT

Pytorch implementation for ICLR24:"Online GNN Evaluation Under Test-Time Graph Distribution Shifts"

160Shell, PythonMar 14, 20242 years agoMIT

Similar Papers

Cite this paper

@article{zheng2024online,
  title  = {Online GNN Evaluation Under Test-time Graph Distribution Shifts},
  author = {Xin Zheng and Dongjin Song and Qingsong Wen and Bo Du and Shirui Pan},
  year   = {2024},
  eprint = {2403.09953},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2403.09953},
  journal = {ICLR 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.09953.svg)](https://codeofpaper.com/paper/2403.09953)

Discussion