Learning Overlapping Representations for the Estimation of Individualized Treatment Effects

Yao Zhang, Alexis Bellot, Mihaela Schaar
Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics, PMLR 108:1005-1014, 2020.

Abstract

The choice of making an intervention depends on its potential benefit or harm in comparison to alternatives. Estimating the likely outcome of alternatives from observational data is a challenging problem as all outcomes are never observed, and selection bias precludes the direct comparison of differently intervened groups. Despite their empirical success, we show that algorithms that learn domain-invariant representations of inputs (on which to make predictions) are often inappropriate, and develop generalization bounds that demonstrate the dependence on domain overlap and highlight the need for invertible latent maps. Based on these results, we develop a deep kernel regression algorithm and posterior regularization framework that substantially outperforms the state-of-the-art on a variety of benchmarks data sets.

Cite this Paper

BibTeX
@InProceedings{pmlr-v108-zhang20c, title = {Learning Overlapping Representations for the Estimation of Individualized Treatment Effects}, author = {Zhang, Yao and Bellot, Alexis and van der Schaar, Mihaela}, booktitle = {Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics}, pages = {1005--1014}, year = {2020}, editor = {Chiappa, Silvia and Calandra, Roberto}, volume = {108}, series = {Proceedings of Machine Learning Research}, month = {26--28 Aug}, publisher = {PMLR}, pdf = {http://proceedings.mlr.press/v108/zhang20c/zhang20c.pdf}, url = {https://proceedings.mlr.press/v108/zhang20c.html}, abstract = {The choice of making an intervention depends on its potential benefit or harm in comparison to alternatives. Estimating the likely outcome of alternatives from observational data is a challenging problem as all outcomes are never observed, and selection bias precludes the direct comparison of differently intervened groups. Despite their empirical success, we show that algorithms that learn domain-invariant representations of inputs (on which to make predictions) are often inappropriate, and develop generalization bounds that demonstrate the dependence on domain overlap and highlight the need for invertible latent maps. Based on these results, we develop a deep kernel regression algorithm and posterior regularization framework that substantially outperforms the state-of-the-art on a variety of benchmarks data sets.} }
Endnote
%0 Conference Paper %T Learning Overlapping Representations for the Estimation of Individualized Treatment Effects %A Yao Zhang %A Alexis Bellot %A Mihaela Schaar %B Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics %C Proceedings of Machine Learning Research %D 2020 %E Silvia Chiappa %E Roberto Calandra %F pmlr-v108-zhang20c %I PMLR %P 1005--1014 %U https://proceedings.mlr.press/v108/zhang20c.html %V 108 %X The choice of making an intervention depends on its potential benefit or harm in comparison to alternatives. Estimating the likely outcome of alternatives from observational data is a challenging problem as all outcomes are never observed, and selection bias precludes the direct comparison of differently intervened groups. Despite their empirical success, we show that algorithms that learn domain-invariant representations of inputs (on which to make predictions) are often inappropriate, and develop generalization bounds that demonstrate the dependence on domain overlap and highlight the need for invertible latent maps. Based on these results, we develop a deep kernel regression algorithm and posterior regularization framework that substantially outperforms the state-of-the-art on a variety of benchmarks data sets.
APA
Zhang, Y., Bellot, A. & Schaar, M.. (2020). Learning Overlapping Representations for the Estimation of Individualized Treatment Effects. Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics, in Proceedings of Machine Learning Research 108:1005-1014 Available from https://proceedings.mlr.press/v108/zhang20c.html.

Related Material