Partial Identification of Policy-Relevant Treatment Effects with Instrumental Variables via Optimal Transport
Jiyuan Tan, Jose Blanchet, Vasilis Syrgkanis
TLDR
This paper uses optimal transport to derive sharper bounds for policy-relevant treatment effects, improving identification with instrumental variables.
Key contributions
- Formulates PRTE partial identification as a Constrained Conditional Optimal Transport (CCOT) problem.
- Reduces complex CCOT to separable one-dimensional optimal transport problems for sharp, closed-form bounds.
- Develops DML for discrete instruments and characterizes nonparametric rates for continuous instruments.
- Achieves substantially tighter bounds for policy-relevant treatment effects than prior moment-relaxation methods.
Why it matters
Policy-Relevant Treatment Effects are crucial for evidence-based policy but often hard to identify. This paper offers a novel, more powerful approach using optimal transport, leading to significantly tighter and more reliable bounds. This advancement improves the precision of causal inference in policy evaluation.
Original Abstract
Policy-Relevant Treatment Effects (PRTEs) are generally not point-identified under standard instrumental variable (IV) assumptions when the instrument generates limited support in treatment propensity. Existing approaches typically optimize over marginal treatment response functions subject to moment restrictions and can discard identifying distributional information. We show that PRTE partial identification in the generalized Roy model can instead be formulated as a Constrained Conditional Optimal Transport (CCOT) problem. The resulting multidimensional CCOT problem reduces analytically to separable one-dimensional OT problems with product costs, yielding sharp closed-form bounds and avoiding direct solution of the original high-dimensional CCOT problem. We also develop estimation and inference procedures for these bounds: for discrete instruments, a Double Machine Learning (DML) approach based on Neyman-orthogonal scores that accommodates high-dimensional covariates while achieving the parametric $\sqrt{n}$ rate and asymptotic normality; for continuous instruments, we explicitly characterize the corresponding nonparametric convergence rates. The framework accommodates covariates, discrete and continuous instruments, and extensions to general treatment settings. In simulations and a bed-net subsidy application, the resulting bounds are substantially tighter than existing moment-relaxation methods.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.