MeMo: Meaningful, Modular Controllers via Noise Injection

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Tjandrasuwita, Megan, Xu, Jie, Solar-Lezama, Armando, Matusik, Wojciech
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910821009850368
author Tjandrasuwita, Megan
Xu, Jie
Solar-Lezama, Armando
Matusik, Wojciech
author_facet Tjandrasuwita, Megan
Xu, Jie
Solar-Lezama, Armando
Matusik, Wojciech
contents Robots are often built from standardized assemblies, (e.g. arms, legs, or fingers), but each robot must be trained from scratch to control all the actuators of all the parts together. In this paper we demonstrate a new approach that takes a single robot and its controller as input and produces a set of modular controllers for each of these assemblies such that when a new robot is built from the same parts, its control can be quickly learned by reusing the modular controllers. We achieve this with a framework called MeMo which learns (Me)aningful, (Mo)dular controllers. Specifically, we propose a novel modularity objective to learn an appropriate division of labor among the modules. We demonstrate that this objective can be optimized simultaneously with standard behavior cloning loss via noise injection. We benchmark our framework in locomotion and grasping environments on simple to complex robot morphology transfer. We also show that the modules help in task transfer. On both structure and task transfer, MeMo achieves improved training efficiency to graph neural network and Transformer baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2407_01567
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MeMo: Meaningful, Modular Controllers via Noise Injection
Tjandrasuwita, Megan
Xu, Jie
Solar-Lezama, Armando
Matusik, Wojciech
Robotics
Machine Learning
Robots are often built from standardized assemblies, (e.g. arms, legs, or fingers), but each robot must be trained from scratch to control all the actuators of all the parts together. In this paper we demonstrate a new approach that takes a single robot and its controller as input and produces a set of modular controllers for each of these assemblies such that when a new robot is built from the same parts, its control can be quickly learned by reusing the modular controllers. We achieve this with a framework called MeMo which learns (Me)aningful, (Mo)dular controllers. Specifically, we propose a novel modularity objective to learn an appropriate division of labor among the modules. We demonstrate that this objective can be optimized simultaneously with standard behavior cloning loss via noise injection. We benchmark our framework in locomotion and grasping environments on simple to complex robot morphology transfer. We also show that the modules help in task transfer. On both structure and task transfer, MeMo achieves improved training efficiency to graph neural network and Transformer baselines.
title MeMo: Meaningful, Modular Controllers via Noise Injection
topic Robotics
Machine Learning
url https://arxiv.org/abs/2407.01567