A Fixed-Effects Causal Forest for Staggered Adoption, with an Application to Medicaid Expansion

• URL: http://arxiv.org/abs/2607.19644v1
• Date: Wed, 22 Jul 2026 00:53:27 GMT
• Title: A Fixed-Effects Causal Forest for Staggered Adoption, with an Application to Medicaid Expansion
• Abstract summary: We estimate the tau_g,t(x) object with a fixed-effects causal forest.<n>Difference-in-differences with staggered adoption identifies group-time average treatment effects ATT(g,t)<n>We apply the method to the Callaway-Sant'Anna minimum-wage panel as a validation and to the staggered county-level rollout of the ACA Medicaid expansion.
• Score: 0.0
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Difference-in-differences with staggered adoption identifies group-time average treatment effects ATT(g,t) by comparing each cohort to units not yet treated, which avoids the "forbidden comparisons" that bias two-way fixed-effects estimators when effects are heterogeneous. This paper studies the covariate-conditional version of that object, tau_{g,t}(x), and estimates it with a fixed-effects causal forest. Within each (g,t) comparison block, the outcome and treatment are residualized on unit and period fixed effects inside each tree node, and honest causal trees split on treatment-effect heterogeneity in the covariates. The estimand is not new: Hatamyar, Kreif, Rocha and Huber (2023) introduced it using a doubly-robust R-learner, and Imai, Qin and Yanagi (2023) study it for a single continuous covariate. What we add is a different way to estimate it. Where those methods remove confounding by modeling nuisance functions, we remove it by differencing out unit and period effects within each tree node, following the fixed-effects residualization of Kattenberg, Scheer and Thiel (2023) and Gavrilova, Langorgen and Zoutman (2025) and carrying it into the Callaway-Sant'Anna group-time structure. In Monte Carlo experiments the estimator is the only forest-based method that stays unbiased and correctly covered for the overall effect under staggered timing with cohort-varying effects; two-way fixed effects and a pooled causal forest inherit large forbidden-comparison bias. We apply the method to the Callaway-Sant'Anna minimum-wage panel as a validation and to the staggered county-level rollout of the ACA Medicaid expansion, where it recovers an average 2.25 percentage-point fall in the uninsured rate and a conditional surface on which poorer and lower-income counties gained substantially more coverage -- heterogeneity measured along socioeconomic covariates that are not lags of the outcome.

Related papers

• Splitting the Difference: Interpretable Causal Forests for Treatment Effect Heterogeneity and Bias [0.0]
  We introduce an algorithm based on decision trees and random forests for estimating individual treatment effects.<n>Our algorithm is simple: it operates exactly like a standard random forest, but with a different splitting criterion.<n>It handles observational studies with varying treatment propensities without requiring separate estimation of the full propensity function.
  arXiv  Detail & Related papers  (2026-09-15T10:43:57Z)
• Attenuated Heterogeneity in Fixed-Effects Causal Forests, and a Cross-Fitted Correction [0.0]
  Causal forests estimate conditional average treatment effects by averaging honest leaf-level effects across trees.<n>We show that this averaging systematically attenuates the estimated heterogeneity.<n>We adapt the best-linear-predictor calibration of Chernozhukov et al., estimating the de-attenuation slope out-of-bag.
  arXiv  Detail & Related papers  (2026-07-24T20:18:11Z)
• Topological Ignorability for Structural Causal Effects Beyond Means [0.0]
  Interventions alter the structure of an outcome distribution rather than its mean.<n>Mean-based causal estimands such as the average treatment effect may miss important structural effects.<n>We introduce topological-geometrical causal metrics based on summaries of interventional outcome laws.
  arXiv  Detail & Related papers  (2026-05-31T11:56:53Z)
• Calibrated Inference for the Conditional Average Treatment Effect in the Few-Placebo Regime via Gaussian Processes [0.0]
  We study the few-placebo regime, in which one treatment arm is much smaller than the other.<n>We show that intervals under-cover: they contain the true effect less often than their nominal level.<n>We propose GP-CATE, which models each arm's outcome surface with a Gaussian process.
  arXiv  Detail & Related papers  (2026-05-26T11:18:42Z)
• Disentangled Double Machine Learning for Accurate Causal Effect Estimation [80.07136847623498]
  Confounding bias is a key challenge in causal effect estimation from observational data.<n>Double Machine Learning (DML) addresses this issue by estimating treatment and outcome nuisance functions.<n>DML often produces biased and unstable estimates in highdimensional or finite-sample scenarios.<n>We propose Disentangled Double Machine Learning (DDML), a novel algorithm that integrates two key strategies.
  arXiv  Detail & Related papers  (2026-05-24T01:40:43Z)
• Doubly Robust Proxy Causal Learning with Neural Mean Embeddings [58.8808794752741]
  We develop a neural doubly robust framework for proxy causal learning with continuous and structured treatments.<n>The framework covers population, heterogeneous, and conditional dose-response functions, yielding full response-curve estimators.<n>Across synthetic and image-valued benchmarks, the proposed estimators outperform existing baselines and single-bridge neural estimators.
  arXiv  Detail & Related papers  (2026-05-10T12:50:25Z)
• DARTS: Targeting Prognostic Covariates in Budget-Constrained Sequential Experiments [47.61857875238484]
  We introduce Dynamic Adaptive Rerandomization via Thompson Sampling (DARTS)<n>DARTS treats covariate acquisition as a sequential optimization problem embedded within a design-based causal inference task.<n>We derive a Bayes risk bound for the acquisition layer that matches the minimax lower bound up to logarithmic factors.
  arXiv  Detail & Related papers  (2026-05-07T17:27:51Z)
• Unity Forests: Improving Interaction Modelling and Interpretability in Random Forests [45.88028371034407]
  Random forests (RFs) are widely used for prediction and variable importance analysis.<n>In UFOs, the first few splits of each tree are optimized jointly.<n>UFOs achieved higher discrimination and predictive accuracy than standard RFs.
  arXiv  Detail & Related papers  (2026-01-11T17:36:58Z)
• Accounting for Missing Covariates in Heterogeneous Treatment Estimation [17.09751619857397]
  We introduce a novel partial identification strategy based on ideas from ecological inference. We show that our framework can produce bounds that are much tighter than would otherwise be possible.
  arXiv  Detail & Related papers  (2024-10-21T05:47:07Z)
• Clustered Switchback Designs for Experimentation Under Spatio-temporal Interference [44.644520116360106]
  We estimate the global average treatment effect (GATE), the difference between average outcomes having exposed all units at all times to treatment or to control.<n>We propose a clustered switchback design, where units are grouped into clusters and time steps are grouped into blocks.<n>We show that for graphs that admit good clustering, a truncated Horvitz-Thompson estimator achieves a $tilde O(1/NT)$ mean squared error (MSE)<n>Our results simultaneously generalize the results from citethu2022switchback,ugander2013graph and citetleung2022rate
  arXiv  Detail & Related papers  (2023-12-25T01:00:58Z)
• B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confounding [51.74479522965712]
  We propose a meta-learner called the B-Learner, which can efficiently learn sharp bounds on the CATE function under limits on hidden confounding. We prove its estimates are valid, sharp, efficient, and have a quasi-oracle property with respect to the constituent estimators under more general conditions than existing methods.
  arXiv  Detail & Related papers  (2023-04-20T18:07:19Z)
• Robust and Agnostic Learning of Conditional Distributional Treatment Effects [44.31792000298105]
  We provide a new robust and model-agnostic methodology for learning the conditional DTE (CDTE) for a class of problems.<n>Our method is model-agnostic in that it can provide the best projection of CDTE onto the regression model class.<n>We investigate the behavior of our proposal in simulations, as well as in a case study of 401(k) eligibility effects on wealth.
  arXiv  Detail & Related papers  (2022-05-23T17:40:31Z)
• Treatment Effect Risk: Bounds and Inference [58.442274475425144]
  Since the average treatment effect measures the change in social welfare, even if positive, there is a risk of negative effect on, say, some 10% of the population. In this paper we consider how to nonetheless assess this important risk measure, formalized as the conditional value at risk (CVaR) of the ITE distribution. Some bounds can also be interpreted as summarizing a complex CATE function into a single metric and are of interest independently of being a bound.
  arXiv  Detail & Related papers  (2022-01-15T17:21:26Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.