让智能体自定目标预训练,提升探索与适应新任务能力
Unsupervised Learning of Efficient Exploration: Pre-training Adaptive Policies via Self-Imposed Goals
- 通过自设目标和元学习框架优化多轮探索效率
- 在迷你网格环境中实现零样本与少样本性能显著提升
- 适合需要快速适应未知任务的强化学习场景
无监督预训练可为强化学习智能体提供先验知识,加速下游任务学习。受人类发展启发,本文研究智能体自主设定并追求目标的学习方式。核心挑战在于如何有效生成、选择并从这些目标中学习。研究聚焦于下游任务分布广泛且无法零样本解决的场景,此类情况常见于目标任务超出预训练分布或身份未知时。本文提出ULee方法,(i)在元学习框架内优化多轮探索与适应效率,(ii)利用对智能体后适应性能的动态估计引导训练课程。ULee结合上下文学习器与对抗性目标生成策略,在XLand-MiniGrid基准上展现出更强的探索与适应能力,能泛化至新目标、环境动态和地图结构。所获策略在零样本与少样本任务中表现更优,并为长周期微调提供良好初始化。其性能优于从头学习、DIAYN预训练及其它课程设计。
原文摘要 · Abstract (English)
Unsupervised pre-training can equip reinforcement learning agents with prior knowledge and accelerate learning in downstream tasks. A promising direction, grounded in human development, investigates agents that learn by setting and pursuing their own goals. The core challenge lies in how to effectively generate, select, and learn from such goals. Our focus is on broad distributions of downstream tasks where solving every task zero-shot is infeasible. Such settings naturally arise when the target tasks lie outside of the pre-training distribution or when their identities are unknown to the agent. In this work, we (i) optimize for efficient multi-episode exploration and adaptation within a meta-learning framework, and (ii) guide the training curriculum with evolving estimates of the agent's post-adaptation performance. We present ULEE, an unsupervised meta-learning method that combines an in-context learner with an adversarial goal-generation strategy that maintains training at the frontier of the agent's capabilities. On XLand-MiniGrid benchmarks, ULEE pre-training yields improved exploration and adaptation abilities that generalize to novel objectives, environment dynamics, and map structures. The resulting policy attains improved zero-shot and few-shot performance, and provides a strong initialization for longer fine-tuning processes. It outperforms learning from scratch, DIAYN pre-training, and alternative curricula. Code is available at: https://github.com/Octavio-Pappalardo/ulee-jax
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。