arXiv:2602.17497cs.LG2026-02NeurIPS被引 4

用大模型回溯学习,让稀疏奖励变密集信号,提升强化学习效率

Retrospective In-Context Learning for Temporal Credit Assignment with Large Language Models

  • 通过大模型回溯式上下文学习,将稀疏奖励转化为密集优势函数
  • 在4个BabyAI任务中,样本效率远超传统在线强化学习算法
  • 适合追求高效训练的强化学习研究者和智能体开发者

从自采数据和稀疏环境反馈中学习仍是训练自演进智能体的核心挑战。时间信用分配通过将稀疏反馈转换为密集监督信号缓解此问题。然而,以往方法通常依赖特定任务的价值函数进行信用分配,存在样本效率低、泛化能力差的问题。本文提出利用预训练大语言模型(LLM)通过回溯式上下文学习(RICL),将稀疏奖励转化为密集训练信号(即优势函数)。进一步设计在线学习框架RICOL,基于RICL的信用分配结果迭代优化策略。实验表明,RICL可在少量样本下准确估计优势函数,并有效识别环境中关键状态。在四个BabyAI场景上的扩展评估显示,RICOL在收敛性能上可媲美传统在线强化学习算法,且样本效率显著更高。研究结果表明,利用大模型进行时间信用分配具有巨大潜力,为更高效、更通用的强化学习范式开辟道路。

原文摘要 · Abstract (English)

Learning from self-sampled data and sparse environmental feedback remains a fundamental challenge in training self-evolving agents. Temporal credit assignment mitigates this issue by transforming sparse feedback into dense supervision signals. However, previous approaches typically depend on learning task-specific value functions for credit assignment, which suffer from poor sample efficiency and limited generalization. In this work, we propose to leverage pretrained knowledge from large language models (LLMs) to transform sparse rewards into dense training signals (i.e., the advantage function) through retrospective in-context learning (RICL). We further propose an online learning framework, RICOL, which iteratively refines the policy based on the credit assignment results from RICL. We empirically demonstrate that RICL can accurately estimate the advantage function with limited samples and effectively identify critical states in the environment for temporal credit assignment. Extended evaluation on four BabyAI scenarios show that RICOL achieves comparable convergent performance with traditional online RL algorithms with significantly higher sample efficiency. Our findings highlight the potential of leveraging LLMs for temporal credit assignment, paving the way for more sample-efficient and generalizable RL paradigms.

强化学习大模型信用分配样本效率

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