Causality-Aware Spatiotemporal Graph Neural Networks for Spatiotemporal Time Series Imputation

Abstract

Spatiotemporal time series are usually collected via monitoring sensors placed at different locations, which usually contain missing values due to various failures, such as mechanical damages and Internet outages. Imputing the missing values is crucial for analyzing time series. When recovering a specific data point, most existing methods consider all the information relevant to that point regardless of the cause-and-effect relationship. During data collection, it is inevitable that some unknown confounders are included, e.g., background noise in time series and non-causal shortcut edges in the constructed sensor network. These confounders could open backdoor paths and establish non-causal correlations between the input and output. Over-exploiting these non-causal correlations could cause overfitting. In this paper, we first revisit spatiotemporal time series imputation from a causal perspective and show how to block the confounders via the frontdoor adjustment. Based on the results of frontdoor adjustment, we introduce a novel Causality-Aware Spatiotemporal Graph Neural Network (Casper), which contains a novel Prompt Based Decoder (PBD) and a Spatiotemporal Causal Attention (SCA). PBD could reduce the impact of confounders and SCA could discover the sparse causal relationships among embeddings. Theoretical analysis reveals that SCA discovers causal relationships based on the values of gradients. We evaluate Casper on three real-world datasets, and the experimental results show that Casper could outperform the baselines and could effectively discover causal relationships.

Accepted by CIKM'2024. Fixed typos

References in corpus (12)

• Causal Attention for Interpretable and Generalizable Graph Classification
• HDMI: High-order Deep Multiplex Infomax
• CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation
• Should Graph Convolution Trust Neighbors? A Simple Causal Inference Method
• Graph Communal Contrastive Learning
• Networked Time Series Imputation via Position-aware Graph Enhanced Variational Autoencoders
• Network of Tensor Time Series
• Learning Continuous System Dynamics from Irregularly-Sampled Partial Observations
• X-GOAL: Multiplex Heterogeneous Graph Prototypical Contrastive Learning
• Learning to Reconstruct Missing Data from Spatiotemporal Graphs with Sparse Observations
• CLEAR: Generative Counterfactual Explanations on Graphs
• A Look into Causal Effects under Entangled Treatment in Graphs: Investigating the Impact of Contact on MRSA Infection