Learning Robust Decision Policies from Observational Data
Muhammad Osama, Dave Zachariah, Peter Stoica
Abstract
We address the problem of learning a decision policy from observational data of past decisions in contexts with features and associated outcomes. The past policy maybe unknown and in safety-critical applications, such as medical decision support, it is of interest to learn robust policies that reduce the risk of outcomes with high costs. In this paper, we develop a method for learning policies that reduce tails of the cost distribution at a specified level and, moreover, provide a statistically valid bound on the cost of each decision. These properties are valid under finite samples -- even in scenarios with uneven or no overlap between features for different decisions in the observed data -- by building on recent results in conformal prediction. The performance and statistical properties of the proposed method are illustrated using both real and synthetic data.
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 papers2
- Conformal Off-Policy Prediction in Contextual BanditsMuhammad Faaiz Taufiq, Jean-Francois Ton, Rob Cornish, Yee Whye Teh et al.NeurIPS 2022 · 34 citations
- Doubly Robust Fusion of Many Treatments for Policy LearningKe Zhu, Jianing Chu, Ilya Lipkovich, Wenyu Ye et al.ICML 2025
Builds on1
Related papers
- Calibrating Decision Robustness via Inverse Conformal Risk ControlWenbin Zhou, Shixiang ZhuICML 2026
- Optimal Decision-Making Based on Prediction SetsTao Wang, Edgar DobribanICML 2026
- Towards Safe Policy Learning under Partial Identifiability: A Causal ApproachShalmali Joshi, Junzhe Zhang, Elias BareinboimAAAI 2024 · 10 citations
- Utility-Directed Conformal Prediction: A Decision-Aware Framework for Actionable Uncertainty QuantificationSantiago Cortes-Gomez, Carlos Miguel Patiño, Yewon Byun, Steven Wu et al.ICLR 2025
- Conformal Policy ControlDrew Prinster, Clara Fannjiang, Ji Won Park, Kyunghyun Cho et al.ICML 2026 · 3 citations
