Salvato in:
Dettagli Bibliografici
Autori principali: Bai, Chenjia, Yang, Rushuai, Zhang, Qiaosheng, Xu, Kang, Chen, Yi, Xiao, Ting, Li, Xuelong
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:https://arxiv.org/abs/2405.16030
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910459134738432
author Bai, Chenjia
Yang, Rushuai
Zhang, Qiaosheng
Xu, Kang
Chen, Yi
Xiao, Ting
Li, Xuelong
author_facet Bai, Chenjia
Yang, Rushuai
Zhang, Qiaosheng
Xu, Kang
Chen, Yi
Xiao, Ting
Li, Xuelong
contents Unsupervised Reinforcement Learning (RL) provides a promising paradigm for learning useful behaviors via reward-free per-training. Existing methods for unsupervised RL mainly conduct empowerment-driven skill discovery or entropy-based exploration. However, empowerment often leads to static skills, and pure exploration only maximizes the state coverage rather than learning useful behaviors. In this paper, we propose a novel unsupervised RL framework via an ensemble of skills, where each skill performs partition exploration based on the state prototypes. Thus, each skill can explore the clustered area locally, and the ensemble skills maximize the overall state coverage. We adopt state-distribution constraints for the skill occupancy and the desired cluster for learning distinguishable skills. Theoretical analysis is provided for the state entropy and the resulting skill distributions. Based on extensive experiments on several challenging tasks, we find our method learns well-explored ensemble skills and achieves superior performance in various downstream tasks compared to previous methods.
format Preprint
id arxiv_https___arxiv_org_abs_2405_16030
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Constrained Ensemble Exploration for Unsupervised Skill Discovery
Bai, Chenjia
Yang, Rushuai
Zhang, Qiaosheng
Xu, Kang
Chen, Yi
Xiao, Ting
Li, Xuelong
Machine Learning
Unsupervised Reinforcement Learning (RL) provides a promising paradigm for learning useful behaviors via reward-free per-training. Existing methods for unsupervised RL mainly conduct empowerment-driven skill discovery or entropy-based exploration. However, empowerment often leads to static skills, and pure exploration only maximizes the state coverage rather than learning useful behaviors. In this paper, we propose a novel unsupervised RL framework via an ensemble of skills, where each skill performs partition exploration based on the state prototypes. Thus, each skill can explore the clustered area locally, and the ensemble skills maximize the overall state coverage. We adopt state-distribution constraints for the skill occupancy and the desired cluster for learning distinguishable skills. Theoretical analysis is provided for the state entropy and the resulting skill distributions. Based on extensive experiments on several challenging tasks, we find our method learns well-explored ensemble skills and achieves superior performance in various downstream tasks compared to previous methods.
title Constrained Ensemble Exploration for Unsupervised Skill Discovery
topic Machine Learning
url https://arxiv.org/abs/2405.16030