# Counterfactual Prediction with Deep Instrumental Variables Networks

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

## Key facts

- Cited by 5 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 30 Dec 2016.

## Built on by (most cited first, 5 of 5)

- [Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured Proxies](https://mlsanity.com/paper/2203.09672/) (2022)
- [Stochastic Online Instrumental Variable Regression: Regrets for Endogeneity and Bandit Feedback](https://mlsanity.com/paper/2302.09357/) (2023)
- [A framework for massive scale personalized promotion](https://mlsanity.com/paper/2108.12100/) (2021)
- [Learning to Ignore: Fair and Task Independent Representations](https://mlsanity.com/paper/2101.04047/) (2021)
- [Doubly Robust Proxy Causal Learning with Neural Mean Embeddings](https://mlsanity.com/paper/2605.09514/) (2026)

## Abstract (arXiv)

We are in the middle of a remarkable rise in the use and capability of artificial intelligence. Much of this growth has been fueled by the success of deep learning architectures: models that map from observables to outputs via multiple layers of latent representations. These deep learning algorithms are effective tools for unstructured prediction, and they can be combined in AI systems to solve complex automated reasoning problems. This paper provides a recipe for combining ML algorithms to solve for causal effects in the presence of instrumental variables -- sources of treatment randomization that are conditionally independent from the response. We show that a flexible IV specification resolves into two prediction tasks that can be solved with deep neural nets: a first-stage network for treatment prediction and a second-stage network whose loss function involves integration over the conditional treatment distribution. This Deep IV framework imposes some specific structure on the stochastic gradient descent routine used for training, but it is general enough that we can take advantage of off-the-shelf ML capabilities and avoid extensive algorithm customization. We outline how to obtain out-of-sample causal validation in order to avoid over-fit. We also introduce schemes for both Bayesian and frequentist inference: the former via a novel adaptation of dropout training, and the latter via a data splitting routine.
