Saved in:
Bibliographic Details
Main Authors: Lan, Hui, Chang, Haoge, Dillon, Eleanor, Syrgkanis, Vasilis
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2502.04699
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918000736600064
author Lan, Hui
Chang, Haoge
Dillon, Eleanor
Syrgkanis, Vasilis
author_facet Lan, Hui
Chang, Haoge
Dillon, Eleanor
Syrgkanis, Vasilis
contents We address the problem of estimating heterogeneous treatment effects in panel data, adopting the popular Difference-in-Differences (DiD) framework under the conditional parallel trends assumption. We propose a novel doubly robust meta-learner for the Conditional Average Treatment Effect on the Treated (CATT), reducing the estimation to a convex risk minimization problem involving a set of auxiliary models. Our framework allows for the flexible estimation of the CATT, when conditioning on any subset of variables of interest using generic machine learning. Leveraging Neyman orthogonality, our proposed approach is robust to estimation errors in the auxiliary models. As a generalization to our main result, we develop a meta-learning approach for the estimation of general conditional functionals under covariate shift. We also provide an extension to the instrumented DiD setting with non-compliance. Empirical results demonstrate the superiority of our approach over existing baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2502_04699
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Meta-learner for Heterogeneous Effects in Difference-in-Differences
Lan, Hui
Chang, Haoge
Dillon, Eleanor
Syrgkanis, Vasilis
Machine Learning
We address the problem of estimating heterogeneous treatment effects in panel data, adopting the popular Difference-in-Differences (DiD) framework under the conditional parallel trends assumption. We propose a novel doubly robust meta-learner for the Conditional Average Treatment Effect on the Treated (CATT), reducing the estimation to a convex risk minimization problem involving a set of auxiliary models. Our framework allows for the flexible estimation of the CATT, when conditioning on any subset of variables of interest using generic machine learning. Leveraging Neyman orthogonality, our proposed approach is robust to estimation errors in the auxiliary models. As a generalization to our main result, we develop a meta-learning approach for the estimation of general conditional functionals under covariate shift. We also provide an extension to the instrumented DiD setting with non-compliance. Empirical results demonstrate the superiority of our approach over existing baselines.
title A Meta-learner for Heterogeneous Effects in Difference-in-Differences
topic Machine Learning
url https://arxiv.org/abs/2502.04699