Estimation of Local Average Treatment Effect by Data Combination

Authors

  • Kazuhiko Shinoda Graduate School of Economics, Keio University
  • Takahiro Hoshino Faculty of Economics, Keio University RIKEN AIP

DOI:

https://doi.org/10.1609/aaai.v36i8.20804

Keywords:

Machine Learning (ML)

Abstract

It is important to estimate the local average treatment effect (LATE) when compliance with a treatment assignment is incomplete. The previously proposed methods for LATE estimation required all relevant variables to be jointly observed in a single dataset; however, it is sometimes difficult or even impossible to collect such data in many real-world problems for technical or privacy reasons. We consider a novel problem setting in which LATE, as a function of covariates, is nonparametrically identified from the combination of separately observed datasets. For estimation, we show that the direct least squares method, which was originally developed for estimating the average treatment effect under complete compliance, is applicable to our setting. However, model selection and hyperparameter tuning for the direct least squares estimator can be unstable in practice since it is defined as a solution to the minimax problem. We then propose a weighted least squares estimator that enables simpler model selection by avoiding the minimax objective formulation. Unlike the inverse probability weighted (IPW) estimator, the proposed estimator directly uses the pre-estimated weight without inversion, avoiding the problems caused by the IPW methods. We demonstrate the effectiveness of our method through experiments using synthetic and real-world datasets.

Downloads

Published

2022-06-28

How to Cite

Shinoda, K., & Hoshino, T. (2022). Estimation of Local Average Treatment Effect by Data Combination. Proceedings of the AAAI Conference on Artificial Intelligence, 36(8), 8295-8303. https://doi.org/10.1609/aaai.v36i8.20804

Issue

Section

AAAI Technical Track on Machine Learning III