arXiv · 2504.07337
FLASH: Flexible Learning of Adaptive Sampling from History in Temporal Graph Neural Networks
Abstract
Aggregating temporal signals from historic interactions is a key step in future link prediction on dynamic graphs. However, incorporating long histories is resource-intensive. Hence, temporal graph neural networks (TGNNs) often rely on historical neighbors sampling heuristics such as uniform sampling or recent neighbors selection. These heuristics are static and fail to adapt to the underlying graph structure. We introduce FLASH, a learnable and graph-adaptive neighborhood selection mechanism that generalizes existing heuristics. FLASH integrates seamlessly into TGNNs and is trained end-to-end using a self-supervised ranking loss. We provide theoretical evidence that commonly used heuristics hinder TGNNs performance, motivating our design. Extensive experiments across multiple benchmarks demonstrate consistent and significant performance improvements for TGNNs equipped with FLASH.
Explore related subjects
Keep this discovery
Or Feldman, Krishna Sri Ipsit Mantri, Carola-Bibiane Schönlieb, Chaim Baskin, Moshe Eliasof. 2025-04-09. FLASH: Flexible Learning of Adaptive Sampling from History in Temporal Graph Neural Networks. https://arxiv.org/abs/2504.07337
Cite the original work for its findings. Save a collection to share your selection of sources.