cs.LGJun 25, 2026

Explaining Temporal Graph Neural Networks via Feature-induced Information Flow

Authors: Ping XiongThomas SchnakeKlaus-Robert MüllerShinichi Nakajima

Organizations: Berlin Institute for the Foundations of Learning and Data – BIFOLD, 10623 Berlin, Germany · Machine Learning Group, Technical University of Berlin, Berlin, Germany · Department of Chemistry, Chemical Physics Theory Group, University of Toronto, Toronto, Canada · Vector Institute for Artificial Intelligence, Toronto, Canada · Acceleration Consortium, University of Toronto, Toronto, Canada · Department of Artificial Intelligence, Korea University, Seoul, Korea · Max Planck Institute for Informatics, Saarbrücken, Germany · RIKEN AIP, Tokyo, Japan

Abstract

Event-based Temporal Graph Neural Networks (ETGNNs) have demonstrated strong performance across a wide range of applications, including social network analysis, epidemic tracing, recommender systems, and political event forecasting. However, their increasing complexity poses significant challenges for explainability. Existing explanation methods focus only on a subset of the information flow within ETGNNs, typically tracing contributions from the event-related embeddings to the output. Consequently, they overlook the important pathways through event-induced variables, which mediate interactions between nodes and thereby play a central role in capturing long-range temporal dependencies. To overcome this limitation, we propose a novel attribution method that analyzes the entire information flow through all event-associated variables. Our method is built upon the recent Normalized Relevance Measure (NRM) framework, which enables explicit quantification of information flow originating from event embeddings as well as information flow passing through event-induced variables. It also ensures comparability of latent variables across layers, and supports higher-order analysis of interactions between events. To handle the architectural complexity of ETGNNs, we extend the NRM framework with a modular decomposition procedure that facilitates the systematic construction of relevance structure for complex neural architectures. We evaluate our approach on two synthetic datasets for epidemic tracing and social dynamics, as well as a real-world dataset of political event networks. Our qualitative and quantitative experiments show that our method consistently outperforms existing explanation approaches while producing more human-interpretable explanations.

Explore similar work

CardsList