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

微调强化学习模型秘密上是一个遗忘缓解问题

Fine-tuning Reinforcement Learning Models is Secretly a Forgetting Mitigation Problem

IDEAS NCBR;University of Warsaw(华沙大学);Warsaw University of Technology(华沙理工大学);Jagiellonian University(雅盖隆大学);Google DeepMind(谷歌DeepMind);Institute of Mathematics, Polish Academy of Sciences(波兰科学院数学研究所)

Maciej Wołczyk, Bartłomiej Cupiał, Mateusz Ostaszewski, Michał Bortkiewicz, Michał Zając, Razvan Pascanu, Łukasz Kuciński, Piotr Miłoś

AI总结:

本文揭示微调强化学习模型中的性能下降源于对预训练能力的遗忘,提出用知识保留技术缓解,并在NetHack和Montezuma's Revenge上验证,取得新最先进结果。

摘要:

微调是一种广泛使用的技术,允许从业者迁移预训练的能力,正如最近基础模型成功应用所展示的那样。然而,微调强化学习(RL)模型仍然是一个挑战。这项工作将迁移效果不佳的一个具体原因概念化,该原因在RL设置中因动作与观察之间的相互作用而加剧:即对预训练能力的遗忘。具体而言,模型在下游任务的初始微调阶段未访问的状态子空间上性能恶化,而该子空间上模型因预训练而表现良好。这样,我们失去了预期的迁移收益。我们确定了该问题发生的条件,表明它很常见,并且在许多情况下是灾难性的。通过对具有挑战性的NetHack和Montezuma's Revenge环境的详细实证分析,我们表明标准的保留知识技术可以缓解该问题,从而使我们能够充分利用预训练的能力。特别是在NetHack中,我们在Human Monk场景中实现了神经模型的新最先进水平,将之前的最佳分数从5K提高到超过10K点。

英文摘要:

Fine-tuning is a widespread technique that allows practitioners to transfer pre-trained capabilities, as recently showcased by the successful applications of foundation models. However, fine-tuning reinforcement learning (RL) models remains a challenge. This work conceptualizes one specific cause of poor transfer, accentuated in the RL setting by the interplay between actions and observations: forgetting of pre-trained capabilities. Namely, a model deteriorates on the state subspace of the downstream task not visited in the initial phase of fine-tuning, on which the model behaved well due to pre-training. This way, we lose the anticipated transfer benefits. We identify conditions when this problem occurs, showing that it is common and, in many cases, catastrophic. Through a detailed empirical analysis of the challenging NetHack and Montezuma's Revenge environments, we show that standard knowledge retention techniques mitigate the problem and thus allow us to take full advantage of the pre-trained capabilities. In particular, in NetHack, we achieve a new state-of-the-art for neural models, improving the previous best score from $5$K to over $10$K points in the Human Monk scenario.

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

↑