通过可到达性加权采样,提升离线目标条件强化学习性能
Reachability Weighted Offline Goal-conditioned Resampling
- 用可达性分类器为状态-动作对分配优先级,指导采样
- 在六项机器人任务中平均提升性能,手部抓取任务提升近50%
- 无需修改主算法,可直接接入现有离线RL框架
离线目标条件强化学习依赖固定数据集,其中许多潜在目标未在轨迹中显式表示。传统均匀采样需海量数据且生成大量不可达状态-目标-动作组合,损害策略表现。本文提出可达性加权采样(RWS),利用正-未标记学习训练的可达性分类器,将目标条件状态-动作值映射为可达性分数,并以此作为采样优先级。RWS为即插即用模块,可无缝集成至标准离线强化学习算法。在六项复杂仿真机器人操作任务(含机械臂与灵巧手)上的实验表明,RWS显著提升性能,在HandBlock-Z任务中相对基线提升近50%。结果验证了可达性加权采样的有效性。
原文摘要 · Abstract (English)
Offline goal-conditioned reinforcement learning (RL) relies on fixed datasets where many potential goals share the same state and action spaces. However, these potential goals are not explicitly represented in the collected trajectories. To learn a generalizable goal-conditioned policy, it is common to sample goals and state-action pairs uniformly using dynamic programming methods such as Q-learning. Uniform sampling, however, requires an intractably large dataset to cover all possible combinations and creates many unreachable state-goal-action pairs that degrade policy performance. Our key insight is that sampling should favor transitions that enable goal achievement. To this end, we propose Reachability Weighted Sampling (RWS). RWS uses a reachability classifier trained via positive-unlabeled (PU) learning on goal-conditioned state-action values. The classifier maps these values to a reachability score, which is then used as a sampling priority. RWS is a plug-and-play module that integrates seamlessly with standard offline RL algorithms. Experiments on six complex simulated robotic manipulation tasks, including those with a robot arm and a dexterous hand, show that RWS significantly improves performance. In one notable case, performance on the HandBlock-Z task improved by nearly 50 percent relative to the baseline. These results indicate the effectiveness of reachability-weighted sampling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。