When Does Trajectory-Level Supervision Permit Efficient Offline Reinforcement Learning?
AuthorsXuanfei Ren, Tengyang Xie
Resources
This paper asks when you can learn good offline RL policies without per-step rewards, and shows both the limits and the exact sample complexity of doing so from trajectory-level labels or preferences.
Key results
Leading horizon exponent in O(H^2 Csa(pi*)/n) for OPAC
Matching horizon exponent in Ω(H^4/ε^2) trajectories for ε-optimality
What the paper found
This paper gives a sharp sample-complexity theory for offline reinforcement learning when the dataset contains only trajectory-level supervision rather than per-step rewards. For the standard cumulative-return objective, the authors propose OPAC, a pessimistic actor-critic method that learns a latent per-step reward from scalar trajectory labels and then optimizes a policy with layer-wise exponential weights; under realizability and completeness, they prove a high-probability suboptimality rate of O(H^2 Csa(pi*)/n), and a matching lower bound of Ω(H^4/ε^2) trajectories even in deterministic two-action chains with constant concentrability, showing the extra horizon factor is the statistical price of compressing H rewards into one label. They extend the same framework to Bradley–Terry–Luce trajectory preferences, preserving the leading H^2 Csa(pi*)/n dependence up to preference-constant factors. Finally, for generalized trajectory objectives, they show that unrestricted nonlinear aggregation can be exponentially hard: the all-success objective requires Ω(2^H) trajectories. They then identify a tractable regime for Bellman-recursive aggregations via two new complexity measures, the reward-process coefficient κµ(σ) and Bellman inverse coefficient χµ(σ), and prove generalized OPAC achieves polynomial sample complexity scaling like Vmax^2 L κµ(σ)H^2 Csa(pi*)/n plus Vmax^2 L χµ(σ)H^4/n.
Original abstract
Offline reinforcement learning is typically analyzed under process-level reward supervision, yet many sequential decision datasets record only trajectory-level outcomes. We develop a statistical theory for offline policy optimization from such outcome-level supervision. We first study the canonical setting where the target remains the expected cumulative reward, but each offline trajectory provides only a scalar label whose conditional mean is the cumulative return. We propose OPAC, a pessimistic actor-critic algorithm that learns a latent reward model and optimizes a policy from trajectory-level labels. We prove a high-probability guarantee of order $\widetilde O(H^2\sqrt{C_{sa}(π^\star)/n})$ and a matching lower bound, characterizing the sharp statistical cost of replacing process-level rewards with one trajectory-level label. We then extend the principle to preference-based feedback, preserving the leading horizon and concentrability dependence up to preference-model constants. Finally, we study generalized outcome-based offline RL, where both the supervision and the objective are trajectory-level quantities induced by a nonlinear aggregation of latent per-step rewards. This problem is not learnable in general: for all-success objectives, any offline learner may require $Ω(2^H)$ trajectories even with deterministic transitions and constant concentrability. We then identify a tractable regime through two structural coefficients, $κ_μ(σ)$ and $χ_μ(σ)$, capturing information loss in outcome aggregation and generalized Bellman updates, under which generalized OPAC achieves polynomial sample complexity. Together, our results delineate when outcome-level supervision enables sample-efficient offline control and when missing process-level rewards create fundamental statistical barriers.
Read the original paperMore in Reinforcement Learning
Browse all 54 papers →Res-HIL: Human-Guided Residual Reinforcement Learning for Sample-Efficient Dexterous Manipulation
Mariia Iavorskaia, Christian Dietz, Sebastian Albrecht, Majid Khadiv
Res-HIL lets humans efficiently improve robot manipulation skills by teaching a small corrective policy on top of an existing imitation policy.
Selecting Diverse SFT Traces Improves Post-RL Generalization
Dylan Zhang, Mingyuan Wu, Jinning Li
Choosing varied reasoning paths—not just correct ones—can make reinforcement-trained language models generalize better.
Verifiable Hidden Dynamics Play: Generating Agentic RL Environments from Solved Mechanisms
Xinjie Shen, Wei Fan, Xudong Guo, Jianhong Tu, Yang Su, Chuqiao Kuang, Yinger Zhang, Dayiheng Liu
VHD-Play turns solved mathematical mechanisms into cheap, stateful, self-verifying worlds where language-model agents can practice long-horizon decision-making.