Loading video...

Video Failed to Load

Go Home

Goal-conditioned RL (GCRL) is great - unsupervised, can use data (in offline mode), flexibility to define tasks at test time. But can we run GCRL on *language data*?? In our new work we show that language GCRL enables sophisticated test-time reasoning for interactive tasks! 🧵👇

18,782 views • 1 year ago •via X (Twitter)

5 Comments

Sergey Levine's profile picture
Sergey Levine1 year ago

The idea is to train goal-conditioned value functions on interaction data (the game Avalon, WebShop, etc.). These value functions predict how likely different outcomes are given a high-level action, e.g., if I accuse Player 2 in Avalon, how likely is that to raise suspicion?

Sergey Levine's profile picture
Sergey Levine1 year ago

At evaluation time, the agent can use these goal-conditioned value functions as part of a test-time compute self-refinement loop, evaluating its actions along many different dimensions via long-horizon prediction.

Sergey Levine's profile picture
Sergey Levine1 year ago

Here are some examples of real interactions with our agent, including reasoning and prediction from the value functions.

Sergey Levine's profile picture
Sergey Levine1 year ago

A really fun project led by Joey Hong, with @ancadianadragan Website: Paper:

Wendy Carlosa's profile picture
Wendy Carlosa1 year ago

gcrl doesn't iterate long enough to see the immorality of making new children when I hate the ones you already did

Related Videos

3D-LLM: Injecting the 3D World into Large Language Models paper page: Large language models (LLMs) and Vision-Language Models (VLMs) have been proven to excel at multiple tasks, such as commonsense reasoning. Powerful as these models can be, they are not grounded in the 3D physical world, which involves richer concepts such as spatial relationships, affordances, physics, layout, and so on. In this work, we propose to inject the 3D world into large language models and introduce a whole new family of 3D-LLMs. Specifically, 3D-LLMs can take 3D point clouds and their features as input and perform a diverse set of 3D-related tasks, including captioning, dense captioning, 3D question answering, task decomposition, 3D grounding, 3D-assisted dialog, navigation, and so on. Using three types of prompting mechanisms that we design, we are able to collect over 300k 3D-language data covering these tasks. To efficiently train 3D-LLMs, we first utilize a 3D feature extractor that obtains 3D features from rendered multi- view images. Then, we use 2D VLMs as our backbones to train our 3D-LLMs. By introducing a 3D localization mechanism, 3D-LLMs can better capture 3D spatial information. Experiments on ScanQA show that our model outperforms state-of-the-art baselines by a large margin (e.g., the BLEU-1 score surpasses state-of-the-art score by 9%). Furthermore, experiments on our held-in datasets for 3D captioning, task composition, and 3D-assisted dialogue show that our model outperforms 2D VLMs. Qualitative examples also show that our model could perform more tasks beyond the scope of existing LLMs and VLMs.

AK

249,572 views • 3 years ago