用后验行为克隆提升强化学习微调效果,让预训练策略更适合作为起点。
Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning
- 通过建模演示数据下的行为后验分布,确保策略覆盖所有示范动作。
- 相比传统行为克隆,新方法在真实机器人任务中实现显著更高的微调性能。
- 无需复杂算法,仅用标准监督学习即可实现,适合实际部署场景。
在机器人到语言等多个领域,标准做法是先在大规模示范数据上预训练策略,再通过强化学习(RL)微调以提升部署表现。尽管微调算法研究众多,但对预训练策略是否能有效初始化微调的关注不足。本文理论证明,标准行为克隆(BC)可能无法覆盖示范者的全部动作,而这是有效微调的必要条件。我们提出后验行为克隆(PostBC),训练策略以建模给定数据集下示范者行为的后验分布,从而保证动作覆盖性,同时保持预训练性能不低于传统BC。实验表明,该方法在现代生成模型支持下可高效实现,在真实机器人控制与操作任务中均显著优于标准行为克隆。
原文摘要 · Abstract (English)
Standard practice across domains from robotics to language is to first pretrain a policy on a large-scale demonstration dataset, and then finetune this policy, typically with reinforcement learning (RL), in order to improve performance on deployment domains. This finetuning step has proved critical in achieving human or super-human performance, yet while much attention has been given to developing more effective finetuning algorithms, little attention has been given to ensuring the pretrained policy is an effective initialization for RL finetuning. In this work we seek to understand how the pretrained policy affects finetuning performance, and how to pretrain policies in order to ensure they are effective initializations for finetuning. We first show theoretically that standard behavioral cloning (BC) -- which trains a policy to directly match the actions played by the demonstrator -- can fail to ensure coverage over the demonstrator's actions, a minimal condition necessary for effective RL finetuning. We then show that if, instead of exactly fitting the observed demonstrations, we train a policy to model the posterior distribution of the demonstrator's behavior given the demonstration dataset, we do obtain a policy that ensures coverage over the demonstrator's actions, enabling more effective finetuning. Furthermore, this policy -- which we refer to as the posterior behavioral cloning (PostBC) policy -- achieves this while ensuring pretrained performance is no worse than that of the BC policy. We then show that PostBC is practically implementable with modern generative models in robotic control domains -- relying only on standard supervised learning -- and leads to significantly improved RL finetuning performance on both realistic robotic control benchmarks and real-world robotic manipulation tasks, as compared to standard behavioral cloning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。