arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.08069cs.LGcs.AI
收藏

Explore-Go:利用探索提升深度强化学习的泛化能力

Explore-Go: Leveraging Exploration for Generalisation in Deep Reinforcement Learning

Delft University of Technology(代尔夫特理工大学)

Max Weltevrede, Felix Kaubek, Matthijs T. J. Spaan, Wendelin Böhmer

AI总结:

针对强化学习智能体对新场景的泛化难题,本文解释了探索助力泛化的直觉,提出扩大起始状态分布的Explore-Go方法,可适配多数主流强化学习算法,实验验证其在说明性环境和Procgen基准上能提升泛化性能。

摘要:

强化学习领域尚存的挑战之一是开发出能够泛化到部署后可能遇到的全新场景的智能体。这一挑战通常被设定在多任务场景中:智能体在一组固定的任务上训练,且必须能泛化到新任务。近期研究表明,在该场景下,增加训练期间的探索可用于提升智能体的泛化性能。当测试期间遇到的状态确实能在训练期间被探索到时,这一结论是合理的。本文提出了一种直觉解释,说明为何探索也能提升对训练期间无法明确遇到的状态的泛化能力。此外,我们提出了一种名为Explore-Go的新方法,该方法通过增加智能体训练所涉及的状态数量来利用这一直觉。Explore-Go有效扩大了智能体的起始状态分布,因此可与大多数现有的同策略(on-policy)或异策略(off-policy)强化学习算法结合使用。我们通过实验证明,该方法在一个说明性环境以及Procgen基准测试上均能提升泛化性能。

英文摘要:

One of the remaining challenges in reinforcement learning is to develop agents that can generalise to novel scenarios they might encounter once deployed. This challenge is often framed in a multi-task setting where agents train on a fixed set of tasks and have to generalise to new tasks. Recent work has shown that in this setting increased exploration during training can be leveraged to increase the generalisation performance of the agent. This makes sense when the states encountered during testing can actually be explored during training. In this paper, we provide intuition why exploration can also benefit generalisation to states that cannot be explicitly encountered during training. Additionally, we propose a novel method Explore-Go that exploits this intuition by increasing the number of states on which the agent trains. Explore-Go effectively increases the starting state distribution of the agent and as a result can be used in conjunction with most existing on-policy or off-policy reinforcement learning algorithms. We show empirically that our method can increase generalisation performance in an illustrative environment and on the Procgen benchmark.

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

↑