arXiv:2410.19982cs.LGcs.AI2024-10被引 1

用随机策略生成预训练数据,让模型在无优策略情况下也能实现上下文强化学习。

Random Policy Enables In-Context Reinforcement Learning within Trust Horizons

  • 通过随机策略在信任范围内筛选最优状态-动作对构建预训练数据
  • 离线评估平均性能提升236.3%,在线评估提升135.2%
  • 首次实现仅依赖随机策略的高效上下文强化学习,适合真实场景

预训练基础模型在上下文学习方面表现出色,可零样本泛化到未见任务。在强化学习中,当基础模型以自回归监督方式在决策问题上预训练时,会形成上下文强化学习(ICRL)。然而,当前最先进的ICRL算法如算法蒸馏、决策预训练变换器和决策重要性变换器,对预训练数据集中的源策略、上下文信息和动作标签有严格要求,通常需要最优策略或充分训练的行为策略。这在真实场景中难以实现,因获取大量环境的最优或良好策略极为困难。为此,我们提出新方法状态-动作蒸馏(SAD),仅依赖随机策略生成有效预训练数据。SAD在信任范围内使用随机策略蒸馏出优秀状态-动作对,并继承经典的自回归监督机制进行预训练。据我们所知,这是首个在随机策略和随机上下文中实现有效ICRL的工作。我们还提供了可信度与性能保证的定量分析。在多个主流ICRL基准环境上的实证结果表明,SAD在离线评估中平均优于最佳基线236.3%,在线评估中提升135.2%。

原文摘要 · Abstract (English)

Pretrained foundation models have exhibited extraordinary in-context learning performance, allowing zero-shot generalization to new tasks not encountered during pretraining. In the case of reinforcement learning (RL), in-context RL (ICRL) emerges when pretraining FMs on decision-making problems in an autoregressive-supervised manner. Nevertheless, current state-of-the-art ICRL algorithms, like Algorithm Distillation, Decision Pretrained Transformer and Decision Importance Transformer, impose stringent requirements on the pretraining dataset concerning the source policies, context information, and action labels. Notably, these algorithms either demand optimal policies or require varying degrees of well-trained behavior policies for all pretraining environments. This significantly hinders the application of ICRL to real-world scenarios, where acquiring optimal or well-trained policies for a substantial volume of real-world training environments can be intractable. To overcome this challenge, we introduce a novel approach, termed State-Action Distillation (SAD), that allows to generate an effective pretraining dataset guided solely by random policies. In particular, SAD selects query states and corresponding action labels by distilling outstanding state-action pairs from the entire state and action spaces by using random policies within a trust horizon, and then inherits the classical autoregressive-supervised mechanism during pretraining. To the best of our knowledge, this is the first work that enables effective ICRL under random policies and random contexts. We also establish quantitative analysis of the trustworthiness as well as the performance guarantees of SAD. Moreover, our empirical results across multiple popular ICRL benchmark environments demonstrate that, on average, SAD outperforms the best baseline by 236.3% in the offline evaluation and by 135.2% in the online evaluation.

强化学习上下文学习随机策略预训练

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