Forests for Differences: Robust Causal Inference Beyond Parametric DiD

2025-05-14

AI Summary

This paper addresses the failure of conventional difference-in-differences (DiD) estimators under staggered treatment adoption and heterogeneous treatment effects. We propose DiD-BCF—a novel framework that reparameterizes the parallel trends assumption as an identifiable nonlinear function, thereby relaxing standard linearity and homogeneity restrictions. Integrating Bayesian Causal Forests (BCF), nonparametric modeling, and the DiD structure, DiD-BCF employs MCMC-based posterior inference and tree ensembles to deliver unified, robust estimation of the average treatment effect (ATE), group-level ATE (GATE), and conditional ATE (CATE). Simulations demonstrate substantial gains over leading DiD methods in settings with nonlinearity, selection bias, and strong heterogeneity. Empirically, applying DiD-BCF to U.S. minimum wage policy reveals significant population-size–dependent conditional treatment effects—heterogeneity entirely overlooked by conventional DiD approaches—highlighting the model’s improved identification power and substantive interpretability.

Technology Category

• Reasoning under Uncertainty: Causality
• Machine Learning: Causal Learning
• Search and Optimization: Mixed Discrete/Continuous Search

Application Category

• User Modeling, Personalization and Recommendation: Attacks and countermeasures in recommendation systems
• Economics, Online Markets and Human Computation: Fairness, privacy, and diversity in economic environments
• Search and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for ranking

Abstract

This paper introduces the Difference-in-Differences Bayesian Causal Forest (DiD-BCF), a novel non-parametric model addressing key challenges in DiD estimation, such as staggered adoption and heterogeneous treatment effects. DiD-BCF provides a unified framework for estimating Average (ATE), Group-Average (GATE), and Conditional Average Treatment Effects (CATE). A core innovation, its Parallel Trends Assumption (PTA)-based reparameterization, enhances estimation accuracy and stability in complex panel data settings. Extensive simulations demonstrate DiD-BCF's superior performance over established benchmarks, particularly under non-linearity, selection biases, and effect heterogeneity. Applied to U.S. minimum wage policy, the model uncovers significant conditional treatment effect heterogeneity related to county population, insights obscured by traditional methods. DiD-BCF offers a robust and versatile tool for more nuanced causal inference in modern DiD applications.

Problem

• Addresses staggered adoption in DiD estimation
• Estimates ATE, GATE, and CATE effects
• Handles non-linearity and selection biases

Innovation

• Non-parametric DiD-BCF model for causal inference
• PTA-based reparameterization enhances estimation accuracy
• Estimates ATE, GATE, CATE in unified framework