Offline-to-online Reinforcement Learning for Image-based Grasping with Scarce Demonstrations

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Chan, Bryan, Leung, Anson, Bergstra, James
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915117075005440
author Chan, Bryan
Leung, Anson
Bergstra, James
author_facet Chan, Bryan
Leung, Anson
Bergstra, James
contents Offline-to-online reinforcement learning (O2O RL) aims to obtain a continually improving policy as it interacts with the environment, while ensuring the initial policy behaviour is satisficing. This satisficing behaviour is necessary for robotic manipulation where random exploration can be costly due to catastrophic failures and time. O2O RL is especially compelling when we can only obtain a scarce amount of (potentially suboptimal) demonstrations$\unicode{x2014}$a scenario where behavioural cloning (BC) is known to suffer from distribution shift. Previous works have outlined the challenges in applying O2O RL algorithms under the image-based environments. In this work, we propose a novel O2O RL algorithm that can learn in a real-life image-based robotic vacuum grasping task with a small number of demonstrations where BC fails majority of the time. The proposed algorithm replaces the target network in off-policy actor-critic algorithms with a regularization technique inspired by neural tangent kernel. We demonstrate that the proposed algorithm can reach above 90\% success rate in under two hours of interaction time, with only 50 human demonstrations, while BC and existing commonly-used RL algorithms fail to achieve similar performance.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14957
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Offline-to-online Reinforcement Learning for Image-based Grasping with Scarce Demonstrations
Chan, Bryan
Leung, Anson
Bergstra, James
Robotics
Artificial Intelligence
Machine Learning
Offline-to-online reinforcement learning (O2O RL) aims to obtain a continually improving policy as it interacts with the environment, while ensuring the initial policy behaviour is satisficing. This satisficing behaviour is necessary for robotic manipulation where random exploration can be costly due to catastrophic failures and time. O2O RL is especially compelling when we can only obtain a scarce amount of (potentially suboptimal) demonstrations$\unicode{x2014}$a scenario where behavioural cloning (BC) is known to suffer from distribution shift. Previous works have outlined the challenges in applying O2O RL algorithms under the image-based environments. In this work, we propose a novel O2O RL algorithm that can learn in a real-life image-based robotic vacuum grasping task with a small number of demonstrations where BC fails majority of the time. The proposed algorithm replaces the target network in off-policy actor-critic algorithms with a regularization technique inspired by neural tangent kernel. We demonstrate that the proposed algorithm can reach above 90\% success rate in under two hours of interaction time, with only 50 human demonstrations, while BC and existing commonly-used RL algorithms fail to achieve similar performance.
title Offline-to-online Reinforcement Learning for Image-based Grasping with Scarce Demonstrations
topic Robotics
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.14957