Statistical Inference Under Constrained Selection Bias
Santiago Cortes-Gomez, Mateo Dulce-Rubio, Carlos Miguel Patiño, Bryan Wilder
Abstract
Large-scale datasets are increasingly being used to inform decision making. While this effort aims to ground policy in real-world evidence, challenges have arisen as selection bias and other forms of distribution shifts often plague observational data. Previous attempts to provide robust inference have given guarantees depending on a user-specified amount of possible distribution shift (e.g., the maximum KL divergence between the observed and target distributions). However, decision makers will often have additional knowledge about the target distribution which constrains the kind of possible shifts. To leverage such information, we propose a framework that enables statistical inference in the presence of selection bias which obeys user-specified constraints in the form of functions whose expectation is known under the target distribution. The output is high-probability bounds on the value of an estimand for the target distribution. Hence, our method leverages domain knowledge in order to partially identify a wide class of estimands. We analyze the computational and statistical properties of methods to estimate these bounds and show that our method can produce informative bounds on a variety of simulated and semisynthetic tasks, as well as in a real-world use case.
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 094ab8f2-e9a6-40ca-b18c-b50b6a3168a5Cited by top-tier papers2
- Domain constraints improve risk prediction when outcome data is missingSidhika Balachandar, Nikhil Garg, Emma PiersonICLR 2024 · 11 citations
- Conformal Mixed-Integer Constraint Learning with Feasibility GuaranteesDaniel Ovalle, Lorenz T. Biegler, Ignacio E. Grossmann, Carl D. Laird et al.NeurIPS 2025 · 2 citations
Builds on4
- Retiring Adult: New Datasets for Fair Machine LearningFrances Ding, Moritz Hardt, John Miller, Ludwig SchmidtNeurIPS 2021 · 671 citations
- A Generative Adversarial Framework for Bounding Confounded Causal EffectsYaowei Hu, Yongkai Wu, Lu Zhang, Xintao WuAAAI 2021 · 32 citations
- Partial Identification of Treatment Effects with Implicit Generative ModelsVahid Balazadeh Meresht, Vasilis Syrgkanis, Rahul G. KrishnanNeurIPS 2022 · 25 citations
- The s-value: evaluating stability with respect to distributional shiftsSuyash Gupta, Dominik RothenhäuslerNeurIPS 2023 · 21 citations
Related papers
- Robust Generalization despite Distribution Shift via Minimum Discriminating InformationTobias Sutter, Andreas Krause, Daniel KuhnNeurIPS 2021 · 13 citations
- Falsification before Extrapolation in Causal Effect EstimationZeshan M. Hussain, Michael Oberst, Ming-Chieh Shih, David A. SontagNeurIPS 2022 · 11 citations
- What's the Harm? Sharp Bounds on the Fraction Negatively Affected by TreatmentNathan KallusNeurIPS 2022 · 40 citations
- Towards Estimating Bounds on the Effect of Policies under Unobserved ConfoundingAlexis Bellot, Silvia ChiappaNeurIPS 2024 · 6 citations
- Estimation of Bounds on Potential Outcomes For Decision MakingMaggie Makar, Fredrik D. Johansson, John V. Guttag, David A. SontagICML 2020 · 11 citations
