Causal ML Auto-Inference
PyTorch implementation of the Farrell-Liang-Misra framework for automatic inference on individual heterogeneity in treatment effects.
Why it matters: Enables practitioners to estimate heterogeneous treatment effects with valid inference using deep learning, without manually deriving influence functions.
Problem
- Estimating individual-level heterogeneity in treatment effects requires combining ML with causal inference
- Computing valid standard errors for CATE estimates needs influence function calculations
- Automatic differentiation can simplify gradient computation for influence functions
Approach
- Implements the framework from 'Deep Learning for Individual Heterogeneity: An Automatic Inference Framework' (Farrell, Liang, Misra)
- Uses PyTorch for building and training neural networks with MSE loss
- Leverages automatic differentiation to compute gradients for the influence function estimator
Highlights / Evidence
- Estimates Conditional Average Treatment Effects (CATE) using deep learning
- Illustrates estimation and inference for the average treatment effect estimator
- Based on arXiv:2010.14694
How to Run
git clone https://github.com/rmmomin/causal-ml-auto-inference.git
cd causal-ml-auto-inference
pip install -r requirements.txt
Requires Python 3.7+, PyTorch 1.7+. See README for full documentation.