Deep Reinforcement Learning Based on Local GNN for Goal-conditioned Deformable Object Rearranging

Abstract

Object rearranging is one of the most common deformable manipulation tasks, where the robot needs to rearrange a deformable object into a goal configuration. Previous studies focus on designing an expert system for each specific task by model-based or data-driven approaches and the application scenarios are therefore limited. Some research has been attempting to design a general framework to obtain more advanced manipulation capabilities for deformable rearranging tasks, with lots of progress achieved in simulation. However, transferring from simulation to reality is difficult due to the limitation of the end-to-end CNN architecture. To address these challenges, we design a local GNN (Graph Neural Network) based learning method, which utilizes two representation graphs to encode keypoints detected from images. Self-attention is applied for graph updating and cross-attention is applied for generating manipulation actions. Extensive experiments have been conducted to demonstrate that our framework is effective in multiple 1-D (rope, rope ring) and 2-D (cloth) rearranging tasks in simulation and can be easily transferred to a real robot by fine-tuning a keypoint detector.

arXiv commenthas been accepted by IEEE/RSJ International Conference on Intelligent Robots and Systems 2022

References in corpus (5)

• Unsupervised Learning of Object Keypoints for Perception and Control
• CLIPort: What and Where Pathways for Robotic Manipulation
• Deep Reinforcement Learning for Robotic Pushing and Picking in Cluttered Environment
• Learning Predictive Representations for Deformable Objects Using Contrastive Estimation
• Manipulating Deformable Objects by Interleaving Prediction, Planning, and Control