arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.01903cs.LGcs.AIcs.CV
收藏

面向策略学习的文本感知扩散模型

Text-Aware Diffusion for Policy Learning

Brown University(布朗大学)

Calvin Luo, Mandy He, Zilai Zeng, Chen Sun

AI总结:

针对强化学习为新目标设计奖励函数困难的问题,提出TADPoLe方法,利用冻结的预训练文本条件扩散模型生成零样本密集奖励,在多种环境下无需演示和真实奖励即可学习符合文本要求且更自然的策略。

摘要:

训练智能体达成特定目标或执行期望行为通常通过强化学习实现,尤其是在缺乏专家演示的场景下。然而,通过强化学习支持新的目标或行为需要临时设计合适的奖励函数,这一过程很快会变得难以处理。为解决这一挑战,我们提出面向策略学习的文本感知扩散模型(Text-Aware Diffusion for Policy Learning,TADPoLe),该方法利用预训练的冻结文本条件扩散模型,为文本对齐的策略学习计算密集的零样本奖励信号。我们假设,大规模预训练生成模型编码了丰富的先验知识,这些先验不仅能监督策略以符合文本要求的方式行动,还能使其与从互联网规模训练数据中总结出的“自然性”概念保持一致。实验中我们证明,TADPoLe能够在人形机器人(Humanoid)和狗形机器人(Dog)环境中,学习由自然语言指定的新型目标达成和连续运动行为的策略。这些行为是在无真实奖励或专家演示的零样本条件下习得的,且根据人类评估,其行为在定性上更自然。我们进一步表明,当应用于Meta-World环境中的机器人操作任务时,在无需访问任何领域内演示的情况下,TADPoLe的表现仍具有竞争力。

英文摘要:

Training an agent to achieve particular goals or perform desired behaviors is often accomplished through reinforcement learning, especially in the absence of expert demonstrations. However, supporting novel goals or behaviors through reinforcement learning requires the ad-hoc design of appropriate reward functions, which quickly becomes intractable. To address this challenge, we propose Text-Aware Diffusion for Policy Learning (TADPoLe), which uses a pretrained, frozen text-conditioned diffusion model to compute dense zero-shot reward signals for text-aligned policy learning. We hypothesize that large-scale pretrained generative models encode rich priors that can supervise a policy to behave not only in a text-aligned manner, but also in alignment with a notion of naturalness summarized from internet-scale training data. In our experiments, we demonstrate that TADPoLe is able to learn policies for novel goal-achievement and continuous locomotion behaviors specified by natural language, in both Humanoid and Dog environments. The behaviors are learned zero-shot without ground-truth rewards or expert demonstrations, and are qualitatively more natural according to human evaluation. We further show that TADPoLe performs competitively when applied to robotic manipulation tasks in the Meta-World environment, without having access to any in-domain demonstrations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑