Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tirinzoni, Andrea, Touati, Ahmed, Farebrother, Jesse, Guzek, Mateusz, Kanervisto, Anssi, Xu, Yingchen, Lazaric, Alessandro, Pirotta, Matteo
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916691461537792
author Tirinzoni, Andrea
Touati, Ahmed
Farebrother, Jesse
Guzek, Mateusz
Kanervisto, Anssi
Xu, Yingchen
Lazaric, Alessandro
Pirotta, Matteo
author_facet Tirinzoni, Andrea
Touati, Ahmed
Farebrother, Jesse
Guzek, Mateusz
Kanervisto, Anssi
Xu, Yingchen
Lazaric, Alessandro
Pirotta, Matteo
contents Unsupervised reinforcement learning (RL) aims at pre-training agents that can solve a wide range of downstream tasks in complex environments. Despite recent advancements, existing approaches suffer from several limitations: they may require running an RL process on each downstream task to achieve a satisfactory performance, they may need access to datasets with good coverage or well-curated task-specific samples, or they may pre-train policies with unsupervised losses that are poorly correlated with the downstream tasks of interest. In this paper, we introduce a novel algorithm regularizing unsupervised RL towards imitating trajectories from unlabeled behavior datasets. The key technical novelty of our method, called Forward-Backward Representations with Conditional-Policy Regularization, is to train forward-backward representations to embed the unlabeled trajectories to the same latent space used to represent states, rewards, and policies, and use a latent-conditional discriminator to encourage policies to ``cover'' the states in the unlabeled behavior dataset. As a result, we can learn policies that are well aligned with the behaviors in the dataset, while retaining zero-shot generalization capabilities for reward-based and imitation tasks. We demonstrate the effectiveness of this new approach in a challenging humanoid control problem: leveraging observation-only motion capture datasets, we train Meta Motivo, the first humanoid behavioral foundation model that can be prompted to solve a variety of whole-body tasks, including motion tracking, goal reaching, and reward optimization. The resulting model is capable of expressing human-like behaviors and it achieves competitive performance with task-specific methods while outperforming state-of-the-art unsupervised RL and model-based baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2504_11054
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Models
Tirinzoni, Andrea
Touati, Ahmed
Farebrother, Jesse
Guzek, Mateusz
Kanervisto, Anssi
Xu, Yingchen
Lazaric, Alessandro
Pirotta, Matteo
Machine Learning
Unsupervised reinforcement learning (RL) aims at pre-training agents that can solve a wide range of downstream tasks in complex environments. Despite recent advancements, existing approaches suffer from several limitations: they may require running an RL process on each downstream task to achieve a satisfactory performance, they may need access to datasets with good coverage or well-curated task-specific samples, or they may pre-train policies with unsupervised losses that are poorly correlated with the downstream tasks of interest. In this paper, we introduce a novel algorithm regularizing unsupervised RL towards imitating trajectories from unlabeled behavior datasets. The key technical novelty of our method, called Forward-Backward Representations with Conditional-Policy Regularization, is to train forward-backward representations to embed the unlabeled trajectories to the same latent space used to represent states, rewards, and policies, and use a latent-conditional discriminator to encourage policies to ``cover'' the states in the unlabeled behavior dataset. As a result, we can learn policies that are well aligned with the behaviors in the dataset, while retaining zero-shot generalization capabilities for reward-based and imitation tasks. We demonstrate the effectiveness of this new approach in a challenging humanoid control problem: leveraging observation-only motion capture datasets, we train Meta Motivo, the first humanoid behavioral foundation model that can be prompted to solve a variety of whole-body tasks, including motion tracking, goal reaching, and reward optimization. The resulting model is capable of expressing human-like behaviors and it achieves competitive performance with task-specific methods while outperforming state-of-the-art unsupervised RL and model-based baselines.
title Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Models
topic Machine Learning
url https://arxiv.org/abs/2504.11054