Distributional Mismatch

Distributional mismatch is the main problem in offline RL. A learned policy is optimized by looking for actions that maximize Q(s,a)Q(s,a), but many actions the policy considers were never taken in the dataset. To evaluate those actions, the value function must extrapolate outside the data distribution, and extrapolation is unreliable.

This is the core difference from ordinary supervised learning. A supervised model usually predicts yy from inputs xx that resemble the training set. Offline RL must reason about counterfactuals: What would have happened if a different action had been taken? What reward would have occurred? How would future states have evolved? Those questions require extrapolation, not just interpolation.

Distributional mismatch

When Q-values leave the dataset

Pick an action the learned policy might consider. Watch how support, value error, and policy preference interact.

Dataset support

0%

Estimated Q

9.6

True value

1.8

Error

7.8

Policy probability after greedy improvement82%

The policy is now optimizing an extrapolation error. Offline data cannot correct it by exploring.

Offline RL is hard because bad counterfactual estimates cannot be corrected by collecting new counterfactual data.

Bootstrapped value learning makes this especially delicate. A value update often uses a target like

Q(s,a)←r+γEs′[V(s′)]Q(s,a)\leftarrow r+\gamma\mathbb{E}_{s'}[V(s')]

If Q(s,a)Q(s,a) is estimated for an unsupported action, the immediate reward, next-state distribution, and future value may all be extrapolated. Neural value functions can assign arbitrarily high Q-values to unfamiliar state-action pairs because the data never constrained those predictions. The policy then exploits these errors and selects actions the dataset provides no evidence for.

Equation walkthrough

Bootstrapping value from the future

Click each term to see why this update is useful, and why it becomes dangerous outside the dataset support.

+[]

Current Q-value

value of taking action a in state s

This is the action-value estimate being updated. In offline RL, it becomes risky when s or a is poorly covered by the fixed dataset.

Terms

The target mixes observed reward with learned future value; offline RL has to keep both terms grounded in D.

A soft policy improvement step makes the amplification visible:

π(a∣s)∝exp⁡(Q(s,a))\pi(a\mid s)\propto \exp(Q(s,a))

A slightly inflated QQ can become a much larger action probability. Even tiny overestimates get magnified, pushing the policy deeper into out-of-distribution regions where QQ becomes even less grounded.

Equation walkthrough

Why small Q-errors move the policy

Click each term to see how a Q-value becomes an action preference.

Policy probability

chance of choosing action a in state s

This is the learned policy's probability for action a in state s. The higher this value is, the more often the agent chooses that action.

Terms

The exponential policy update can turn a small error in Q(s,a) into a large behavioral shift.
ℹ

The Offline RL Rule of Thumb

The learned policy should stay close to the dataset support, roughly π(a∣s)≈β(a∣s)\pi(a\mid s)\approx\beta(a\mid s), or the algorithm must explicitly penalize unsupported actions. Every major offline RL method is a different way of enforcing this pressure.

◆

Example: Finance

A trading dataset may contain many ordinary market days and very few crisis states. If an offline policy recommends a leveraged action in a crisis-like state, the value estimate may be based on weak extrapolation. High predicted return can be less important than the question: have we ever seen enough evidence for this action under these conditions?

Checkpoint

What makes distribution shift harder to correct in offline RL than online RL?