跳到主要导航 跳到搜索 跳到主要内容

STCTN:一种基于时域偏倚校正与空域因果传递的时空因果表示学习方法

  • Pan Deng
  • , Jun Ting Liu
  • , Xiao Wang
  • , Xiao Feng Jia*
  • , Yu Zhao
  • , Mu Lan Wang
  • , Xing Yuan Dai
  • *此作品的通讯作者
  • Beihang University
  • School of Artificial Intelligence of Anhui University
  • Beijing Big Data Centre
  • CAS - Institute of Automation

科研成果: 期刊稿件文章同行评审

摘要

Spatio-temporal data, sampled from complex dynamical systems, is ubiquitous in real world, e. g., traffic flow, meteorological records and energy consumption. Learning effective feature representation from spatiotemporal data is the foundation of spatiotemporal data mining. The existing models overemphasize the statistical correlations in the spatiotemporal data and are susceptible to spurious correlations, which makes it hard to extract unbiased and robust feature representation. We model the generation process of spatiotemporal data based on the structural causal model (SCM), analyze the causes of spurious correlations in observations, and propose a spatiotemporal causal representation learning method based on temporal bias adjustment and spatial causal transition. Here, we focus on two challenges in spatio-temporal representation learning: (1) Eliminating temporal confounding bias. Existing models fail to handle causal relationships and certainly not eliminate the influence of confounders in temporal domain. Hence, the second challenge is to remove confounding bias and extract unbiased temporal representations. (2) Modelling spatial causal relationships. Limited by predefined graph structures, existing models are susceptible to non-causal spatial spurious correlations so it is significant to recover underlying spatial causal structure under causal constraints. First, we eliminate temporal spurious correlation based on backdoor adjustment. Then, we construct causal transition network for eliminating spatial spurious correlation. Finally, the downstream feature decoder applies the causal representation to downstream tasks. We tackle spatio-temporal representation learning tasks from a causal perspective and analyze the causes of spatial and temporal spurious correlation in observation data. To the best of our knowledge, this is the first attempt to apply causal theory to spatio-temporal representation learning. Experiments on two real-world datasets show that the proposed spatiotemporal representation learning method effectively avoids the interference of spurious correlations, enhances the stability of the model, and reduces the prediction error of nodes with sparse data in two downstream prediction tasks by 3% and 10% respectively.

投稿的翻译标题STCTN: A Spatio-temporal Causal Representation Learning Method Based on Temporal Bias Adjustment and Spatial Causal Transition
源语言繁体中文
页(从-至)2335-2550
页数216
期刊Jisuanji Xuebao/Chinese Journal of Computers
46
12
DOI
出版状态已出版 - 12月 2023

联合国可持续发展目标

此成果有助于实现下列可持续发展目标:

  1. 可持续发展目标 7 - 经济适用的清洁能源
    可持续发展目标 7 经济适用的清洁能源

关键词

  • backdoor adjustment
  • causality relation
  • spatio-temporal representation learning
  • spurious correlations
  • structural causal model

指纹

探究 'STCTN:一种基于时域偏倚校正与空域因果传递的时空因果表示学习方法' 的科研主题。它们共同构成独一无二的指纹。

引用此