Learning under random distributional shifts
Not on the map. Why?
Abstract (arXiv)
Many existing approaches for generating predictions in settings with distribution shift model distribution shifts as adversarial or low-rank in suitable representations. In various real-world settings, however, we might expect shifts to arise through the superposition of many small and random changes in the population and environment. Thus, we consider a class of random distribution shift models that capture arbitrary changes in the underlying covariate space, and dense, random shocks to the relationship between the covariates and the outcomes. In this setting, we characterize the benefits and drawbacks of several alternative prediction strategies: the standard approach that directly predicts the long-term outcome of interest, the proxy approach that directly predicts a shorter-term proxy outcome, and a hybrid approach that utilizes both the long-term policy outcome and (shorter-term) proxy outcome(s). We show that the hybrid approach is robust to the strength of the distribution shift and the proxy relationship. We apply this method to datasets in two high-impact domains: asylum-seeker assignment and early childhood education. In both settings, we find that the proposed approach results in substantially lower mean-squared error than current approaches.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 20 of its 60 references that are in the map. First posted on arXiv 5 Jun 2023.
Builds on
20 of 60 references.
References · 20
- 2013Intriguing properties of neural networkscited by 5,155
- 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,917
- 2019Invariant Risk Minimizationcited by 1,854
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
- 2020In Search of Lost Domain Generalizationcited by 1,054
- 2019Unlabeled Data Improves Adversarial Robustnesscited by 416
- 2017Wild Patterns: Ten Years After the Rise of Adversarial Machine Learningcited by 414
- 2019Using Pre-Training Can Improve Model Robustness and Uncertaintycited by 373
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 335
- 2020Understanding and Mitigating the Tradeoff Between Robustness and Accuracycited by 149
- 2020Understanding Self-Training for Gradual Domain Adaptationcited by 149
- 2018Anchor regression: heterogeneous data meets causalitycited by 118
8 more
- 2020Self-training Avoids Using Spurious Features Under Domain Shiftcited by 69
- 2020Domain adaptation under structural causal modelscited by 28
- 2020Targeting for long-term outcomescited by 23
- 2021Fundamental Tradeoffs in Distributionally Adversarial Trainingcited by 12
- 2021Dynamic Placement in Refugee Resettlementcited by 7
- 2020Outcome-Driven Dynamic Refugee Assignment with Allocation Balancingcited by 6
- 2022Calibrated inference: statistical inference that accounts for both sampling uncertainty and distributional uncertaintycited by 5
- 2022Distributionally robust and generalizable inferencecited by 3
Built on by
Citing papers per year: 2024: 1; 2025: 1; 2026: 4.
6 papers.
- 2024Beyond Reweighting: On the Predictive Role of Covariate Shift in Effect Generalizationcited by 4
- 2025Optimal Empirical Risk Minimization under Temporal Distribution Shiftscited by 2
- 2026Diagnostics for Individual-Level Prediction Instability in Machine Learning for Healthcarecited by 0
- 2026Robustness of Refugee-Matching Gains to Off-Policy Evaluation Choicescited by 0
- 2026Harnessing non-adversarial robustness in large language modelscited by 0
- 2026Augmented Inverse Hybrid Weighting: Robust Inference under Deterministic and Random Distribution Shiftscited by 0
- 2026Augmented Inverse Hybrid Weighting: Robust Inference under Deterministic and Random Distribution Shifts1 Aug 2026
- 2026Harnessing non-adversarial robustness in large language models28 May 2026
- 2026Robustness of Refugee-Matching Gains to Off-Policy Evaluation Choices25 Apr 2026
- 2026Diagnostics for Individual-Level Prediction Instability in Machine Learning for Healthcare27 Feb 2026
- 2025Optimal Empirical Risk Minimization under Temporal Distribution Shifts17 Jul 2025
- 2024Beyond Reweighting: On the Predictive Role of Covariate Shift in Effect Generalization12 Dec 2024
Similar, not cited
Close in wording, no citation either way
- 2023Foundation Model's Embedded Representations May Detect Distribution Shift
- 2026A Unified Causal-Origin Taxonomy of Distributional Shifts in Reinforcement Learning
- 2025Recursive Training Loops in LLMs: How training data properties modulate distribution shift in generated data?
- 2024Mitigating Covariate Shift in Misspecified Regression with Applications to Reinforcement Learning
- 2024Assessing the Impact of Distribution Shift on Reinforcement Learning Performance
- 2024Mitigating Distribution Shift in Model-based Offline RL via Shifts-aware Reward Learning
- 2024Distributionally Robust Policy Evaluation under General Covariate Shift in Contextual Bandits
- 2025Distribution Shift Alignment Helps LLMs Simulate Survey Response Distributions