用欧几里得变换增强状态数据,提升连续控制的强化学习效率
Reinforcement Learning with Euclidean Data Augmentation for State-Based Continuous Control
- 基于肢体构型特征进行旋转等欧氏变换生成新数据
- 在多个任务上显著提升数据效率与最终性能
- 适合追求高效强化学习的机器人控制研究者
数据增强通过变换原始数据生成新样本,提升强化学习(RL)在连续控制任务中的数据效率。以往工作多采用扰动式增强(如随机裁剪、平移),在图像观测任务中表现优异。本文聚焦状态驱动的连续控制,提出基于欧氏对称性的新增强策略,将变换应用于肢体构型(即关节连接的刚体)而非传统关节配置。我们发现,传统关节配置特征不适用于欧氏变换;而肢体构型特征在旋转等变换下能生成丰富新数据。经少量超参数调整,该方法在多种基准任务上显著提升数据效率与最终性能。
原文摘要 · Abstract (English)
Data augmentation creates new data points by transforming the original ones for a reinforcement learning (RL) agent to learn from, which has been shown to be effective for the objective of improving the data efficiency of RL for continuous control. Prior work towards this objective has been largely restricted to perturbation-based data augmentation where new data points are created by perturbing the original ones, which has been impressively effective for tasks where the RL agent observes control states as images with perturbations including random cropping, shifting, etc. This work focuses on state-based control, where the RL agent can directly observe raw kinematic and task features, and considers an alternative data augmentation applied to these features based on Euclidean symmetries under transformations like rotations. We show that the default state features used in exiting benchmark tasks that are based on joint configurations are not amenable to Euclidean transformations. We therefore advocate using state features based on configurations of the limbs (i.e., the rigid bodies connected by the joints) that instead provide rich augmented data under Euclidean transformations. With minimal hyperparameter tuning, we show this new Euclidean data augmentation strategy significantly improves both data efficiency and asymptotic performance of RL on a wide range of continuous control tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。