atlas

Research map / Large language models / Language, safety and interpretability

Large Language Models as Generalizable Policies for Embodied Tasks

ICLR 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 8 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LLaRP, which adapts a frozen pretrained LLM (LLaMA-7B) with trained observation encoder and action head via PPO RL to act from egocentric vision and language instructions, and releases the Language Rearrangement benchmark.

Problem

Generalizable language-conditioned embodied policies that take visual observations and text instructions and act directly in the environment; existing LLM-for-embodied work relies on static expert data or zero-shot planning that is blind to vision.

Approach

Use a pre-trained frozen LLM backbone (input/output layers stripped) with learnable visual encoder MLP and action output MLP, trained only by online PPO RL with sparse/subgoal rewards, so the LLM's world knowledge transfers to unseen instructions and behaviors.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use LLaRP instead of the strongest learned baseline, LSTM-FlanLanguage Rearrangement evaluation · Unseen tasks (Total)success rate: 25 { $ \pm$ 1 } -> 42 { $ \pm$ 2 }Helped
Use LLaRP instead of LSTM-FlanLanguage Rearrangement evaluation · Behavior Generalizationsuccess rate: 28 { $ \pm$ 1 } -> 45 { $ \pm$ 3 }Helped
Use a pretrained LLaMA backbone rather than training the transformer from scratch (ablation)LLaRP · Language Rearrangement unseen tasks (Total)success rate: 17 { $ \pm$ 4 } -> 42 { $ \pm$ 2 }Helped
Scale the LLaMA backbone from 7B to 13B (ablation)LLaRP · Language Rearrangement evaluationsuccess rate: +4%Helped
Train with reinforcement learning instead of imitation learningLLaRP · Language Rearrangementsuccess rateHelped
Use a LLaRP LLM-based policy rather than LSTM-FlanLanguage Rearrangement continual learning · Multiple Rearrangementsenvironment steps to near-perfect performance: 1.5M vs 500kHelped
Use a pretrained LLM backbone rather than training the transformer from scratch (ablation)LLaRP · Atari 2600 (55 games)human-normalized score: between 3x and 4xHelped

Surprising findings

Limitations

Experimental setup

Datasets: Language Rearrangement (150k train instructions) ReplicaCAD YCB

Benchmarks: Language Rearrangement Atari 2600 (55 games)

Metrics: success rate human-normalized score interquartile mean sample efficiency

Baselines: ZS-ChatGPT ZS-LLaMA ZS-Flamingo LLaRP-Scratch LSTM-Flan LSTM-LLaMA

Builds on: PPO DD-PPO Habitat 2.0 VC-1

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy