Du Keqing, Yang Xinyu, Chen Hang
Xi'an Jiaotong University, No. 28 Xianning West Road, Xi'an Shaanxi, 710049, PR China.
Neural Netw. 2025 Nov;191:107826. doi: 10.1016/j.neunet.2025.107826. Epub 2025 Jul 3.
Multivariate spatio-temporal forecasting aims to predict the future evolution of multiple interdependent variables distributed across space and time. Effectively capturing the underlying causal dependencies among these variables is essential for enhancing model interpretability, robustness, and decision support in complex systems. However, existing methods often fall short in modeling complete and dynamic causal dependencies due to the presence of latent confounders and the challenges of identifying multidimensional causal interactions. To address these challenges, we propose MCST, a novel framework that systematically refines the causal generation process of each variable through comprehensive causal modeling. MCST first applies variational inference to disentangle variable-specific exogenous factors and identify latent confounders within a shared latent space. To capture dynamic causal dependencies, we design a causal estimator that quantifies both instantaneous and lagged causal transmission across spatial, temporal, and inter-variable dimensions. These estimated causal transmissions are then integrated with exogenous and endogenous components using SCMs, enabling the construction of refined, variable-wise causal generation mechanisms for accurate forecasting. Extensive experiments on three real-world and one synthetic dataset demonstrate that MCST consistently outperforms existing approaches in predictive performance while providing enhanced interpretability through explicit causal reasoning.
多变量时空预测旨在预测跨空间和时间分布的多个相互依赖变量的未来演变。有效捕捉这些变量之间潜在的因果依赖关系对于增强复杂系统中的模型可解释性、稳健性和决策支持至关重要。然而,由于存在潜在混杂因素以及识别多维因果相互作用的挑战,现有方法在对完整和动态因果依赖关系进行建模时往往存在不足。为应对这些挑战,我们提出了MCST,这是一个新颖的框架,通过全面的因果建模系统地优化每个变量的因果生成过程。MCST首先应用变分推理来解开特定变量的外生因素,并在共享潜在空间中识别潜在混杂因素。为了捕捉动态因果依赖关系,我们设计了一个因果估计器,用于量化跨空间、时间和变量间维度的即时和滞后因果传递。然后,使用结构因果模型将这些估计的因果传递与外生和内生成分相结合,从而构建精细的、按变量的因果生成机制以进行准确预测。在三个真实世界数据集和一个合成数据集上进行的大量实验表明,MCST在预测性能方面始终优于现有方法,同时通过显式因果推理提供了增强的可解释性。