Search papers, labs, and topics across Lattice.
This paper introduces QDOS, a novel pipeline for offline-to-online reinforcement learning that enhances policy performance by integrating a Quality-Diversity pretraining objective with a dual dataset reuse strategy. By weighting skill extraction based on the estimated advantage of trajectory segments, QDOS effectively extracts diverse and high-value skills, leading to improved representations in the low-level policy. Experimental results show that QDOS outperforms existing methods in both structured manipulation and unstructured locomotion tasks, demonstrating its efficacy in accelerating exploration and boosting performance in sparse-reward environments.
QDOS transforms offline datasets into a treasure trove of high-value skills, significantly enhancing reinforcement learning performance in challenging tasks.
Recent studies investigate how to leverage pre-collected datasets to improve the policy performance and sample efficiency of RL. One promising approach to achieve this goal is to employ a two-stage strategy: In the first stage, diverse skills are extracted as a low-level policy from a given dataset, and a high-level policy is trained to solve a specific task in the second stage. Typically, extraction of the low-level policy is performed based on unsupervised learning such as trajectory VAE. However, a limitation of this approach is that the quality of the low-level policy highly depends on the quality of the dataset. To address this issue, we introduce QDOS (Quality-Diversity Offline Skill learning), a unified pipeline for robust offline-to-online learning. Our approach incorporates an Advantage-Weighted Quality-Diversity pretraining objective, which weights the skill extraction and diversity objectives by the estimated advantage of each trajectory segment. This approach allows the model to extract diverse and high-value skills. By providing robust and task-relevant skill representations, QDOS significantly improves the quality of the embedded skill space used by the low-level policy. We further integrate this with a dual dataset reuse strategy, where offline data is used both for skill pretraining and for populating the online replay buffer via pseudo-labeling. Experiments demonstrate that QDOS significantly outperforms strong baselines in structured manipulation tasks and unstructured locomotion tasks, confirming its ability to accelerate exploration and improve final returns in challenging sparse-reward domains.
