Transformer-Based Spatial-Temporal Counterfactual Outcomes Estimation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, He, Chi, Haoang, Liu, Mingyu, Huang, Wanrong, Xu, Liyang, Yang, Wenjing
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916811537121280
author Li, He
Chi, Haoang
Liu, Mingyu
Huang, Wanrong
Xu, Liyang
Yang, Wenjing
author_facet Li, He
Chi, Haoang
Liu, Mingyu
Huang, Wanrong
Xu, Liyang
Yang, Wenjing
contents The real world naturally has dimensions of time and space. Therefore, estimating the counterfactual outcomes with spatial-temporal attributes is a crucial problem. However, previous methods are based on classical statistical models, which still have limitations in performance and generalization. This paper proposes a novel framework for estimating counterfactual outcomes with spatial-temporal attributes using the Transformer, exhibiting stronger estimation ability. Under mild assumptions, the proposed estimator within this framework is consistent and asymptotically normal. To validate the effectiveness of our approach, we conduct simulation experiments and real data experiments. Simulation experiments show that our estimator has a stronger estimation capability than baseline methods. Real data experiments provide a valuable conclusion to the causal effect of conflicts on forest loss in Colombia. The source code is available at https://github.com/lihe-maxsize/DeppSTCI_Release_Version-master.
format Preprint
id arxiv_https___arxiv_org_abs_2506_21154
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Transformer-Based Spatial-Temporal Counterfactual Outcomes Estimation
Li, He
Chi, Haoang
Liu, Mingyu
Huang, Wanrong
Xu, Liyang
Yang, Wenjing
Methodology
Artificial Intelligence
Machine Learning
The real world naturally has dimensions of time and space. Therefore, estimating the counterfactual outcomes with spatial-temporal attributes is a crucial problem. However, previous methods are based on classical statistical models, which still have limitations in performance and generalization. This paper proposes a novel framework for estimating counterfactual outcomes with spatial-temporal attributes using the Transformer, exhibiting stronger estimation ability. Under mild assumptions, the proposed estimator within this framework is consistent and asymptotically normal. To validate the effectiveness of our approach, we conduct simulation experiments and real data experiments. Simulation experiments show that our estimator has a stronger estimation capability than baseline methods. Real data experiments provide a valuable conclusion to the causal effect of conflicts on forest loss in Colombia. The source code is available at https://github.com/lihe-maxsize/DeppSTCI_Release_Version-master.
title Transformer-Based Spatial-Temporal Counterfactual Outcomes Estimation
topic Methodology
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2506.21154