Optimal Treatment Regimes for Proximal Causal Learning
Tao Shen, Yifan Cui
Abstract
A common concern when a policymaker draws causal inferences from and makes decisions based on observational data is that the measured covariates are insufficiently rich to account for all sources of confounding, i.e., the standard no confoundedness assumption fails to hold. The recently proposed proximal causal inference framework shows that proxy variables that abound in real-life scenarios can be leveraged to identify causal effects and therefore facilitate decision-making. Building upon this line of work, we propose a novel optimal individualized treatment regime based on so-called outcome and treatment confounding bridges. We then show that the value function of this new optimal treatment regime is superior to that of existing ones in the literature. Theoretical guarantees, including identification, superiority, excess value bound, and consistency of the estimated regime, are established. Furthermore, we demonstrate the proposed optimal regime via numerical experiments and a real data application.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on4
- Minimax Estimation of Conditional Moment ModelsNishanth Dikkala, Greg Lewis, Lester Mackey, Vasilis SyrgkanisNeurIPS 2020 · 125 citations
- Proximal Causal Learning with Kernels: Two-Stage Estimation and Moment RestrictionAfsaneh Mastouri, Yuchen Zhu, Limor Gultchin, Anna Korba et al.ICML 2021 · 78 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
- Representation Learning for Integrating Multi-domain Outcomes to Optimize Individualized TreatmentYuan Chen, Donglin Zeng, Tianchen Xu, Yuanjia WangNeurIPS 2020 · 5 citations
Related papers
- Coupling Generative Modeling and an Autoencoder with the Causal BridgeRuolin Meng, Ming-Yu Chung, Dhanajit Brahma, Ricardo Henao et al.NeurIPS 2025
- Off-Policy Evaluation for Episodic Partially Observable Markov Decision Processes under Non-Parametric ModelsRui Miao, Zhengling Qi, Xiaoke ZhangNeurIPS 2022 · 18 citations
- Density Ratio-Free Doubly Robust Proxy Causal LearningBariscan Bozkurt, Houssam Zenati, Dimitri Meunier, Liyuan Xu et al.NeurIPS 2025 · 2 citations
- Efficient and Sharp Off-Policy Learning under Unobserved ConfoundingKonstantin Hess, Dennis Frauen, Valentyn Melnychuk, Stefan FeuerriegelICLR 2026 · 5 citations
- A Proxy Variable View of Shared ConfoundingYixin Wang, David M. BleiICML 2021 · 14 citations
