Dream to Remember:通用机器人策略持续优化方法
Dream to Remember: Continual Steering of Generalist Robot Policies with World Models
MIT提出D2R方法,让机器人通过想象学习新技能而不遗忘旧技能,在真实机器人上验证成功。
Dream to Remember (D2R) 方法通过冻结的通用策略和预训练世界模型实现机器人技能的持续学习。该方法在CALVIN基准上连续学习8个任务,平均成功率从63.7%提升至74.8%,而LoRA微调则降至11.7%。D2R使用想象回放来保留先前的技能,无需重放缓冲区或新数据。在LIBERO-90任务中,D2R在保持原有性能的同时学习新任务。在Franka机器人上,该方法在连续学习3个任务时完全通过想象实现。
Dream to Remember: Continual Steering of Generalist Robot Policies with World Models
Generalist robot policies are trained with imitation learning and are bounded by the quality and coverage of their demonstrations. Reinforcement learning can improve and extend them, either by fine-tuning the policy weights or by training a small module to steer the input noise of a frozen generative policy. When tasks arrive sequentially, both approaches suffer from catastrophic forgetting. Fine-tuning on a task degrades prior tasks, and steering only shifts forgetting to the shared steering module. We propose Dream to Remember (D2R), which continually steers a frozen generalist policy toward new tasks, while retaining its earlier skills. As the steering module learns a new task, a frozen pretrained world model rehearses prior tasks through imagined rollouts, anchoring the module to its earlier behavior without a replay buffer or new data. When the world model is faithful enough, D2R learns the new task entirely in imagination. Otherwise, it learns the task in the environment while using the world model only to rehearse prior tasks. On CALVIN, D2R improves eight tasks in sequence entirely in imagination, raising average success from 63.7% to 74.8%, while LoRA fine-tuning on the same sequence drops it to 11.7%. With a frozen off-the-shelf generalist policy and world model, $π_0$ and A2World, D2R learns unseen LIBERO-90 tasks in the environment while keeping $π_0$'s performance on the LIBERO suites it was fine-tuned on, even though the world model is not accurate enough for policy optimization. Finally, on a real Franka robot, D2R improves three tasks in sequence entirely in imagination while retaining the earlier improvements.