English

Provable Offline Reinforcement Learning for Structured Cyclic MDPs

Machine Learning 2026-02-13 v1 Artificial Intelligence Machine Learning Optimization and Control Methodology

Abstract

We introduce a novel cyclic Markov decision process (MDP) framework for multi-step decision problems with heterogeneous stage-specific dynamics, transitions, and discount factors across the cycle. In this setting, offline learning is challenging: optimizing a policy at any stage shifts the state distributions of subsequent stages, propagating mismatch across the cycle. To address this, we propose a modular structural framework that decomposes the cyclic process into stage-wise sub-problems. While generally applicable, we instantiate this principle as CycleFQI, an extension of fitted Q-iteration enabling theoretical analysis and interpretation. It uses a vector of stage-specific Q-functions, tailored to each stage, to capture within-stage sequences and transitions between stages. This modular design enables partial control, allowing some stages to be optimized while others follow predefined policies. We establish finite-sample suboptimality error bounds and derive global convergence rates under Besov regularity, demonstrating that CycleFQI mitigates the curse of dimensionality compared to monolithic baselines. Additionally, we propose a sieve-based method for asymptotic inference of optimal policy values under a margin condition. Experiments on simulated and real-world Type 1 Diabetes data sets demonstrate CycleFQI's effectiveness.

Keywords

Cite

@article{arxiv.2602.11679,
  title  = {Provable Offline Reinforcement Learning for Structured Cyclic MDPs},
  author = {Kyungbok Lee and Angelica Cristello Sarteau and Michael R. Kosorok},
  journal= {arXiv preprint arXiv:2602.11679},
  year   = {2026}
}

Comments

65 pages, 4 figures. Submitted to JMLR

R2 v1 2026-07-01T10:33:12.473Z