Contextual Markov Decision Processes
Not on the map. Why?
Abstract (arXiv)
We consider a planning problem where the dynamics and rewards of the environment depend on a hidden static parameter referred to as the context. The objective is to learn a strategy that maximizes the accumulated reward across all contexts. The new model, called Contextual Markov Decision Process (CMDP), can model a customer's behavior when interacting with a website (the learner). The customer's behavior depends on gender, age, location, device, etc. Based on that behavior, the website objective is to determine customer characteristics, and to optimize the interaction between them. Our work focuses on one basic scenario--finite horizon with a small known number of possible contexts. We suggest a family of algorithms with provable guarantees that learn the underlying models and the latent contexts, and optimize the CMDPs. Bounds are obtained for specific naive implementations, and extensions of the framework are discussed, laying the ground for future research.
Key facts. Cited by 213 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Feb 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 19; 2022: 36; 2023: 41; 2024: 38; 2025: 43; 2026: 35.
213 papers.
Sort:
By area:
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environmentcited by 99
- 2023Factually Consistent Summarization via Reinforcement Learning with Textual Entailment Feedbackcited by 71
- 2021RL for Latent MDPs: Regret Guarantees and a Lower Boundcited by 69
- 2024Understanding the Weakness of Large Language Model Agents within a Complex Android Environmentcited by 60
- 2022Exploration via Elliptical Episodic Bonusescited by 51
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 41
- 2021Learning a Large Neighborhood Search Algorithm for Mixed Integer Programscited by 41
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 41
- 2022Efficient Risk-Averse Reinforcement Learningcited by 41
- 2022Automated Dynamic Algorithm Configurationcited by 37
38 more
- 2021CORA: Benchmarks, Baselines, and Metrics as a Platform for Continual Reinforcement Learning Agentscited by 36
- 2023AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agentscited by 36
- 2022The Importance of Non-Markovianity in Maximum State Entropy Explorationcited by 35
- 2024Diversity Empowers Intelligence: Integrating Expertise of Software Engineering Agentscited by 35
- 2022Sequence Model Imitation Learning with Unobserved Contextscited by 34
- 2022Trustworthy Reinforcement Learning Against Intrinsic Vulnerabilities: Robustness, Safety, and Generalizabilitycited by 32
- 2022Model-based RL with Optimistic Posterior Sampling: Structural Conditions and Sample Complexitycited by 31
- 2024Unsupervised Zero-Shot Reinforcement Learning via Functional Reward Encodingscited by 31
- 2022Factored Adaptation for Non-Stationary Reinforcement Learningcited by 30
- 2023On the Importance of Exploration for Generalization in Reinforcement Learningcited by 30
- 2021DACBench: A Benchmark Library for Dynamic Algorithm Configurationcited by 28
- 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
- 2023Seeing is not Believing: Robust Reinforcement Learning against Spurious Correlationcited by 25
- 2023The Generalization Gap in Offline Reinforcement Learningcited by 25
- 2022A Comprehensive Survey of Data Augmentation in Visual Reinforcement Learningcited by 23
- 2023Causal Reinforcement Learning: A Surveycited by 22
- 2021CARL: A Benchmark for Contextual and Adaptive Reinforcement Learningcited by 21
- 2023Universal Morphology Control via Contextual Modulationcited by 21
- 2022Language Control Diffusion: Efficiently Scaling through Space, Time, and Taskscited by 20
- 2022Deciding What to Model: Value-Equivalent Sampling for Reinforcement Learningcited by 19
- 2023A Study of Global and Episodic Bonuses for Exploration in Contextual MDPscited by 19
- 2022Generalization in Cooperative Multi-Agent Systemscited by 18
- 2022Hypernetworks for Zero-shot Transfer in Reinforcement Learningcited by 18
- 2023Dynamics Generalisation in Reinforcement Learning via Adaptive Context-Aware Policiescited by 18
- 2021Unsupervised Reinforcement Learning in Multiple Environmentscited by 17
- 2022Robust Policy Learning over Multiple Uncertainty Setscited by 16
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculumcited by 16
- 2023Structure in Deep Reinforcement Learning: A Survey and Open Problemscited by 16
- 2024LLM-Powered Code Vulnerability Repair with Reinforcement Learning and Semantic Rewardcited by 16
- 2024Towards an Information Theoretic Framework of Context-Based Offline Meta-Reinforcement Learningcited by 16
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalizationcited by 16
- 2021Reinforcement Learning in Reward-Mixing MDPscited by 15
- 2022Learning Mixtures of Linear Dynamical Systemscited by 15
- 2022An Adaptive Deep RL Method for Non-Stationary Environments with Piecewise Stable Contextcited by 15
- 2025NeSyC: A Neuro-symbolic Continual Learner For Complex Embodied Tasks In Open Domainscited by 15
- 2021On Covariate Shift of Latent Confounders in Imitation and Reinforcement Learningcited by 14
50 most cited of 213 All 213 (JSON)
- 2026Optimal Multi-Reward Reinforcement Learning29 Sep 2026
- 2026From Static Policies to Adaptive Priors in Offline Reinforcement Learning26 Sep 2026
- 2026Skill Composition for Legged Robot Reinforcement Learning13 Sep 2026
- 2026Toward Unified Robot Learning: Bridging Representation, Vision-Language-Action, and World Models3 Sep 2026
- 2026Reinforcement learning to choose optimizers1 Sep 2026
- 2026Dynamic Context Scheduling: Learning Beyond the Static Universe21 Aug 2026
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026A Distribution Mapping Approach to Counterfactually Fair Reinforcement Learning9 Aug 2026
- 2026DishSeg24k: A Large-Scale Benchmark for Food Segmentation with Stochastic Expert Decoding25 Jul 2026
- 2026Adjustment Speed as a Safety Constraint for Nonstationary Reinforcement Learning21 Jul 2026
- 2026Building2Building: A Large Scale Benchmark for Generalizable Real-World Reinforcement Learning17 Jul 2026
- 2026Generalization in offline RL: The structure is more important than the amount of pessimism2 Jul 2026
38 more
- 2026Deep Reinforcement Learning for Spacecraft Attitude Control During Atmospheric Re-Entry30 Jun 2026
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalization18 Jun 2026
- 2026Functional Cache Grafting: Robust and Rapid Code-Policy Synthesis for Embodied Agents11 Jun 2026
- 2026ConTraIRL: Factorized Contrastive Abstractions for Transferable IRL2 Jun 2026
- 2026Bilevel Optimization of Synthetic Trajectories for Multi-Turn LLM Fine-Tuning23 May 2026
- 2026Curriculum reinforcement learning with measurable task representation learning22 May 2026
- 2026MATE: Solving Contextual Markov Decision Processes with Memory of Accumulated Transition Embeddings17 May 2026
- 2026Adaptive Estimation and Optimal Control in Offline Contextual MDPs without Stationarity5 May 2026
- 2026Data Deletion Can Help in Adaptive RL30 Apr 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
- 2026Self Paced Gaussian Contextual Reinforcement Learning24 Mar 2026
- 2026NASimJax: A GPU-Accelerated Policy Learning Framework for Penetration Testing20 Mar 2026
- 2026Generalization in Online Reinforcement Learning for Mobile Agents8 Mar 2026
- 2026Beyond Mimicry: Toward Lifelong Adaptability in Imitation Learning23 Feb 2026
- 2026Contextual Intelligence The Next Leap for Reinforcement Learning17 Feb 2026
- 2026Solving Parameter-Robust Avoid Problems with Unknown Feasibility using Reinforcement Learning17 Feb 2026
- 2026Near-Optimal Regret for Policy Optimization in Contextual MDPs with General Offline Function Approximation14 Feb 2026
- 2026Pushing Forward Pareto Frontiers of Proactive Agents with Behavioral Agentic Optimization11 Feb 2026
- 2026Adaptive Exploration for Latent-State Bandits4 Feb 2026
- 2026HoRD: Robust Humanoid Control via History-Conditioned Reinforcement Learning and Online Distillation4 Feb 2026
- 2026C-IDS: Solving Contextual POMDP via Information-Directed Objective3 Feb 2026
- 2026Structure Detection for Contextual Reinforcement Learning13 Jan 2026
- 2025GB-DQN: Gradient Boosted DQN Models for Non-stationary Reinforcement Learning18 Dec 2025
- 2025Knowledge Diversion for Efficient Morphology Control and Policy Transfer10 Dec 2025
- 2025Direct transfer of optimized controllers to similar systems using dimensionless MPC9 Dec 2025
- 2025Robust Agents in Open-Ended Worlds9 Dec 2025
- 2025Deep Reinforcement Learning for Dynamic Algorithm Configuration: A Case Study on Optimizing OneMax with the (1+(λ,λ))-GA3 Dec 2025
- 2025From monoliths to modules: Decomposing transducers for efficient world modelling1 Dec 2025
- 2025Representative Action Selection for Large Action Space: From Bandits to MDPs27 Nov 2025
- 2025Out-of-Distribution Generalization with a SPARC: Racing 100 Unseen Vehicles with a Single Policy12 Nov 2025
- 2025Curriculum Design for Trajectory-Constrained Agent: Compressing Chain-of-Thought Tokens in LLMs4 Nov 2025
- 2025Learning Personalized Ad Impact via Contextual Reinforcement Learning under Delayed Rewards22 Oct 2025
- 2025DAIL: Beyond Task Ambiguity for Language-Conditioned Reinforcement Learning22 Oct 2025
- 2025Multi-Task Reinforcement Learning with Language-Encoded Gated Policy Networks7 Oct 2025
- 2025Improved Robustness of Deep Reinforcement Learning for Control of Time-Varying Systems by Bounded Extremum Seeking2 Oct 2025
- 2025Learning to Decide with Just Enough: Information-Theoretic Context Summarization for CMDPs2 Oct 2025
- 2025MDP modeling for multi-stage stochastic programs26 Sep 2025
50 newest of 213 All 213 (JSON)
- 2022Sequence Model Imitation Learning with Unobserved Contextscited by 34
- 2023The Generalization Gap in Offline Reinforcement Learningcited by 25
- 2024Towards an Information Theoretic Framework of Context-Based Offline Meta-Reinforcement Learningcited by 16
- 2021On Covariate Shift of Latent Confounders in Imitation and Reinforcement Learningcited by 14
- 2019Inverse Reinforcement Learning in Contextual MDPscited by 12
- 2023Delphic Offline Reinforcement Learning under Nonidentifiable Hidden Confoundingcited by 7
- 2021Learning without Knowing: Unobserved Context in Continuous Transfer Reinforcement Learningcited by 4
- 2024Offline Reinforcement Learning with Domain-Unlabeled Datacited by 3
- 2024Offline Reinforcement Learning and Sequence Modeling for Downlink Link Adaptationcited by 3
- 2021Provably Improved Context-Based Offline Meta-RL with Attention and Contrastive Learningcited by 2
- 2023Bootstrapping Adaptive Human-Machine Interfaces with Offline Reinforcement Learningcited by 0
- 2024Offline Imitation of Badminton Player Behavior via Experiential Contexts and Brownian Motioncited by 0
12 most cited of 16 All 16 (JSON)
- 2026From Static Policies to Adaptive Priors in Offline Reinforcement Learning26 Sep 2026
- 2026Generalization in offline RL: The structure is more important than the amount of pessimism2 Jul 2026
- 2026ConTraIRL: Factorized Contrastive Abstractions for Transferable IRL2 Jun 2026
- 2026Beyond Mimicry: Toward Lifelong Adaptability in Imitation Learning23 Feb 2026
- 2024Offline Reinforcement Learning and Sequence Modeling for Downlink Link Adaptation30 Oct 2024
- 2024Offline Reinforcement Learning with Domain-Unlabeled Data11 Apr 2024
- 2024Offline Imitation of Badminton Player Behavior via Experiential Contexts and Brownian Motion19 Mar 2024
- 2024Towards an Information Theoretic Framework of Context-Based Offline Meta-Reinforcement Learning4 Feb 2024
- 2023The Generalization Gap in Offline Reinforcement Learning10 Dec 2023
- 2023Bootstrapping Adaptive Human-Machine Interfaces with Offline Reinforcement Learning7 Sep 2023
- 2023Delphic Offline Reinforcement Learning under Nonidentifiable Hidden Confounding1 Jun 2023
- 2022Sequence Model Imitation Learning with Unobserved Contexts3 Aug 2022
12 newest of 16 All 16 (JSON)
- 2024Understanding the Weakness of Large Language Model Agents within a Complex Android Environmentcited by 60
- 2024Diversity Empowers Intelligence: Integrating Expertise of Software Engineering Agentscited by 35
- 2025Exploring Expert Failures Improves LLM Agent Tuningcited by 7
- 2026Generalization in Online Reinforcement Learning for Mobile Agentscited by 6
- 2025cMALC-D: Contextual Multi-Agent LLM-Guided Curriculum Learning with Diversity-Based Context Blendingcited by 2
- 2026Pushing Forward Pareto Frontiers of Proactive Agents with Behavioral Agentic Optimizationcited by 2
- 2025Robust Agents in Open-Ended Worldscited by 1
- 2026Functional Cache Grafting: Robust and Rapid Code-Policy Synthesis for Embodied Agentscited by 0
- 2026Functional Cache Grafting: Robust and Rapid Code-Policy Synthesis for Embodied Agents11 Jun 2026
- 2026Generalization in Online Reinforcement Learning for Mobile Agents8 Mar 2026
- 2026Pushing Forward Pareto Frontiers of Proactive Agents with Behavioral Agentic Optimization11 Feb 2026
- 2025Robust Agents in Open-Ended Worlds9 Dec 2025
- 2025cMALC-D: Contextual Multi-Agent LLM-Guided Curriculum Learning with Diversity-Based Context Blending28 Aug 2025
- 2025Exploring Expert Failures Improves LLM Agent Tuning17 Apr 2025
- 2024Diversity Empowers Intelligence: Integrating Expertise of Software Engineering Agents13 Aug 2024
- 2024Understanding the Weakness of Large Language Model Agents within a Complex Android Environment9 Feb 2024
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environmentcited by 99
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 41
- 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
- 2025From monoliths to modules: Decomposing transducers for efficient world modellingcited by 4
- 2026Toward Unified Robot Learning: Bridging Representation, Vision-Language-Action, and World Modelscited by 0
- 2026Toward Unified Robot Learning: Bridging Representation, Vision-Language-Action, and World Models3 Sep 2026
- 2025From monoliths to modules: Decomposing transducers for efficient world modelling1 Dec 2025
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalization27 Aug 2025
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalization16 Mar 2024
- 2024WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environment19 Feb 2024
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environment12 Apr 2021
- 2021Reinforcement Learning in Reward-Mixing MDPscited by 15
- 2024Contextual Bilevel Reinforcement Learning for Incentive Alignmentcited by 14
- 2022Reward-Mixing MDPs with a Few Latent Contexts are Learnablecited by 5
- 2024Contextual Bilevel Reinforcement Learning for Incentive Alignment3 Jun 2024
- 2022Reward-Mixing MDPs with a Few Latent Contexts are Learnable5 Oct 2022
- 2021Reinforcement Learning in Reward-Mixing MDPs7 Oct 2021
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2023Factually Consistent Summarization via Reinforcement Learning with Textual Entailment Feedbackcited by 71
- 2021RL for Latent MDPs: Regret Guarantees and a Lower Boundcited by 69
- 2022Exploration via Elliptical Episodic Bonusescited by 51
- 2021Learning a Large Neighborhood Search Algorithm for Mixed Integer Programscited by 41
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 41
- 2022Efficient Risk-Averse Reinforcement Learningcited by 41
- 2022Automated Dynamic Algorithm Configurationcited by 37
- 2021CORA: Benchmarks, Baselines, and Metrics as a Platform for Continual Reinforcement Learning Agentscited by 36
- 2023AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agentscited by 36
- 2022The Importance of Non-Markovianity in Maximum State Entropy Explorationcited by 35
12 most cited of 171 All 171 (JSON)
- 2026Optimal Multi-Reward Reinforcement Learning29 Sep 2026
- 2026Skill Composition for Legged Robot Reinforcement Learning13 Sep 2026
- 2026Reinforcement learning to choose optimizers1 Sep 2026
- 2026Dynamic Context Scheduling: Learning Beyond the Static Universe21 Aug 2026
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026A Distribution Mapping Approach to Counterfactually Fair Reinforcement Learning9 Aug 2026
- 2026Adjustment Speed as a Safety Constraint for Nonstationary Reinforcement Learning21 Jul 2026
- 2026Building2Building: A Large Scale Benchmark for Generalizable Real-World Reinforcement Learning17 Jul 2026
- 2026Deep Reinforcement Learning for Spacecraft Attitude Control During Atmospheric Re-Entry30 Jun 2026
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalization18 Jun 2026
- 2026Bilevel Optimization of Synthetic Trajectories for Multi-Turn LLM Fine-Tuning23 May 2026
- 2026Curriculum reinforcement learning with measurable task representation learning22 May 2026
12 newest of 171 All 171 (JSON)