NTH

When Does Trajectory-Level Supervision Permit Efficient Offline Reinforcement Learning?

AuthorsXuanfei Ren, Tengyang Xie

June 22, 2026 2 min read
Watch on YouTube
The one-line take

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

2
Outcome-supervised rate

Leading horizon exponent in O(H^2 Csa(pi*)/n) for OPAC

4
Lower-bound sample complexity

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 paper

More in Reinforcement Learning

Browse all 54 papers →