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

通过对比学习发现强化学习中的层级成就

Discovering Hierarchical Achievements in Reinforcement Learning via Contrastive Learning

Seoul National University(首尔大学);KRAFTON(魁匠团)

Seungyong Moon, Junyoung Yeom, Bumsoo Park, Hyun Oh Song

AI总结:

针对程序生成环境中的层级成就发现问题,本文提出成就蒸馏对比学习方法,增强 PPO 对下一成就的预测能力,并在 Crafter 上以更少参数实现样本高效的最优性能。

摘要:

在程序生成环境中发现具有层级结构的成就,是一项重大挑战。这要求智能体具备广泛能力,包括泛化和长期推理。许多先前方法建立在基于模型或层级式方法之上,认为用于长期规划的显式模块将有利于学习层级依赖关系。然而,这些方法需要过多的环境交互或庞大的模型规模,从而限制了其实用性。在本研究中,我们证明近端策略优化(PPO)这一简单而通用的无模型算法,在采用近期实现实践进行优化时,表现优于先前方法。此外,我们发现 PPO 智能体能够在一定程度上预测下一个待解锁的成就,尽管置信度有限。基于这一观察,我们提出一种名为成就蒸馏的新型对比学习方法,用于增强智能体预测下一成就的能力。我们的方法展现出强大的层级成就发现能力,并以高样本效率在具有挑战性的 Crafter 环境上取得最先进性能,同时使用更少的模型参数。

英文摘要:

Discovering achievements with a hierarchical structure in procedurally generated environments presents a significant challenge. This requires an agent to possess a broad range of abilities, including generalization and long-term reasoning. Many prior methods have been built upon model-based or hierarchical approaches, with the belief that an explicit module for long-term planning would be advantageous for learning hierarchical dependencies. However, these methods demand an excessive number of environment interactions or large model sizes, limiting their practicality. In this work, we demonstrate that proximal policy optimization (PPO), a simple yet versatile model-free algorithm, outperforms previous methods when optimized with recent implementation practices. Moreover, we find that the PPO agent can predict the next achievement to be unlocked to some extent, albeit with limited confidence. Based on this observation, we introduce a novel contrastive learning method, called achievement distillation, which strengthens the agent's ability to predict the next achievement. Our method exhibits a strong capacity for discovering hierarchical achievements and shows state-of-the-art performance on the challenging Crafter environment in a sample-efficient manner while utilizing fewer model parameters.

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

↑