arXiv:2501.16288cs.LGcs.AI2025-01被引 2

用生成器直接输出策略,不用评判器就能高效训练。

Upside Down Reinforcement Learning with Policy Generators

  • 用超网络根据指令生成特定权重的策略
  • 无需评判器,训练收敛更快且回报更高
  • 可零样本生成新任务策略,适合快速部署

Upside Down Reinforcement Learning(UDRL)是一种聚焦于学习命令条件策略的强化学习框架。本文将其扩展至学习命令条件的深度神经网络策略生成器,采用超网络(Hypernetworks)——一种快速权重编程变体——将表示期望回报的命令解码为特定命令的权重矩阵。该方法称为UDRLPG,通过移除生成器权重更新所需的评估器或评判器,简化了现有技术。为缓解因缺少评判器导致的最终回报方差增加问题,我们分离了缓冲区采样概率与策略数量,并结合简单加权策略,提升了算法的实证收敛性。实验表明,相比现有算法,UDRLPG性能相当甚至更优,且在某些情况下超越复杂架构。训练后的生成器能零样本泛化到未见回报任务。该方法有效缓解了学习高度多模态函数的挑战。总体而言,我们认为UDRLPG在提升强化学习的样本效率方面迈出了有前景的一步。完整代码已公开:https://github.com/JacopoD/udrlpg_

原文摘要 · Abstract (English)

Upside Down Reinforcement Learning (UDRL) is a promising framework for solving reinforcement learning problems which focuses on learning command-conditioned policies. In this work, we extend UDRL to the task of learning a command-conditioned generator of deep neural network policies. We accomplish this using Hypernetworks - a variant of Fast Weight Programmers, which learn to decode input commands representing a desired expected return into command-specific weight matrices. Our method, dubbed Upside Down Reinforcement Learning with Policy Generators (UDRLPG), streamlines comparable techniques by removing the need for an evaluator or critic to update the weights of the generator. To counteract the increased variance in last returns caused by not having an evaluator, we decouple the sampling probability of the buffer from the absolute number of policies in it, which, together with a simple weighting strategy, improves the empirical convergence of the algorithm. Compared with existing algorithms, UDRLPG achieves competitive performance and high returns, sometimes outperforming more complex architectures. Our experiments show that a trained generator can generalize to create policies that achieve unseen returns zero-shot. The proposed method appears to be effective in mitigating some of the challenges associated with learning highly multimodal functions. Altogether, we believe that UDRLPG represents a promising step forward in achieving greater empirical sample efficiency in RL. A full implementation of UDRLPG is publicly available at https://github.com/JacopoD/udrlpg_

强化学习策略生成超网络样本效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。