Skip to main navigation Skip to search Skip to main content

Temporal Graph Neural Tangent Kernel with Graphon-Guaranteed

  • University of Illinois at Urbana-Champaign
  • Meta Ai
  • IBM

Research output: Contribution to journalConference articlepeer-review

9 Scopus citations

Abstract

Graph Neural Tangent Kernel (GNTK) fuses graph neural networks and graph kernels, simplifies the process of graph representation learning, interprets the training dynamics of graph neural networks, and serves various applications like protein identification, image segmentation, and social network analysis. In practice, graph data carries complex information among entities that inevitably evolves over time, and previous static graph neural tangent kernel methods may be stuck in the sub-optimal solution in terms of both effectiveness and efficiency. As a result, extending the advantage of GNTK to temporal graphs becomes a critical problem. To this end, we propose the temporal graph neural tangent kernel, which not only extends the simplicity and interpretation ability of GNTK to the temporal setting but also leads to rigorous temporal graph classification error bounds. Furthermore, we prove that when the input temporal graph grows over time in the number of nodes, our temporal graph neural tangent kernel will converge in the limit to the graphon NTK value, which implies the transferability and robustness of the proposed kernel method, named Temporal Graph Neural Tangent Kernel with Graphon-Guaranteed or Temp-G3NTK. In addition to the theoretical analysis, we also perform extensive experiments, not only demonstrating the superiority of Temp-G3NTK in the temporal graph classification task, but also showing that Temp-G3NTK can achieve very competitive performance in node-level tasks like node classification compared with various SOTA graph kernel and representation learning baselines. Our code is available at https://github.com/kthrn22/TempGNTK.

Original languageEnglish
JournalAdvances in Neural Information Processing Systems
Volume37
StatePublished - 2024
Event38th Conference on Neural Information Processing Systems, NeurIPS 2024 - Vancouver, Canada
Duration: Dec 9 2024Dec 15 2024

Fingerprint

Dive into the research topics of 'Temporal Graph Neural Tangent Kernel with Graphon-Guaranteed'. Together they form a unique fingerprint.

Cite this