Swooper: Learning High-Speed Aerial Grasping With a Simple Gripper

TL;DR

Swooper introduces a deep reinforcement learning approach with a lightweight neural network for high-speed aerial grasping, achieving real-time control and high success rates on a simple quadrotor platform.

Contribution

The paper presents a two-stage training strategy for DRL-based aerial grasping, enabling rapid training and deployment on a low-cost onboard computer with high success rates.

Findings

84% grasp success rate in real-world trials

Training completes in under 60 minutes on a standard desktop

Inference time of about 1.0 ms per prediction

Abstract

High-speed aerial grasping presents significant challenges due to the high demands on precise, responsive flight control and coordinated gripper manipulation. In this work, we propose Swooper, a deep reinforcement learning (DRL) based approach that achieves both precise flight control and active gripper control using a single lightweight neural network policy. Training such a policy directly via DRL is nontrivial due to the complexity of coordinating flight and grasping. To address this, we adopt a two-stage learning strategy: we first pre-train a flight control policy, and then fine-tune it to acquire grasping skills. With the carefully designed reward functions and training framework, the entire training process completes in under 60 minutes on a standard desktop with an Nvidia RTX 3060 GPU. To validate the trained policy in the real world, we develop a lightweight quadrotor grasping platform equipped with a simple off-the-shelf gripper, and deploy the policy in a zero-shot manner on the onboard Raspberry Pi 4B computer, where each inference takes only about 1.0 ms. In 25 real-world trials, our policy achieves an 84% grasp success rate and grasping speeds of up to 1.5 m/s without any fine-tuning. This matches the robustness and agility of state-of-the-art classical systems with sophisticated grippers, highlighting the capability of DRL for learning a robust control policy that seamlessly integrates high-speed flight and grasping. The supplementary video is available for more results. Video: https://zikenhuang.github.io/Swooper/.