Joint MDPs and Reinforcement Learning in Coupled-Dynamics Environments

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Kaya, Ege C., Ghasemi, Mahsa, Hashemi, Abolfazl
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915842086666240
author Kaya, Ege C.
Ghasemi, Mahsa
Hashemi, Abolfazl
author_facet Kaya, Ege C.
Ghasemi, Mahsa
Hashemi, Abolfazl
contents Many distributional quantities in reinforcement learning are intrinsically joint across actions, including distributions of gaps and probabilities of superiority. However, the classical Markov decision process (MDP) formalism specifies only marginal laws and leaves the joint law of counterfactual one-step outcomes across multiple possible actions at a state unspecified. We study coupled-dynamics environments with a multi-action generative interface which can sample counterfactual one-step outcomes for multiple actions under shared exogenous randomness. We propose joint MDPs (JMDPs) as a formalism for such environments by augmenting an MDP with a multi-action sample transition model which specifies a coupling of one-step counterfactual outcomes, while preserving standard MDP interaction as marginal observations. We adopt and formalize a one-step coupling regime where dependence across actions is confined to immediate counterfactual outcomes at the queried state. In this regime, we derive Bellman operators for $n$th-order return moments, providing dynamic programming and incremental algorithms with convergence guarantees.
format Preprint
id arxiv_https___arxiv_org_abs_2603_06946
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Joint MDPs and Reinforcement Learning in Coupled-Dynamics Environments
Kaya, Ege C.
Ghasemi, Mahsa
Hashemi, Abolfazl
Machine Learning
Optimization and Control
Many distributional quantities in reinforcement learning are intrinsically joint across actions, including distributions of gaps and probabilities of superiority. However, the classical Markov decision process (MDP) formalism specifies only marginal laws and leaves the joint law of counterfactual one-step outcomes across multiple possible actions at a state unspecified. We study coupled-dynamics environments with a multi-action generative interface which can sample counterfactual one-step outcomes for multiple actions under shared exogenous randomness. We propose joint MDPs (JMDPs) as a formalism for such environments by augmenting an MDP with a multi-action sample transition model which specifies a coupling of one-step counterfactual outcomes, while preserving standard MDP interaction as marginal observations. We adopt and formalize a one-step coupling regime where dependence across actions is confined to immediate counterfactual outcomes at the queried state. In this regime, we derive Bellman operators for $n$th-order return moments, providing dynamic programming and incremental algorithms with convergence guarantees.
title Joint MDPs and Reinforcement Learning in Coupled-Dynamics Environments
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2603.06946