A Pretrained Centralised Model-Based Reinforcement Learning Method for Heterogeneous Collaborative Grasping
摘要
In this paper, we propose a pretrained centralised model-based reinforcement learning method for collaborative grasping by a team of aerial and ground robots. Our method leverages demonstration data generated from the optimal reciprocal collision avoidance (ORCA) algorithm and PID controller to pretrain a dynamics model for the multi-agent system. This model is then utilised to optimise a centralised policy through model predictive control (MPC). We comprehensively conduct experiments under the scenarios with the load of three, four and six grasping points. The experiment results demonstrate that the proposed method outperforms model-free baselines, achieving higher success rates and faster convergence. Furthermore, the algorithm can be applied in a heterogeneous robot team for the collaborative grasping.