PianoMime: Learning a Generalist, Dexterous Piano Player from Internet Demonstrations

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Qian, Cheng, Urain, Julen, Zakka, Kevin, Peters, Jan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929437258285056
author Qian, Cheng
Urain, Julen
Zakka, Kevin
Peters, Jan
author_facet Qian, Cheng
Urain, Julen
Zakka, Kevin
Peters, Jan
contents In this work, we introduce PianoMime, a framework for training a piano-playing agent using internet demonstrations. The internet is a promising source of large-scale demonstrations for training our robot agents. In particular, for the case of piano-playing, Youtube is full of videos of professional pianists playing a wide myriad of songs. In our work, we leverage these demonstrations to learn a generalist piano-playing agent capable of playing any arbitrary song. Our framework is divided into three parts: a data preparation phase to extract the informative features from the Youtube videos, a policy learning phase to train song-specific expert policies from the demonstrations and a policy distillation phase to distil the policies into a single generalist agent. We explore different policy designs to represent the agent and evaluate the influence of the amount of training data on the generalization capability of the agent to novel songs not available in the dataset. We show that we are able to learn a policy with up to 56\% F1 score on unseen songs.
format Preprint
id arxiv_https___arxiv_org_abs_2407_18178
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PianoMime: Learning a Generalist, Dexterous Piano Player from Internet Demonstrations
Qian, Cheng
Urain, Julen
Zakka, Kevin
Peters, Jan
Computer Vision and Pattern Recognition
Artificial Intelligence
Robotics
In this work, we introduce PianoMime, a framework for training a piano-playing agent using internet demonstrations. The internet is a promising source of large-scale demonstrations for training our robot agents. In particular, for the case of piano-playing, Youtube is full of videos of professional pianists playing a wide myriad of songs. In our work, we leverage these demonstrations to learn a generalist piano-playing agent capable of playing any arbitrary song. Our framework is divided into three parts: a data preparation phase to extract the informative features from the Youtube videos, a policy learning phase to train song-specific expert policies from the demonstrations and a policy distillation phase to distil the policies into a single generalist agent. We explore different policy designs to represent the agent and evaluate the influence of the amount of training data on the generalization capability of the agent to novel songs not available in the dataset. We show that we are able to learn a policy with up to 56\% F1 score on unseen songs.
title PianoMime: Learning a Generalist, Dexterous Piano Player from Internet Demonstrations
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Robotics
url https://arxiv.org/abs/2407.18178