Saved in:
Bibliographic Details
Main Authors: Huang, Yuting, Fang, Ziquan, Zeng, Zhihao, Chen, Lu, Gao, Yunjun
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.17637
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909872363143168
author Huang, Yuting
Fang, Ziquan
Zeng, Zhihao
Chen, Lu
Gao, Yunjun
author_facet Huang, Yuting
Fang, Ziquan
Zeng, Zhihao
Chen, Lu
Gao, Yunjun
contents Spatio-temporal prediction plays a crucial role in intelligent transportation, weather forecasting, and urban planning. While integrating multi-modal data has shown potential for enhancing prediction accuracy, key challenges persist: (i) inadequate fusion of multi-modal information, (ii) confounding factors that obscure causal relations, and (iii) high computational complexity of prediction models. To address these challenges, we propose E^2-CSTP, an Effective and Efficient Causal multi-modal Spatio-Temporal Prediction framework. E^2-CSTP leverages cross-modal attention and gating mechanisms to effectively integrate multi-modal data. Building on this, we design a dual-branch causal inference approach: the primary branch focuses on spatio-temporal prediction, while the auxiliary branch mitigates bias by modeling additional modalities and applying causal interventions to uncover true causal dependencies. To improve model efficiency, we integrate GCN with the Mamba architecture for accelerated spatio-temporal encoding. Extensive experiments on 4 real-world datasets show that E^2-CSTP significantly outperforms 9 state-of-the-art methods, achieving up to 9.66% improvements in accuracy as well as 17.37%-56.11% reductions in computational overhead.
format Preprint
id arxiv_https___arxiv_org_abs_2505_17637
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Causal Spatio-Temporal Prediction: An Effective and Efficient Multi-Modal Approach
Huang, Yuting
Fang, Ziquan
Zeng, Zhihao
Chen, Lu
Gao, Yunjun
Machine Learning
Spatio-temporal prediction plays a crucial role in intelligent transportation, weather forecasting, and urban planning. While integrating multi-modal data has shown potential for enhancing prediction accuracy, key challenges persist: (i) inadequate fusion of multi-modal information, (ii) confounding factors that obscure causal relations, and (iii) high computational complexity of prediction models. To address these challenges, we propose E^2-CSTP, an Effective and Efficient Causal multi-modal Spatio-Temporal Prediction framework. E^2-CSTP leverages cross-modal attention and gating mechanisms to effectively integrate multi-modal data. Building on this, we design a dual-branch causal inference approach: the primary branch focuses on spatio-temporal prediction, while the auxiliary branch mitigates bias by modeling additional modalities and applying causal interventions to uncover true causal dependencies. To improve model efficiency, we integrate GCN with the Mamba architecture for accelerated spatio-temporal encoding. Extensive experiments on 4 real-world datasets show that E^2-CSTP significantly outperforms 9 state-of-the-art methods, achieving up to 9.66% improvements in accuracy as well as 17.37%-56.11% reductions in computational overhead.
title Causal Spatio-Temporal Prediction: An Effective and Efficient Multi-Modal Approach
topic Machine Learning
url https://arxiv.org/abs/2505.17637