让自回归模型学会在内部状态中生成高层动作,实现高效探索。
Emergent temporal abstractions in autoregressive models enable hierarchical reinforcement learning
- 用高阶非因果模型控制基础模型的激活流,生成时间抽象动作。
- 在网格世界和MuJoCo任务中,控制器可执行长时序有意义行为并带终止条件。
- 内部强化学习使稀疏奖励下学习成功,适合复杂任务规划场景。
大规模自回归模型在预测下一个词的基础上,经强化学习微调后,在多个领域取得突破性进展。强化学习过程中,这些模型通过逐个词采样动作进行探索,但在奖励稀疏时效率极低。本文提出在自回归模型的内部表示中行动与探索,以克服该问题。具体地,引入一个高阶非因果序列模型,其输出控制基础模型的残差流激活。在具有层次结构的网格世界和MuJoCo任务中,高阶模型学会将长激活序列压缩为内部控制器。每个控制器执行一系列行为上连贯的动作,持续时间长,并带有学习到的终止条件,组合多个控制器可实现新任务的高效探索。我们发现直接对内部控制器进行强化学习(称为“内部强化学习”),可在标准强化学习失败的情况下实现稀疏奖励下的学习。结果表明,自回归模型中的潜在动作生成与强化具有显著优势,提示内部强化学习是实现基础模型内层次强化学习的重要方向。
原文摘要 · Abstract (English)
Large-scale autoregressive models pretrained on next-token prediction and finetuned with reinforcement learning (RL) have achieved unprecedented success on many problem domains. During RL, these models explore by generating new outputs, one token at a time. However, sampling actions token-by-token can result in highly inefficient learning, particularly when rewards are sparse. Here, we show that it is possible to overcome this problem by acting and exploring within the internal representations of an autoregressive model. Specifically, to discover temporally-abstract actions, we introduce a higher-order, non-causal sequence model whose outputs control the residual stream activations of a base autoregressive model. On grid world and MuJoCo-based tasks with hierarchical structure, we find that the higher-order model learns to compress long activation sequence chunks onto internal controllers. Critically, each controller executes a sequence of behaviorally meaningful actions that unfold over long timescales and are accompanied with a learned termination condition, such that composing multiple controllers over time leads to efficient exploration on novel tasks. We show that direct internal controller reinforcement, a process we term "internal RL", enables learning from sparse rewards in cases where standard RL finetuning fails. Our results demonstrate the benefits of latent action generation and reinforcement in autoregressive models, suggesting internal RL as a promising avenue for realizing hierarchical RL within foundation models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。