# Forests for Differences: Robust Causal Inference Beyond Parametric DiD

arXiv: https://arxiv.org/abs/2505.09706 · HTML: https://mlsanity.com/paper/2505.09706/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 20 of its 87 references that are in the map.
- First posted on arXiv 14 May 2025.

## Builds on (most cited first, 20 of 20)

- [Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learning](https://mlsanity.com/paper/1706.03461/) (2017)
- [Generalized Random Forests](https://mlsanity.com/paper/1610.01271/) (2016)
- [Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithms](https://mlsanity.com/paper/2101.10943/) (2021)
- [Discussion on "Bayesian Regression Tree Models for Causal Inference: Regularization, Confounding, and Heterogeneous Effects" by Hahn, Murray and Carvalho](https://mlsanity.com/paper/2108.02836/) (2021)
- [Generalized Optimal Matching Methods for Causal Inference](https://mlsanity.com/paper/1612.08321/) (2016)
- [DoubleML -- An Object-Oriented Implementation of Double Machine Learning in Python](https://mlsanity.com/paper/2104.03220/) (2021)
- [Design-based Analysis in Difference-In-Differences Settings with Staggered Adoption](https://mlsanity.com/paper/1808.05293/) (2018)
- [Stochastic tree ensembles for regularized nonlinear regression](https://mlsanity.com/paper/2002.03375/) (2020)
- [Regularization and confounding in linear regression for treatment effect estimation](https://mlsanity.com/paper/1602.02176/) (2016)
- [Prediction under interventions: evaluation of counterfactual performance using longitudinal observational data](https://mlsanity.com/paper/2304.10005/) (2023)
- [Stochastic Tree Ensembles for Estimating Heterogeneous Effects](https://mlsanity.com/paper/2209.06998/) (2022)
- [Regression Discontinuity Designs](https://mlsanity.com/paper/2108.09400/) (2021)
- [Estimating and evaluating counterfactual prediction models](https://mlsanity.com/paper/2308.13026/) (2023)
- [Trade-off Between Dependence and Complexity for Nonparametric Learning -- an Empirical Process Approach](https://mlsanity.com/paper/2401.08978/) (2024)
- [Universal Difference-in-Differences for Causal Inference in Epidemiology](https://mlsanity.com/paper/2302.00840/) (2023)
- [Fairness Implications of Heterogeneous Treatment Effect Estimation with Machine Learning Methods in Policy-making](https://mlsanity.com/paper/2309.00805/) (2023)
- [Connecting Instrumental Variable methods for causal inference to the Estimand Framework](https://mlsanity.com/paper/2012.03786/) (2020)
- [Nonparametric Estimation via Partial Derivatives](https://mlsanity.com/paper/2209.07672/) (2022)
- [Modified BART for Learning Heterogeneous Effects in Regression Discontinuity Designs](https://mlsanity.com/paper/2407.14365/) (2024)
- [Ablation Studies for Novel Treatment Effect Estimation Models](https://mlsanity.com/paper/2410.15560/) (2024)

## Abstract (arXiv)

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.