atlas

Research map / Vision-language and multimodal / Robot action and navigation

Vision-Language Foundation Models as Effective Robot Imitators

ICLR 2024 · Vision-language and multimodal › Robot action and navigation · approach: Vision-language-action foundation models · 19 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.

Adapts the open-source OpenFlamingo VLM into a robot manipulation policy by adding an LSTM policy head and fine-tuning only the cross-attention, resampler and head via imitation learning on CALVIN.

Problem

Adapting pre-trained vision-language models for low-level, closed-loop language-conditioned robot manipulation, which is hard because VLMs are trained on static image-text pairs and output text rather than actions, and prior approaches need private models or huge co-training data.

Approach

Decouple single-step vision-language understanding (a Flamingo/OpenFlamingo backbone) from history modeling (an explicit LSTM policy head over max-pooled fused tokens), and fine-tune only the resampler, gated cross-attention and policy head with imitation learning on language-annotated manipulation data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use RoboFlamingo instead of HULClanguage-conditioned manipulation policy · CALVIN ABCD→DAvg Len: 3.06 -> 4.09Helped
Use RoboFlamingo instead of RT-1language-conditioned manipulation policy · CALVIN ABC→DAvg Len: 0.90 -> 2.48Helped
freeze the fusion decoder embedding layer (ablation)RoboFlamingo · CALVIN ABCD→D (Enriched)Avg Len: 1.85 -> 2.12Helped
instruction fine-tune the M-3B backbone (ablation)RoboFlamingo with M-3B · CALVIN ABCD→D (10% language data)Avg Len: 0.05 -> 0.13Helped
instruction fine-tune the G-4B backbone (ablation)RoboFlamingo with G-4B · CALVIN ABCD→D (10% language data)Avg Len: 0.48 -> 0.55Helped
increase backbone size from M-3B to M-9B (ablation)RoboFlamingo · CALVIN ABCD→D (10% language data)Avg Len: 0.05 -> 0.83Helped
fine-tune Voltron representations instead of freezing them (ablation)Voltron policy · CALVIN ABC→DAvg Len: 0.03 -> 1.00Helped
co-train with COCO and VQA data instead of robotics-only fine-tuning (ablation)RoboFlamingo M-3B-IFT · CALVIN ABCD→DAvg Len: 4.09 -> 3.76Hurt
fine-tune all model parameters instead of only the resampler, gated cross-attention, and policy head (ablation)RoboFlamingo M-3B-IFT · CALVIN ABCD→DAvg Len: 4.09 -> 0.50Hurt
use a single-step MLP without observation history instead of a history-aware policy head (ablation)RoboFlamingo · CALVIN ABCD→DAvg LenHurt
use MLP with history in the fusion decoder instead of GPT or LSTM policy heads (ablation)RoboFlamingo · CALVIN ABCD→DAvg LenHurt
remove vision-language pre-training from the resampler and cross-attention layers (ablation)RoboFlamingo · CALVIN ABCD→DAvg LenHurt

Surprising findings

Limitations

Experimental setup

Datasets: CALVIN COCO VQAv2

Benchmarks: CALVIN

Metrics: success rate per task in a 5-task sequence average sequence length

Baselines: MCIL HULC RT-1 Voltron R3M

Builds on: Flamingo OpenFlamingo RT-1 HULC

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