arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.19982cs.LGcs.AI
收藏

随机策略在信任范围内实现上下文强化学习

Random Policy Enables In-Context Reinforcement Learning within Trust Horizons

Department of Electrical, Computer, and Systems Engineering, Rensselaer Polytechnic Institute(电气、计算机与系统工程系,拉特格斯理工学院)

Weiqin Chen, Santiago Paternain

AI总结:

提出State-Action Distillation(SAD)方法,仅利用随机策略在信任范围内提取状态-动作对生成预训练数据集,首次在随机策略和上下文下实现有效ICRL,并在多个基准上显著超越基线。

摘要:

预训练基础模型展现出卓越的上下文学习性能,能够对预训练中未遇到的新任务进行零样本泛化。在强化学习(RL)中,当以自回归监督方式在决策问题上预训练基础模型时,便产生了上下文强化学习(ICRL)。然而,当前最先进的ICRL算法,如Algorithm Distillation、Decision Pretrained Transformer和Decision Importance Transformer,对预训练数据集在源策略、上下文信息和动作标签方面施加了严格要求。值得注意的是,这些算法要么要求最优策略,要么要求所有预训练环境具有不同程度的训练良好的行为策略。这极大阻碍了ICRL在现实场景中的应用,因为在现实场景中,为大量真实训练环境获取最优或训练良好的策略可能极其困难。为克服这一挑战,我们引入了一种名为State-Action Distillation(SAD)的新方法,该方法允许仅由随机策略引导生成有效的预训练数据集。具体而言,SAD通过在信任范围内使用随机策略,从整个状态和动作空间中提取出色的状态-动作对,以此选择查询状态及相应的动作标签,并在预训练期间继承经典的自回归监督机制。据我们所知,这是首个在随机策略和随机上下文下实现有效ICRL的工作。我们还对SAD的可靠性及性能保证进行了定量分析。此外,我们在多个流行的ICRL基准环境中的实证结果表明,平均而言,SAD在离线评估中比最佳基线高出236.3%,在在线评估中高出135.2%。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑