Proximal Causal Learning with Kernels: Two-Stage Estimation and Moment Restriction
Afsaneh Mastouri, Yuchen Zhu, Limor Gultchin, Anna Korba, Ricardo Silva, Matt J. Kusner, Arthur Gretton, Krikamol Muandet
Abstract
We address the problem of causal effect estimation in the presence of unobserved confounding, but where proxies for the latent confounder(s) are observed. We propose two kernel-based methods for nonlinear causal effect estimation in this setting: (a) a two-stage regression approach, and (b) a maximum moment restriction approach. We focus on the proximal causal learning setting, but our methods can be used to solve a wider class of inverse problems characterised by a Fredholm integral equation. In particular, we provide a unifying view of two-stage and moment restriction approaches for solving this problem in a nonlinear setting. We provide consistency guarantees for each algorithm, and demonstrate that these approaches achieve competitive results on synthetic data and data simulating a real-world task. In particular, our approach outperforms earlier methods that are not suited to leveraging proxy variables.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b5b6fb9e-81dd-474a-b4d3-1b5737771858Cited by top-tier papers25
- Causal Effect Inference for Structured TreatmentsJean Kaddour, Yuchen Zhu, Qi Liu, Matt J. Kusner et al.NeurIPS 2021 · 62 citations
- Deep Proxy Causal Learning and its Application to Confounded Bandit Policy EvaluationLiyuan Xu, Heishiro Kanagawa, Arthur GrettonNeurIPS 2021 · 52 citations
- A Minimax Learning Approach to Off-Policy Evaluation in Confounded Partially Observable Markov Decision ProcessesChengchun Shi, Masatoshi Uehara, Jiawei Huang, Nan JiangICML 2022 · 31 citations
- Deep Learning Methods for Proximal Inference via Maximum Moment RestrictionBenjamin Kompa, David R. Bellamy, Thomas Kolokotrones, James M. Robins et al.NeurIPS 2022 · 22 citations
- Provably Efficient Offline Reinforcement Learning for Partially Observable Markov Decision ProcessesHongyi Guo, Qi Cai, Yufeng Zhang, Zhuoran Yang et al.ICML 2022 · 17 citations
Builds on2
Related papers
- Density Ratio-Free Doubly Robust Proxy Causal LearningBariscan Bozkurt, Houssam Zenati, Dimitri Meunier, Liyuan Xu et al.NeurIPS 2025 · 2 citations
- A Cross-Moment Approach for Causal Effect EstimationYaroslav Kivva, Saber Salehkaleybar, Negar KiyavashNeurIPS 2023 · 9 citations
- Doubly Robust Proximal Causal Learning for Continuous TreatmentsYong Wu, Yanwei Fu, Shouyan Wang, Xinwei SunICLR 2024 · 4 citations
- Conditional Instrumental Variable Regression with Representation Learning for Causal InferenceDebo Cheng, Ziqi Xu, Jiuyong Li, Lin Liu et al.ICLR 2024 · 14 citations
- Automating the Selection of Proxy Variables of Unmeasured ConfoundersFeng Xie, Zhengming Chen, Shanshan Luo, Wang Miao et al.ICML 2024 · 5 citations
