Explaining Temporal Graph Network Predictions via Memory Backtracking.

Yazheng Liu, Xi Zhang, Sihong Xie, Hui Xiong· July 10, 2026 View original

Summary

Researchers propose a new method to explain predictions made by Temporal Graph Networks (TGNs) by analyzing the influence of historical events. This approach uses topology attribution and memory backtracking trees to quantify how past interactions and memory updates shape node predictions.

A novel methodology has been introduced to enhance the explainability of Temporal Graph Networks (TGNs), which are widely used for predictions on dynamic, real-world graphs. Current explanation methods for TGNs often overlook the crucial memory module, which stores and updates node histories, leaving the impact of past events on predictions largely opaque. The proposed approach addresses this by attributing TGN predictions through two interconnected structures: the topology attribution tree and the memory backtracking tree. The topology attribution tree identifies the influence of neighboring nodes and their memory vectors, while the memory backtracking tree quantifies how specific historical events contribute to the evolution of a node's memory vector. By applying Layer-wise Relevance Propagation (LRP) within TGNs, the method ensures that the total contribution of events precisely sums up to the model's logits. Furthermore, it designs optimization objectives to identify truly important events, overcoming potential unfaithfulness from simple top-k selections on non-linear probability mappings. Experiments across nine temporal graph datasets and various tasks demonstrate that this method provides more faithful explanations and outperforms existing state-of-the-art baselines.

Why it matters

For professionals deploying AI in critical applications, understanding why a TGN makes a particular prediction is crucial for building trust, debugging, and ensuring fairness. This method provides a pathway to greater transparency in dynamic graph-based AI systems.

How to implement this in your domain

  1. 1Integrate this explainability method into your TGN development pipeline for critical applications like fraud detection or recommendation systems.
  2. 2Use the explanations to debug TGN models, identify biases, and improve model robustness.
  3. 3Train data scientists and AI engineers on interpreting the topology and memory backtracking trees.
  4. 4Develop user interfaces that visualize these explanations for stakeholders who need to understand model decisions.

Who benefits

BFSISocial MediaCybersecurityHealthcareLogistics

Key takeaways

  • A new method explains TGN predictions by tracing historical event influence.
  • It uses topology attribution and memory backtracking trees to quantify event impact.
  • The approach ensures faithful explanations by applying LRP and optimized event selection.
  • Improved explainability enhances trust and debugging for dynamic graph-based AI.

Original post by Yazheng Liu, Xi Zhang, Sihong Xie, Hui Xiong

"arXiv:2607.07716v1 Announce Type: cross Abstract: Temporal graphs are ubiquitous in real-world applications and Temporal Graph Networks (TGNs) have achieved superior predictive accuracy. Understanding which historical events drive model predictions can enhance trustworthiness of…"

View on X

Originally posted by Yazheng Liu, Xi Zhang, Sihong Xie, Hui Xiong on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses