Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Hongqiang Lin, Pengfei Wang, Nenggan Zheng• 2026

Related benchmarks

TaskDatasetResultRank
Offline Reinforcement LearningD4RL Walker2d Medium v2
Normalized Return95.5
85
Offline Reinforcement LearningD4RL halfcheetah v2 (medium-replay)
Normalized Score74.7
68
Offline Reinforcement LearningD4RL hopper-expert v2
Normalized Score118.9
66
Offline Reinforcement LearningD4RL walker2d-expert v2
Normalized Score116.3
66
Offline Reinforcement LearningD4RL halfcheetah-expert v2
Normalized Score113.7
66
Offline Reinforcement LearningD4RL Hopper-medium-replay v2
Normalized Return110.7
64
Offline Reinforcement LearningD4RL Hopper-medium-expert v2
Normalized Return116.5
61
Offline Reinforcement LearningD4RL walker2d-medium-expert v2
Normalized Score112.4
54
Offline Reinforcement LearningD4RL HalfCheetah Medium v2
Average Normalized Return74.5
53
Offline Reinforcement LearningD4RL walker2d medium-replay v2
Normalized Score85.4
46
Showing 10 of 20 rows

Other info

Follow for update