Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2405.19062 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866910462580359168 |
|---|---|
| author | Fang, Lanting Yang, Yulian Wang, Kai Feng, Shanshan Feng, Kaiyu Gui, Jie Wang, Shuliang Ong, Yew-Soon |
| author_facet | Fang, Lanting Yang, Yulian Wang, Kai Feng, Shanshan Feng, Kaiyu Gui, Jie Wang, Shuliang Ong, Yew-Soon |
| contents | While dynamic graph neural networks have shown promise in various applications, explaining their predictions on continuous-time dynamic graphs (CTDGs) is difficult. This paper investigates a new research task: self-interpretable GNNs for CTDGs. We aim to predict future links within the dynamic graph while simultaneously providing causal explanations for these predictions. There are two key challenges: (1) capturing the underlying structural and temporal information that remains consistent across both independent and identically distributed (IID) and out-of-distribution (OOD) data, and (2) efficiently generating high-quality link prediction results and explanations. To tackle these challenges, we propose a novel causal inference model, namely the Independent and Confounded Causal Model (ICCM). ICCM is then integrated into a deep learning architecture that considers both effectiveness and efficiency. Extensive experiments demonstrate that our proposed model significantly outperforms existing methods across link prediction accuracy, explanation quality, and robustness to shortcut features. Our code and datasets are anonymously released at https://github.com/2024SIG/SIG. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_19062 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | SIG: Efficient Self-Interpretable Graph Neural Network for Continuous-time Dynamic Graphs Fang, Lanting Yang, Yulian Wang, Kai Feng, Shanshan Feng, Kaiyu Gui, Jie Wang, Shuliang Ong, Yew-Soon Machine Learning Artificial Intelligence While dynamic graph neural networks have shown promise in various applications, explaining their predictions on continuous-time dynamic graphs (CTDGs) is difficult. This paper investigates a new research task: self-interpretable GNNs for CTDGs. We aim to predict future links within the dynamic graph while simultaneously providing causal explanations for these predictions. There are two key challenges: (1) capturing the underlying structural and temporal information that remains consistent across both independent and identically distributed (IID) and out-of-distribution (OOD) data, and (2) efficiently generating high-quality link prediction results and explanations. To tackle these challenges, we propose a novel causal inference model, namely the Independent and Confounded Causal Model (ICCM). ICCM is then integrated into a deep learning architecture that considers both effectiveness and efficiency. Extensive experiments demonstrate that our proposed model significantly outperforms existing methods across link prediction accuracy, explanation quality, and robustness to shortcut features. Our code and datasets are anonymously released at https://github.com/2024SIG/SIG. |
| title | SIG: Efficient Self-Interpretable Graph Neural Network for Continuous-time Dynamic Graphs |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2405.19062 |