arXiv:2505.15143cs.LGcs.RO2025-05被引 1

通过过滤学习历史,提升大模型在新环境中的强化学习表现

Filtering Learning Histories Enhances In-Context Reinforcement Learning

  • 基于改进的损失权重机制,筛选并重加权学习历史数据
  • 在多种任务中显著提升性能,尤其在噪声数据下优势明显
  • 适用于现有SOTA强化学习框架,无需额外训练

Transformer模型在上下文强化学习(ICRL)中展现出强大能力,可泛化至未见环境且无需重新训练。当前方法通常依赖大量预训练环境下的完整学习历史进行模仿,但可能继承源算法/数据集中的次优行为。为此,本文提出学习历史过滤(LHF),通过重加权和筛选具有稳定性和进步性的学习历史,从数据预处理角度缓解次优性问题。LHF是首个通过数据预处理避免源次优性的方法,可与现有最先进ICRL算法(AD、DPT、DICP)结合。在多个经典基准测试中验证,涵盖离散环境与连续机器人操作任务,LHF在不同超参数和采样策略下均表现稳健,尤其在噪声数据场景中性能更优,证明了学习历史过滤的重要性。

原文摘要 · Abstract (English)

Transformer models (TMs) have exhibited remarkable in-context reinforcement learning (ICRL) capabilities, allowing them to generalize to and improve in previously unseen environments without re-training or fine-tuning. This is typically accomplished by imitating the complete learning histories of a source RL algorithm over a substantial amount of pretraining environments, which, however, may transfer suboptimal behaviors inherited from the source algorithm/dataset. Therefore, in this work, we address the issue of inheriting suboptimality from the perspective of dataset preprocessing. Motivated by the success of the weighted empirical risk minimization, we propose a simple yet effective approach, learning history filtering (LHF), to enhance ICRL by reweighting and filtering the learning histories based on their improvement and stability characteristics. To the best of our knowledge, LHF is the first approach to avoid source suboptimality by dataset preprocessing, and can be combined with the current state-of-the-art (SOTA) ICRL algorithms. We substantiate the effectiveness of LHF through a series of experiments conducted on the well-known ICRL benchmarks, encompassing both discrete environments and continuous robotic manipulation tasks, with three SOTA ICRL algorithms (AD, DPT, DICP) as the backbones. LHF exhibits robust performance across a variety of suboptimal scenarios, as well as under varying hyperparameters and sampling strategies. Notably, the superior performance of LHF becomes more pronounced in the presence of noisy data, indicating the significance of filtering learning histories.

强化学习上下文学习数据过滤Transformer

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