Saved in:
Bibliographic Details
Main Authors: Zhang, Haichao, Xu, We, Yu, Haonan
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2302.00935
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914011581251584
author Zhang, Haichao
Xu, We
Yu, Haonan
author_facet Zhang, Haichao
Xu, We
Yu, Haonan
contents Pre-training with offline data and online fine-tuning using reinforcement learning is a promising strategy for learning control policies by leveraging the best of both worlds in terms of sample efficiency and performance. One natural approach is to initialize the policy for online learning with the one trained offline. In this work, we introduce a policy expansion scheme for this task. After learning the offline policy, we use it as one candidate policy in a policy set. We then expand the policy set with another policy which will be responsible for further learning. The two policies will be composed in an adaptive manner for interacting with the environment. With this approach, the policy previously learned offline is fully retained during online learning, thus mitigating the potential issues such as destroying the useful behaviors of the offline policy in the initial stage of online learning while allowing the offline policy participate in the exploration naturally in an adaptive manner. Moreover, new useful behaviors can potentially be captured by the newly added policy through learning. Experiments are conducted on a number of tasks and the results demonstrate the effectiveness of the proposed approach. Code is available at https://github.com/Haichao-Zhang/PEX
format Preprint
id arxiv_https___arxiv_org_abs_2302_00935
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Policy Expansion for Bridging Offline-to-Online Reinforcement Learning
Zhang, Haichao
Xu, We
Yu, Haonan
Artificial Intelligence
Pre-training with offline data and online fine-tuning using reinforcement learning is a promising strategy for learning control policies by leveraging the best of both worlds in terms of sample efficiency and performance. One natural approach is to initialize the policy for online learning with the one trained offline. In this work, we introduce a policy expansion scheme for this task. After learning the offline policy, we use it as one candidate policy in a policy set. We then expand the policy set with another policy which will be responsible for further learning. The two policies will be composed in an adaptive manner for interacting with the environment. With this approach, the policy previously learned offline is fully retained during online learning, thus mitigating the potential issues such as destroying the useful behaviors of the offline policy in the initial stage of online learning while allowing the offline policy participate in the exploration naturally in an adaptive manner. Moreover, new useful behaviors can potentially be captured by the newly added policy through learning. Experiments are conducted on a number of tasks and the results demonstrate the effectiveness of the proposed approach. Code is available at https://github.com/Haichao-Zhang/PEX
title Policy Expansion for Bridging Offline-to-Online Reinforcement Learning
topic Artificial Intelligence
url https://arxiv.org/abs/2302.00935