On Time-Indexing as Inductive Bias in Deep RL for Sequential Manipulation Tasks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qureshi, M. Nomaan, Eisner, Ben, Held, David
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914629575245824
author Qureshi, M. Nomaan
Eisner, Ben
Held, David
author_facet Qureshi, M. Nomaan
Eisner, Ben
Held, David
contents While solving complex manipulation tasks, manipulation policies often need to learn a set of diverse skills to accomplish these tasks. The set of skills is often quite multimodal - each one may have a quite distinct distribution of actions and states. Standard deep policy-learning algorithms often model policies as deep neural networks with a single output head (deterministic or stochastic). This structure requires the network to learn to switch between modes internally, which can lead to lower sample efficiency and poor performance. In this paper we explore a simple structure which is conducive to skill learning required for so many of the manipulation tasks. Specifically, we propose a policy architecture that sequentially executes different action heads for fixed durations, enabling the learning of primitive skills such as reaching and grasping. Our empirical evaluation on the Metaworld tasks reveals that this simple structure outperforms standard policy learning methods, highlighting its potential for improved skill acquisition.
format Preprint
id arxiv_https___arxiv_org_abs_2401_01993
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On Time-Indexing as Inductive Bias in Deep RL for Sequential Manipulation Tasks
Qureshi, M. Nomaan
Eisner, Ben
Held, David
Robotics
Artificial Intelligence
While solving complex manipulation tasks, manipulation policies often need to learn a set of diverse skills to accomplish these tasks. The set of skills is often quite multimodal - each one may have a quite distinct distribution of actions and states. Standard deep policy-learning algorithms often model policies as deep neural networks with a single output head (deterministic or stochastic). This structure requires the network to learn to switch between modes internally, which can lead to lower sample efficiency and poor performance. In this paper we explore a simple structure which is conducive to skill learning required for so many of the manipulation tasks. Specifically, we propose a policy architecture that sequentially executes different action heads for fixed durations, enabling the learning of primitive skills such as reaching and grasping. Our empirical evaluation on the Metaworld tasks reveals that this simple structure outperforms standard policy learning methods, highlighting its potential for improved skill acquisition.
title On Time-Indexing as Inductive Bias in Deep RL for Sequential Manipulation Tasks
topic Robotics
Artificial Intelligence
url https://arxiv.org/abs/2401.01993