Markov Decision Processes with Continuous Side Information
Not on the map. Why?
Abstract (arXiv)
We consider a reinforcement learning (RL) setting in which the agent interacts with a sequence of episodic MDPs. At the start of each episode the agent has access to some side-information or context that determines the dynamics of the MDP for that episode. Our setting is motivated by applications in healthcare where baseline measurements of a patient at the start of a treatment episode form the context that may provide information about how the patient might respond to treatment decisions. We propose algorithms for learning in such Contextual Markov Decision Processes (CMDPs) under an assumption that the unobserved MDP parameters vary smoothly with the observed context. We also give lower and upper PAC bounds under the smoothness assumption. Because our lower bound has an exponential dependence on the dimension, we consider a tractable linear setting where the context is used to create linear combinations of a finite set of MDPs. For the linear setting, we give a PAC learning algorithm based on KWIK learning techniques.
Key facts. Cited by 56 papers in the map (data through 14 Oct 2026). First posted on arXiv 15 Nov 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 9; 2022: 10; 2023: 6; 2024: 7; 2025: 11; 2026: 12.
56 papers.
Sort:
By area:
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 174
- 2022Evolving Curricula with Regret-Based Environment Designcited by 173
- 2021RL for Latent MDPs: Regret Guarantees and a Lower Boundcited by 70
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 46
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 42
- 2021Self-Paced Context Evaluation for Contextual Reinforcement Learningcited by 25
- 2021TempoRL: Learning When to Actcited by 25
- 2021Block Contextual MDPs for Continual Learningcited by 25
- 2021Metadata-based Multi-Task Bandits with Bayesian Hierarchical Modelscited by 24
- 2021A Probabilistic Interpretation of Self-Paced Learning with Applications to Reinforcement Learningcited by 22
- 2021CARL: A Benchmark for Contextual and Adaptive Reinforcement Learningcited by 21
- 2023Domain Randomization via Entropy Maximizationcited by 19
38 more
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculumcited by 17
- 2024Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglementcited by 17
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalizationcited by 16
- 2019Inverse Reinforcement Learning in Contextual MDPscited by 12
- 2022Optimism in Face of a Context: Regret Guarantees for Stochastic Contextual MDPcited by 11
- 2022Non-Linear Reinforcement Learning in Large Action Spaces: Structural Conditions and Sample-efficiency of Posterior Samplingcited by 10
- 2022Eluder-based Regret for Stochastic Contextual MDPscited by 9
- 2023Efficient Rate Optimal Regret for Adversarial Contextual MDPs Using Online Function Approximationcited by 9
- 2023Social Contract AI: Aligning AI Assistants with Implicit Group Normscited by 9
- 2024Offline Oracle-Efficient Learning for Contextual MDPs via Layerwise Exploration-Exploitation Tradeoffcited by 8
- 2025Zero-Shot Adaptation of Behavioral Foundation Models to Unseen Dynamicscited by 8
- 2024Proximal Curriculum with Task Correlations for Deep Reinforcement Learningcited by 7
- 2024Model-Based Transfer Learning for Contextual Reinforcement Learningcited by 7
- 2024Sample Complexity Characterization for Linear Contextual MDPscited by 5
- 2024Inferring Behavior-Specific Context Improves Zero-Shot Generalization in Reinforcement Learningcited by 5
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalizationcited by 5
- 2025Regret Bounds for Adversarial Contextual Bandits with General Function Approximation and Delayed Feedbackcited by 5
- 2022Multi-User Reinforcement Learning with Low Rank Rewardscited by 4
- 2022Hyperparameters in Contextual RL are Highly Situationalcited by 4
- 2025Can Context Bridge the Reality Gap? Sim-to-Real Transfer of Context-Aware Policiescited by 4
- 2022Tiered Reinforcement Learning: Pessimism in the Face of Uncertainty and Constant Regretcited by 3
- 2022Provably Efficient Lifelong Reinforcement Learning with Linear Function Approximationcited by 2
- 2025Pessimism Principle Can Be Effective: Towards a Framework for Zero-Shot Transfer Reinforcement Learningcited by 2
- 2025When Is Diversity Rewarded in Cooperative Multi-Agent Learning?cited by 2
- 2025Learning to Decide with Just Enough: Information-Theoretic Context Summarization for CMDPscited by 2
- 2025Curriculum Design for Trajectory-Constrained Agent: Compressing Chain-of-Thought Tokens in LLMscited by 2
- 2026Structure Detection for Contextual Reinforcement Learningcited by 2
- 2026Near-Optimal Regret for Policy Optimization in Contextual MDPs with General Offline Function Approximationcited by 2
- 2023Reinforcement Learning under External Influence: Guarantees, Algorithms, and Sample Complexitycited by 1
- 2023Scaling Distributed Multi-task Reinforcement Learning with Experience Sharingcited by 1
- 2023JoinGym: An Efficient Query Optimization Environment for Reinforcement Learningcited by 1
- 2025On the Fly Adaptation of Behavior Tree-Based Policies through Reinforcement Learningcited by 1
- 2026Contextual Intelligence The Next Leap for Reinforcement Learningcited by 1
- 2026Contextual Multi-Task Reinforcement Learning for Autonomous Reef Monitoringcited by 1
- 2025Multi-Objective Planning with Contextual Lexicographic Reward Preferencescited by 0
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contextscited by 0
- 2026TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learningcited by 0
- 2026C-IDS: Solving Contextual POMDP via Information-Directed Objectivecited by 0
50 most cited of 56 All 56 (JSON)
- 2026Dynamic Context Scheduling: Learning Beyond the Static Universe21 Aug 2026
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalization18 Jun 2026
- 2026Trajectory-Level Data Augmentation for Offline Reinforcement Learning13 May 2026
- 2026Task-specific Subnetwork Discovery in Reinforcement Learning for Autonomous Underwater Navigation23 Apr 2026
- 2026Contextual Multi-Task Reinforcement Learning for Autonomous Reef Monitoring14 Apr 2026
- 2026Contextual Intelligence The Next Leap for Reinforcement Learning17 Feb 2026
- 2026Near-Optimal Regret for Policy Optimization in Contextual MDPs with General Offline Function Approximation14 Feb 2026
- 2026Adaptive Exploration for Latent-State Bandits4 Feb 2026
- 2026C-IDS: Solving Contextual POMDP via Information-Directed Objective3 Feb 2026
- 2026TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning2 Feb 2026
- 2026Structure Detection for Contextual Reinforcement Learning13 Jan 2026
38 more
- 2025Can Context Bridge the Reality Gap? Sim-to-Real Transfer of Context-Aware Policies6 Nov 2025
- 2025Curriculum Design for Trajectory-Constrained Agent: Compressing Chain-of-Thought Tokens in LLMs4 Nov 2025
- 2025Regret Bounds for Adversarial Contextual Bandits with General Function Approximation and Delayed Feedback10 Oct 2025
- 2025Learning to Decide with Just Enough: Information-Theoretic Context Summarization for CMDPs2 Oct 2025
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalization27 Aug 2025
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contexts10 Jul 2025
- 2025When Is Diversity Rewarded in Cooperative Multi-Agent Learning?11 Jun 2025
- 2025Pessimism Principle Can Be Effective: Towards a Framework for Zero-Shot Transfer Reinforcement Learning24 May 2025
- 2025Zero-Shot Adaptation of Behavioral Foundation Models to Unseen Dynamics19 May 2025
- 2025On the Fly Adaptation of Behavior Tree-Based Policies through Reinforcement Learning8 Mar 2025
- 2025Multi-Objective Planning with Contextual Lexicographic Reward Preferences13 Feb 2025
- 2024Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglement15 Oct 2024
- 2024Model-Based Transfer Learning for Contextual Reinforcement Learning8 Aug 2024
- 2024Offline Oracle-Efficient Learning for Contextual MDPs via Layerwise Exploration-Exploitation Tradeoff28 May 2024
- 2024Proximal Curriculum with Task Correlations for Deep Reinforcement Learning3 May 2024
- 2024Inferring Behavior-Specific Context Improves Zero-Shot Generalization in Reinforcement Learning15 Apr 2024
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalization16 Mar 2024
- 2024Sample Complexity Characterization for Linear Contextual MDPs5 Feb 2024
- 2023Domain Randomization via Entropy Maximization3 Nov 2023
- 2023Social Contract AI: Aligning AI Assistants with Implicit Group Norms26 Oct 2023
- 2023JoinGym: An Efficient Query Optimization Environment for Reinforcement Learning21 Jul 2023
- 2023Scaling Distributed Multi-task Reinforcement Learning with Experience Sharing11 Jul 2023
- 2023Reinforcement Learning under External Influence: Guarantees, Algorithms, and Sample Complexity25 May 2023
- 2023Efficient Rate Optimal Regret for Adversarial Contextual MDPs Using Online Function Approximation2 Mar 2023
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculum24 Dec 2022
- 2022Hyperparameters in Contextual RL are Highly Situational21 Dec 2022
- 2022Eluder-based Regret for Stochastic Contextual MDPs27 Nov 2022
- 2022Multi-User Reinforcement Learning with Low Rank Rewards11 Oct 2022
- 2022Optimism in Face of a Context: Regret Guarantees for Stochastic Contextual MDP22 Jul 2022
- 2022Provably Efficient Lifelong Reinforcement Learning with Linear Function Approximation1 Jun 2022
- 2022Tiered Reinforcement Learning: Pessimism in the Face of Uncertainty and Constant Regret25 May 2022
- 2022Non-Linear Reinforcement Learning in Large Action Spaces: Structural Conditions and Sample-efficiency of Posterior Sampling15 Mar 2022
- 2022Evolving Curricula with Regret-Based Environment Design2 Mar 2022
- 2022Contextualize Me -- The Case for Context in Reinforcement Learning9 Feb 2022
- 2021Block Contextual MDPs for Continual Learning13 Oct 2021
- 2021CARL: A Benchmark for Contextual and Adaptive Reinforcement Learning5 Oct 2021
- 2021Metadata-based Multi-Task Bandits with Bayesian Hierarchical Models13 Aug 2021
- 2021TempoRL: Learning When to Act9 Jun 2021
50 newest of 56 All 56 (JSON)
- 2024Proximal Curriculum with Task Correlations for Deep Reinforcement Learningcited by 7
- 2024Model-Based Transfer Learning for Contextual Reinforcement Learningcited by 7
- 2024Sample Complexity Characterization for Linear Contextual MDPscited by 5
- 2024Inferring Behavior-Specific Context Improves Zero-Shot Generalization in Reinforcement Learningcited by 5
- 2025Can Context Bridge the Reality Gap? Sim-to-Real Transfer of Context-Aware Policiescited by 4
- 2025Pessimism Principle Can Be Effective: Towards a Framework for Zero-Shot Transfer Reinforcement Learningcited by 2
- 2026Structure Detection for Contextual Reinforcement Learningcited by 2
- 2025On the Fly Adaptation of Behavior Tree-Based Policies through Reinforcement Learningcited by 1
- 2026Contextual Intelligence The Next Leap for Reinforcement Learningcited by 1
- 2026Contextual Multi-Task Reinforcement Learning for Autonomous Reef Monitoringcited by 1
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contextscited by 0
- 2026Task-specific Subnetwork Discovery in Reinforcement Learning for Autonomous Underwater Navigationcited by 0
12 most cited of 13 All 13 (JSON)
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026Task-specific Subnetwork Discovery in Reinforcement Learning for Autonomous Underwater Navigation23 Apr 2026
- 2026Contextual Multi-Task Reinforcement Learning for Autonomous Reef Monitoring14 Apr 2026
- 2026Contextual Intelligence The Next Leap for Reinforcement Learning17 Feb 2026
- 2026Structure Detection for Contextual Reinforcement Learning13 Jan 2026
- 2025Can Context Bridge the Reality Gap? Sim-to-Real Transfer of Context-Aware Policies6 Nov 2025
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contexts10 Jul 2025
- 2025Pessimism Principle Can Be Effective: Towards a Framework for Zero-Shot Transfer Reinforcement Learning24 May 2025
- 2025On the Fly Adaptation of Behavior Tree-Based Policies through Reinforcement Learning8 Mar 2025
- 2024Model-Based Transfer Learning for Contextual Reinforcement Learning8 Aug 2024
- 2024Proximal Curriculum with Task Correlations for Deep Reinforcement Learning3 May 2024
- 2024Inferring Behavior-Specific Context Improves Zero-Shot Generalization in Reinforcement Learning15 Apr 2024
12 newest of 13 All 13 (JSON)
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 42
- 2024Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglementcited by 17
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalizationcited by 16
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalizationcited by 5
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalization27 Aug 2025
- 2024Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglement15 Oct 2024
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalization16 Mar 2024
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environment12 Apr 2021
- 2026Trajectory-Level Data Augmentation for Offline Reinforcement Learning13 May 2026
- 2019Inverse Reinforcement Learning in Contextual MDPs23 May 2019
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 174
- 2022Evolving Curricula with Regret-Based Environment Designcited by 173
- 2021RL for Latent MDPs: Regret Guarantees and a Lower Boundcited by 70
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 46
- 2021Self-Paced Context Evaluation for Contextual Reinforcement Learningcited by 25
- 2021TempoRL: Learning When to Actcited by 25
- 2021Block Contextual MDPs for Continual Learningcited by 25
- 2021Metadata-based Multi-Task Bandits with Bayesian Hierarchical Modelscited by 24
- 2021A Probabilistic Interpretation of Self-Paced Learning with Applications to Reinforcement Learningcited by 22
- 2021CARL: A Benchmark for Contextual and Adaptive Reinforcement Learningcited by 21
- 2023Domain Randomization via Entropy Maximizationcited by 19
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculumcited by 17
12 most cited of 36 All 36 (JSON)
- 2026Dynamic Context Scheduling: Learning Beyond the Static Universe21 Aug 2026
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalization18 Jun 2026
- 2026Near-Optimal Regret for Policy Optimization in Contextual MDPs with General Offline Function Approximation14 Feb 2026
- 2026Adaptive Exploration for Latent-State Bandits4 Feb 2026
- 2026C-IDS: Solving Contextual POMDP via Information-Directed Objective3 Feb 2026
- 2026TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning2 Feb 2026
- 2025Regret Bounds for Adversarial Contextual Bandits with General Function Approximation and Delayed Feedback10 Oct 2025
- 2025Learning to Decide with Just Enough: Information-Theoretic Context Summarization for CMDPs2 Oct 2025
- 2025When Is Diversity Rewarded in Cooperative Multi-Agent Learning?11 Jun 2025
- 2025Zero-Shot Adaptation of Behavioral Foundation Models to Unseen Dynamics19 May 2025
- 2025Multi-Objective Planning with Contextual Lexicographic Reward Preferences13 Feb 2025
- 2024Offline Oracle-Efficient Learning for Contextual MDPs via Layerwise Exploration-Exploitation Tradeoff28 May 2024
12 newest of 36 All 36 (JSON)