Towards robust model-based reinforcement learning against adversarial corruption

C Ye, J He, Q Gu, T Zhang - arXiv preprint arXiv:2402.08991, 2024 - arxiv.org
arXiv preprint arXiv:2402.08991, 2024arxiv.org
This study tackles the challenges of adversarial corruption in model-based reinforcement
learning (RL), where the transition dynamics can be corrupted by an adversary. Existing
studies on corruption-robust RL mostly focus on the setting of model-free RL, where robust
least-square regression is often employed for value function estimation. However, these
techniques cannot be directly applied to model-based RL. In this paper, we focus on model-
based RL and take the maximum likelihood estimation (MLE) approach to learn transition …
This study tackles the challenges of adversarial corruption in model-based reinforcement learning (RL), where the transition dynamics can be corrupted by an adversary. Existing studies on corruption-robust RL mostly focus on the setting of model-free RL, where robust least-square regression is often employed for value function estimation. However, these techniques cannot be directly applied to model-based RL. In this paper, we focus on model-based RL and take the maximum likelihood estimation (MLE) approach to learn transition model. Our work encompasses both online and offline settings. In the online setting, we introduce an algorithm called corruption-robust optimistic MLE (CR-OMLE), which leverages total-variation (TV)-based information ratios as uncertainty weights for MLE. We prove that CR-OMLE achieves a regret of , where denotes the cumulative corruption level after episodes. We also prove a lower bound to show that the additive dependence on is optimal. We extend our weighting technique to the offline setting, and propose an algorithm named corruption-robust pessimistic MLE (CR-PMLE). Under a uniform coverage condition, CR-PMLE exhibits suboptimality worsened by , nearly matching the lower bound. To the best of our knowledge, this is the first work on corruption-robust model-based RL algorithms with provable guarantees.
arxiv.org
以上显示的是最相近的搜索结果。 查看全部搜索结果

Google学术搜索按钮

example.edu/paper.pdf
搜索
获取 PDF 文件
引用
References