A Unified Causal-Origin Taxonomy of Distributional Shifts in Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Reinforcement learning (RL) systems often degrade when operating conditions differ from those previously encountered, reflecting distributional shifts in the underlying data-generating process. Such shifts may occur between training and evaluation, as in In-Distribution (ID) and Out-of-Distribution (OOD) generalization, or within non-stationary settings where environment dynamics evolve over time. However, the formal relationship between these views remains unclear, and existing work mainly focuses on mitigation rather than the causal origin of shift within the agent-environment interaction. This work develops a unified causal-origin taxonomy that characterizes sources of distributional shift in RL and relates ID/OOD generalization to non-stationary settings. We transfer the classical dataset-shift principle from supervised learning to RL by reformulating distributional shift in terms of the generative interaction process. Using a Partially Observable Markov Decision Process (POMDP), we decompose the interaction into structural components, including the state distribution, observation process, policy, reward, and transition dynamics, together with the shifted-time boundary. The proposed taxonomy distinguishes internal (agent-driven) and external (environment-driven) distributional shifts. The shifted-time boundary perspective further characterizes explicit, implicit, and hybrid shifts. This formulation unifies ID/OOD generalization and non-stationarity as structured changes in the underlying process. We also introduce an evaluation framework for measuring shift impact and adaptation through performance degradation and recovery metrics. By grounding distributional shift in the causal-origin structure of RL, this work supports systematic analysis of robustness under distributional shift.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 39 of its 72 references that are in the map. First posted on arXiv 15 Jun 2026.
Builds on
39 of 72 references.
References · 39
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,211
- 2016Concrete Problems in AI Safetycited by 1,582
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,548
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,274
- 2019Learned Step Size Quantizationcited by 604
- 2018Predictive Uncertainty Estimation via Prior Networkscited by 555
- 2016RL2: Fast Reinforcement Learning via Slow Reinforcement Learningcited by 490
- 2020Learning under Concept Drift: A Reviewcited by 470
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2020Sim-to-Real Transfer in Deep Reinforcement Learning for Robotics: a Surveycited by 388
27 more
- 2018Quantifying Generalization in Reinforcement Learningcited by 345
- 2020Towards Continual Reinforcement Learning: A Review and Perspectivescited by 268
- 2020Benchmarking Multi-Agent Deep Reinforcement Learning Algorithms in Cooperative Taskscited by 207
- 2018A Study on Overfitting in Deep Reinforcement Learningcited by 170
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2018Assessing Generalization in Deep Reinforcement Learningcited by 132
- 2017Continuous Adaptation via Meta-Learning in Nonstationary and Competitive Environmentscited by 106
- 2019Dealing with Non-Stationarity in Multi-Agent Deep Reinforcement Learningcited by 90
- 2020A Survey of Reinforcement Learning Algorithms for Dynamically Varying Environmentscited by 62
- 2023RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalizationcited by 27
- 2022Sample-Efficient Reinforcement Learning of Partially Observable Markov Gamescited by 26
- 2021Taxonomy of Machine Learning Safety: A Survey and Primercited by 22
- 2025A Survey of State Representation Learning for Deep Reinforcement Learningcited by 17
- 2022DROPO: Sim-to-Real Transfer with Offline Domain Randomizationcited by 16
- 2025Handling Out-of-Distribution Data: A Surveycited by 14
- 2019Robust Visual Domain Randomization for Reinforcement Learningcited by 13
- 2025Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learningcited by 12
- 2022Distributionally Adaptive Meta Reinforcement Learningcited by 11
- 2021Dealing with Non-Stationarity in MARL via Trust-Region Decompositioncited by 5
- 2024A Behavior-Aware Approach for Deep Reinforcement Learning in Non-stationary Environments without Known Change Pointscited by 5
- 2025An Extended Benchmarking of Multi-Agent Reinforcement Learning Algorithms in Complex Fully Cooperative Taskscited by 5
- 2024Assessing the Impact of Distribution Shift on Reinforcement Learning Performancecited by 4
- 2025Evaluation-Time Policy Switching for Offline Reinforcement Learningcited by 4
- 2023CROP: Towards Distributional-Shift Robust Reinforcement Learning using Compact Reshaped Observation Processingcited by 3
- 2024Robust Model-Based Reinforcement Learning with an Adversarial Auxiliary Modelcited by 3
- 2023Non-Stationary Policy Learning for Multi-Timescale Multi-Agent Reinforcement Learningcited by 2
- 2024Mitigating Distribution Shift in Model-based Offline RL via Shifts-aware Reward Learningcited by 2
Built on by
Not cited yet.