arXiv:2410.22308cs.RO2024-10ICRA被引 15

让无人机无需重训就能跑新赛道,靠动态调整环境来训练。

Environment as Policy: Learning to Race in Unseen Tracks

  • 用另一个强化学习策略动态设计训练环境,平衡难易度。
  • 单个策略在仿真和真实世界中均能成功通过复杂未见赛道。
  • 适合想提升机器人泛化能力的研究者与开发者。

强化学习在复杂机器人控制任务中表现卓越,例如无人机竞速,其智能体已在已知赛道上超越人类冠军。然而,当面对未见过的赛道布局时,这些智能体表现失效,需重新训练。本文旨在开发无需重训即可泛化至新赛道的强化学习智能体。直接在多样赛道上训练会因环境复杂性过高而阻碍学习效果。为此,提出一种自适应环境塑造框架,利用次级强化学习策略动态调整训练环境,确保挑战性与可达成性之间的平衡,使智能体能逐步适应并提升。实验表明,仅需一个单一策略,该方法便能在仿真与真实世界中成功完成多种复杂未见赛道竞速,显著优于现有环境塑造技术。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has achieved outstanding success in complex robot control tasks, such as drone racing, where the RL agents have outperformed human champions in a known racing track. However, these agents fail in unseen track configurations, always requiring complete retraining when presented with new track layouts. This work aims to develop RL agents that generalize effectively to novel track configurations without retraining. The naive solution of training directly on a diverse set of track layouts can overburden the agent, resulting in suboptimal policy learning as the increased complexity of the environment impairs the agent's ability to learn to fly. To enhance the generalizability of the RL agent, we propose an adaptive environment-shaping framework that dynamically adjusts the training environment based on the agent's performance. We achieve this by leveraging a secondary RL policy to design environments that strike a balance between being challenging and achievable, allowing the agent to adapt and improve progressively. Using our adaptive environment shaping, one single racing policy efficiently learns to race in diverse challenging tracks. Experimental results validated in both simulation and the real world show that our method enables drones to successfully fly complex and unseen race tracks, outperforming existing environment-shaping techniques. Project page: http://rpg.ifi.uzh.ch/env_as_policy.

强化学习无人机竞速泛化能力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。