Regularized Offline Policy Optimization with Posterior Hybrid Bayesian Belief
About
Offline reinforcement learning (RL) aims to optimize policies from pre-collected datasets. A bottleneck of this paradigm is managing epistemic uncertainty, which arises from limited data coverage (sample-level) and the ambiguity in identifying transition dynamics from finite data (model-level). To provide a unified quantification of these uncertainties, Bayesian RL has been proposed by treating the dynamics model as a random variable and maintaining a corresponding belief. Despite its theoretical appeal, policy optimization in Bayesian RL remains computationally challenging as it requires solving composite objectives with expectations. Prior methods either employ search-based techniques with poor computational scalability or impose restrictive posterior assumptions that sacrifice the adaptability of Bayesian RL. To address these limitations, we propose Posterior Hybrid Bayesian Belief (PhyB), which reformulates the expectation as a convex combination over a subset of dynamics models. Theoretical analysis demonstrates that the objective discrepancy induced by this approximation remains bounded. Based on PhyB, we develop an iterative regularized policy optimization algorithm that provides metric-agnostic guarantees for monotonic improvement until convergence. Empirical results demonstrate that PhyB achieves state-of-the-art performance on various benchmarks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Offline Reinforcement Learning | D4RL Walker2d Medium v2 | Normalized Return95.5 | 85 | |
| Offline Reinforcement Learning | D4RL halfcheetah v2 (medium-replay) | Normalized Score74.7 | 68 | |
| Offline Reinforcement Learning | D4RL hopper-expert v2 | Normalized Score118.9 | 66 | |
| Offline Reinforcement Learning | D4RL walker2d-expert v2 | Normalized Score116.3 | 66 | |
| Offline Reinforcement Learning | D4RL halfcheetah-expert v2 | Normalized Score113.7 | 66 | |
| Offline Reinforcement Learning | D4RL Hopper-medium-replay v2 | Normalized Return110.7 | 64 | |
| Offline Reinforcement Learning | D4RL Hopper-medium-expert v2 | Normalized Return116.5 | 61 | |
| Offline Reinforcement Learning | D4RL walker2d-medium-expert v2 | Normalized Score112.4 | 54 | |
| Offline Reinforcement Learning | D4RL HalfCheetah Medium v2 | Average Normalized Return74.5 | 53 | |
| Offline Reinforcement Learning | D4RL walker2d medium-replay v2 | Normalized Score85.4 | 46 |