Exchangeable Gaussian Processes for Staggered-Adoption Policy Evaluation

2026-02-24

AI Summary

This study addresses the challenge of estimating causal effects in panel data with staggered policy adoption, particularly when there are many treated units and nonlinear time trends. The authors propose a novel approach based on exchangeable multi-task Gaussian processes, which models the joint evolution of outcomes for treated and control units through a Gaussian process prior. This framework enforces exchangeability across units while flexibly capturing complex, nonlinear temporal dynamics, thereby enabling robust counterfactual prediction. By innovatively integrating exchangeable Gaussian processes into a causal inference setting, the method supports both pointwise and cumulative treatment effect estimation with principled uncertainty quantification. Extensive simulations and empirical analyses demonstrate its superior predictive accuracy and well-calibrated uncertainty intervals, confirming its effectiveness and adaptability in complex staggered treatment designs.

Technology Category

• Machine Learning: Causal Learning
• Reasoning under Uncertainty: Causality
• Multiagent Systems: Mechanism Design

Application Category

• Search and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for ranking
• Economics, Online Markets and Human Computation: Incentives in network design for Web infrastructures and ecosystems
• User Modeling, Personalization and Recommendation: Explainable and interpretable methods for personalization

Abstract

We study the use of exchangeable multi-task Gaussian processes (GPs) for causal inference in panel data, applying the framework to two settings: one with a single treated unit subject to a once-and-for-all treatment and another with multiple treated units and staggered treatment adoption. Our approach models the joint evolution of outcomes for treated and control units through a GP prior that ensures exchangeability across units while allowing for flexible nonlinear trends over time. The resulting posterior predictive distribution for the untreated potential outcomes of the treated unit provides a counterfactual path, from which we derive pointwise and cumulative treatment effects, along with credible intervals to quantify uncertainty. We implement several variations of the exchangeable GP model using different kernel functions. To assess prediction accuracy, we conduct a placebo-style validation within the pre-intervention window by selecting a ``fake''intervention date. Ultimately, this study illustrates how exchangeable GPs serve as a flexible tool for policy evaluation in panel data settings and proposes a novel approach to staggered-adoption designs with a large number of treated and control units.

Problem

• staggered-adoption
• policy evaluation
• causal inference
• panel data
• counterfactual

Innovation

• exchangeable Gaussian processes
• staggered adoption
• causal inference
• panel data
• counterfactual prediction