1
Return

DML-Geo: an ensemble double machine learning framework for estimating spatially heterogeneous causal effects

delete2026-06-05
delete0
PRE
AI
P
Pengfei Chen
M
Mengjie Gong
J
Jingyu Wang
Y
Yixian Cai
Y
Yiliang Wan *
DOI:10.1080/13658816.2026.2680024delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Causal inference in geographical sciences faces the challenge of isolating treatment effects from high-dimensional observational data, complicated by spatial non-stationarity and persistent confounding. Double machine learning (DML) offers a powerful solution for high-dimensional debiasing through orthogonalization and cross-fitting, but traditional variants overlook spatial heterogeneity by treating space as a simple covariate. To address this, we introduce DML-Geo, an ensemble extension of DML for estimating spatially varying causal effects. Retaining the orthogonalization procedure of DML at its first stage, DML-Geo augments the second stage with three complementary estimators, namely a linear regression model for covariate-driven effects, a generalized additive model (GAM) for spatially smoothed additive effects, and geographically weighted regression (GWR) for localized patterns. Robustness is further enhanced by an adaptive weighting scheme based on inter-model correlations to aggregate outputs from these variants, complemented by a bootstrap procedure for significance testing. Extensive simulations confirm DML-Geo’s superior precision and stability relative to its component models and competing baselines. In real-world applications to housing prices and mental health outcomes, DML-Geo uncovers interpretable spatial causal effect patterns, offering place-specific insights to support policy decisions. DML-Geo provides a flexible toolkit for geospatial causal inference that does not require causal graphs or strong structural assumptions.
Keywords:
Double machine learning
spatial heterogeneity
ensemble approach
geographically weighted regression
causal inference

Journal

International Journal of Geographical Information Science cover
International Journal of Geographical Information Science
IF:
5.1
Papers:
2.7K
Citations:
9.3K

Organization

H
hunan normal university
Scholars:
2.4K
Papers: 773
Citations: 0
Cited Papers

Cited Papers

Citing Papers

Citing Papers