arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.16618cs.ROcs.AI

基于RL的控制器与PID控制器在6-DOF游泳机器人上的比较:混合水下目标跟踪

A comparison of RL-based and PID controllers for 6-DOF swimming robots: hybrid underwater object tracking

McGill University(麦吉尔大学);Independent Robotics Inc(独立机器人公司)

Faraz Lotfi, Khalil Virji, Nicholas Dudek, Gregory Dudek

AI总结:

针对6自由度游泳机器人水下目标跟踪场景,提出先用经典控制器安全探索再切换为集中式DQN控制的方案,在Unity仿真中验证其性能优于独立PID控制器,提供了可训练水下RL智能体的有效框架。

摘要:

本文针对6自由度(6DOF)游泳机器人领域普遍使用的PID控制器,探索并评估了采用集中式深度Q网络(deep Q-network, DQN)控制器作为替代方案的效果,重点以水下目标跟踪这一具体场景展示该控制方案的转换过程。DQN具备数据效率高、离策略学习的优势,且相较于其他强化学习(reinforcement learning, RL)方法实现更简单。由于研究使用的机器人缺乏动力学模型,我们提出了一种RL智能体来控制该多输入多输出(multi-input-multi-output, MIMO)系统,集中式控制器相比独立的PID控制器可提供更鲁棒的控制效果。我们的方法是先使用经典控制器完成安全探索,再逐步切换为DQN完全接管机器人控制。\n我们将水下跟踪任务划分为视觉模块和控制模块,采用成熟的视觉跟踪方法,同时引入集中式DQN控制器。通过将视觉模块输出的边界框数据传输至控制模块,可实现对不同目标的适配,且视觉系统可轻松替换。此外,低维数据的处理也让控制器的在线学习成本更低。我们在基于Unity的模拟器中开展了实验,验证了集中式RL智能体相较于独立PID控制器的有效性,证明了所提框架可用于训练水下RL智能体,且性能优于传统控制方法。真实场景与仿真实现的代码可在https://github.com/FARAZLOTFI/underwater-object-tracking 获取。

英文摘要:

In this paper, we present an exploration and assessment of employing a centralized deep Q-network (DQN) controller as a substitute for the prevalent use of PID controllers in the context of 6DOF swimming robots. Our primary focus centers on illustrating this transition with the specific case of underwater object tracking. DQN offers advantages such as data efficiency and off-policy learning, while remaining simpler to implement than other reinforcement learning methods. Given the absence of a dynamic model for our robot, we propose an RL agent to control this multi-input-multi-output (MIMO) system, where a centralized controller may offer more robust control than distinct PIDs. Our approach involves initially using classical controllers for safe exploration, then gradually shifting to DQN to take full control of the robot. We divide the underwater tracking task into vision and control modules. We use established methods for vision-based tracking and introduce a centralized DQN controller. By transmitting bounding box data from the vision module to the control module, we enable adaptation to various objects and effortless vision system replacement. Furthermore, dealing with low-dimensional data facilitates cost-effective online learning for the controller. Our experiments, conducted within a Unity-based simulator, validate the effectiveness of a centralized RL agent over separated PID controllers, showcasing the applicability of our framework for training the underwater RL agent and improved performance compared to traditional control methods. The code for both real and simulation implementations is at https://github.com/FARAZLOTFI/underwater-object-tracking.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑