TFWaveFormer: Temporal-Frequency Collaborative Multi-level Wavelet Transformer for Dynamic Link Prediction
About
Dynamic link prediction plays a crucial role in diverse applications including social network analysis, communication forecasting, and financial modeling. While recent Transformer-based approaches have demonstrated promising results in temporal graph learning, their performance remains limited when capturing complex multi-scale temporal dynamics. In this paper, we propose TFWaveFormer, a novel Transformer architecture that integrates temporal-frequency analysis with multi-resolution wavelet decomposition to enhance dynamic link prediction. Our framework comprises three key components: (i) a temporal-frequency coordination mechanism that jointly models temporal and spectral representations, (ii) a learnable multi-resolution wavelet decomposition module that adaptively extracts multi-scale temporal patterns through parallel convolutions, replacing traditional iterative wavelet transforms, and (iii) a hybrid Transformer module that effectively fuses local wavelet features with global temporal dependencies. Extensive experiments on benchmark datasets demonstrate that TFWaveFormer achieves state-of-the-art performance, outperforming existing Transformer-based and hybrid models by significant margins across multiple metrics. The superior performance of TFWaveFormer validates the effectiveness of combining temporal-frequency analysis with wavelet decomposition in capturing complex temporal dynamics for dynamic link prediction tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Link Prediction | Reddit (inductive) | AP68.54 | 81 | |
| Link Prediction | Enron (inductive) | AP81.26 | 66 | |
| Inductive dynamic link prediction | Reddit (inductive) | AUC-ROC (%)66.97 | 65 | |
| Link Prediction | Enron (transductive) | AP92.7 | 49 | |
| Inductive dynamic link prediction | Wikipedia (inductive) | AUC-ROC0.6878 | 44 | |
| Dynamic Link Prediction | Wikipedia (inductive) | AP70.08 | 44 | |
| Dynamic Link Detection | ENRON | -- | 44 | |
| Future Link Prediction | MOOC (inductive) | AP79.99 | 39 | |
| transductive dynamic link prediction | Wikipedia | AUC ROC99.31 | 37 | |
| Dynamic Link Prediction | MOOC (transductive) | AUC91.89 | 34 |