atlas

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

Mastering Robot Manipulation with Multimodal Prompts through Pretraining and Multi-task Fine-tuning

ICML 2024 · Vision-language and multimodal › Robot action and navigation · approach: Multimodal prompt-driven manipulation · 1 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 MIDAS, a two-stage pipeline (inverse dynamics pretraining on motion-following tasks, then multi-task finetuning) with a T5 prompt encoder plus residual connection and autoregressive per-dimension action decoding, reaching new SOTA on VIMA-BENCH and showing in-context learning on held-out tasks.

Problem

Training robots to follow multimodal prompts that interleave images and text, which requires understanding transition dynamics and fine-grained visual details, and dependencies among action dimensions.

Approach

Reformulate any robot trajectory as a motion-following task for inverse dynamics pretraining, augment a pretrained T5 prompt encoder with a residual connection from visual tokens, and decode each action dimension as an autoregressive token to model inter-dimension dependencies.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add inverse-dynamics pretraining (ablation)MIDAS · VIMA-BENCH L1success rate (Avg): 91.6 -> 97.8Helped
add inverse-dynamics pretraining (ablation)MIDAS · VIMA-BENCH L2success rate (Avg): 91.8 -> 97.9Helped
add inverse-dynamics pretraining (ablation)MIDAS · VIMA-BENCH L3success rate (Avg): 88.3 -> 93.4Helped
add inverse-dynamics pretraining (ablation)MIDAS · VIMA-BENCH L4success rate (Avg): 49.1 -> 59.1Helped
replace motion-following inverse-dynamics pretraining with masked-action pretraining (ablation)MIDAS · VIMA-BENCH L1success rate (Overall): 97.8 -> 86.0Hurt
replace motion-following inverse-dynamics pretraining with masked-action pretraining (ablation)MIDAS · VIMA-BENCH L2success rate (Overall): 97.9 -> 86.3Hurt
replace motion-following inverse-dynamics pretraining with masked-action pretraining (ablation)MIDAS · VIMA-BENCH L3success rate (Overall): 93.4 -> 86.0Hurt
replace motion-following inverse-dynamics pretraining with masked-action pretraining (ablation)MIDAS · VIMA-BENCH L4success rate (Overall): 59.1 -> 48.5Hurt
replace decoder-only architecture with encoder-decoder architecture (ablation)MIDAS · VIMA-BENCH L1success rate (Overall): 97.8 -> 91.0Hurt
replace decoder-only architecture with encoder-decoder architecture (ablation)MIDAS · VIMA-BENCH L2success rate (Overall): 97.9 -> 90.5Hurt
replace decoder-only architecture with encoder-decoder architecture (ablation)MIDAS · VIMA-BENCH L3success rate (Overall): 93.4 -> 87.2Hurt
replace decoder-only architecture with encoder-decoder architecture (ablation)MIDAS · VIMA-BENCH L4success rate (Overall): 59.1 -> 69.8Helped

Surprising findings

Experimental setup

Datasets: VIMA-BENCH

Benchmarks: VIMA-BENCH

Metrics: success rate

Baselines: VIMA Gato OBJ Gato Flamingo GPT

Builds on: VIMA Frozen Gato SayCan

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