1
Return

Enhancing Offline Model-Based RL via Active Model Selection: A Bayesian Optimization Perspective

delete2026-07-17
delete0
delete
OA
AI
Y
Yuwei Yang
W
Wei Hung
Y
Yun-Ming Chan
X
Xi Liu
P
Ping-Chun Hsieh *
DOI:10.1007/s10994-026-07119-6delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Offline model-based reinforcement learning (MBRL) serves as a competitive framework that can learn well-performing policies solely from pre-collected data with the help of learned dynamics models. To fully unleash the power of offline MBRL, model selection plays a pivotal role in determining the dynamics model utilized for downstream policy learning. However, offline MBRL conventionally relies on validation or off-policy evaluation, which are rather inaccurate due to the inherent distribution shift in offline RL. To tackle this, we propose BOMS, an active model selection framework that enhances model selection in offline MBRL with only a small online interaction budget, through the lens of Bayesian optimization (BO). Specifically, we recast model selection as BO and enable probabilistic inference in BOMS by proposing a novel model-induced kernel, which is theoretically grounded and computationally efficient. Through extensive experiments, we show that BOMS improves over the baseline methods with a small amount of online interaction comparable to only $$1\%$$ - $$2.5\%$$ of offline training data on various RL tasks.
Keywords:
Offline reinforcement learning
Bayesian optimization
Dynamics model selection
Dynamics model distance
Gaussian process
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Machine Learning cover
Machine Learning
IF:
2.9
Papers:
2.6K
Citations:
3.4W

Organization

D
department of computer science
Scholars:
541
Papers: 283
Citations: 0
Cited Papers

Cited Papers

Citing Papers

Citing Papers