arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.04136cs.LGcs.CVstat.ML

CURL: Contrastive Unsupervised Representations for Reinforcement Learning

University of California, Berkeley, BAIR(加州大学伯克利分校伯克利人工智能研究所)

Aravind Srinivas, Michael Laskin, Pieter Abbeel

英文摘要:

We present CURL: Contrastive Unsupervised Representations for Reinforcement Learning. CURL extracts high-level features from raw pixels using contrastive learning and performs off-policy control on top of the extracted features. CURL outperforms prior pixel-based methods, both model-based and model-free, on complex tasks in the DeepMind Control Suite and Atari Games showing 1.9x and 1.2x performance gains at the 100K environment and interaction steps benchmarks respectively. On the DeepMind Control Suite, CURL is the first image-based algorithm to nearly match the sample-efficiency of methods that use state-based features. Our code is open-sourced and available at https://github.com/MishaLaskin/curl.

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

↑