A Transfer Framework for Enhancing Temporal Graph Learning in Data-Scarce Settings

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Agarwal, Sidharth, Dubey, Tanishq, Gupta, Shubham, Bedathur, Srikanta
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914259501318144
author Agarwal, Sidharth
Dubey, Tanishq
Gupta, Shubham
Bedathur, Srikanta
author_facet Agarwal, Sidharth
Dubey, Tanishq
Gupta, Shubham
Bedathur, Srikanta
contents Dynamic interactions between entities are prevalent in domains like social platforms, financial systems, healthcare, and e-commerce. These interactions can be effectively represented as time-evolving graphs, where predicting future connections is a key task in applications such as recommendation systems. Temporal Graph Neural Networks (TGNNs) have achieved strong results for such predictive tasks but typically require extensive training data, which is often limited in real-world scenarios. One approach to mitigating data scarcity is leveraging pre-trained models from related datasets. However, direct knowledge transfer between TGNNs is challenging due to their reliance on node-specific memory structures, making them inherently difficult to adapt across datasets. To address this, we introduce a novel transfer approach that disentangles node representations from their associated features through a structured bipartite encoding mechanism. This decoupling enables more effective transfer of memory components and other learned inductive patterns from one dataset to another. Empirical evaluations on real-world benchmarks demonstrate that our method significantly enhances TGNN performance in low-data regimes, outperforming non-transfer baselines by up to 56\% and surpassing existing transfer strategies by 36\%
format Preprint
id arxiv_https___arxiv_org_abs_2503_00852
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Transfer Framework for Enhancing Temporal Graph Learning in Data-Scarce Settings
Agarwal, Sidharth
Dubey, Tanishq
Gupta, Shubham
Bedathur, Srikanta
Machine Learning
Information Theory
Dynamic interactions between entities are prevalent in domains like social platforms, financial systems, healthcare, and e-commerce. These interactions can be effectively represented as time-evolving graphs, where predicting future connections is a key task in applications such as recommendation systems. Temporal Graph Neural Networks (TGNNs) have achieved strong results for such predictive tasks but typically require extensive training data, which is often limited in real-world scenarios. One approach to mitigating data scarcity is leveraging pre-trained models from related datasets. However, direct knowledge transfer between TGNNs is challenging due to their reliance on node-specific memory structures, making them inherently difficult to adapt across datasets. To address this, we introduce a novel transfer approach that disentangles node representations from their associated features through a structured bipartite encoding mechanism. This decoupling enables more effective transfer of memory components and other learned inductive patterns from one dataset to another. Empirical evaluations on real-world benchmarks demonstrate that our method significantly enhances TGNN performance in low-data regimes, outperforming non-transfer baselines by up to 56\% and surpassing existing transfer strategies by 36\%
title A Transfer Framework for Enhancing Temporal Graph Learning in Data-Scarce Settings
topic Machine Learning
Information Theory
url https://arxiv.org/abs/2503.00852