atlas

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

SAM-E: Leveraging Visual Foundation Model with Sequence Imitation for Embodied Manipulation

ICML 2024 · Vision-language and multimodal › Robot action and navigation · approach: Memory and subgoal-based 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 SAM-E, a multi-view 3D manipulation policy that LoRA-fine-tunes a SAM ViT-B encoder and predicts a 5-step action sequence in a single pass via multi-channel heatmaps.

Problem

Multi-task 3D robot manipulation from language instructions requires scene understanding and long-horizon action prediction, but existing multi-view policies need many high-quality trajectories, generalize poorly to unseen tasks, and execute inefficiently step by step.

Approach

Use the promptable SAM image encoder (LoRA-fine-tuned on embodied data) as the visual foundation for a multi-view transformer, and add an action-sequence policy head that outputs multi-channel pose heatmaps for a short horizon of future keyframe actions, exploiting temporal smoothness of end-effector actions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use SAM-E with SAM and action-sequence prediction instead of RVTRLBench multi-task policy · RLBench 18-task multi-tasksuccess rate: 62.9 -> 70.6Helped
predict action sequences rather than next keypoint poses step by stepSAM-E · RLBench 18-task multi-taskinference steps (sum): 6158±64 -> 1130±12Helped
initialize the policy from multi-task training and adapt to new tasksSAM-E · RLBench few-shot adaptation, 6 taskssuccess rate: 37.1±1.0 -> 63.2±1.5Helped
use SAM-E rather than Hiveformer10-task comparison · Hiveformer 10-task comparisonsuccess rate: 83.3 -> 88.6±0.7Helped
replace the SAM encoder with RVT's encoder (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 65.3±0.6Hurt
add LoRA to the key projection (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 69.2±0.9Hurt
remove LoRA fine-tuning (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 67.2±1.0Hurt
fully fine-tune the SAM encoder instead of using LoRA (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 65.8±1.0Hurt
set the action horizon to 1 rather than 5 (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 30.6±1.4Hurt
set the action horizon to 3 rather than 5 (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 64.0±0.6Hurt
set the action horizon to 7 rather than 5 (ablation)SAM-E · RLBench 18-task multi-tasksuccess rate: 70.6±0.7 -> 66.5±1.2Hurt
use an action horizon of 7 instead of 5 (ablation)SAM-E · RLBench 18-task multi-taskinference steps (sum): 1130±12 -> 919±12Helped

Surprising findings

Limitations

Experimental setup

Datasets: RLBench demonstrations (RVT/PerAct training set) real-world Franka demonstrations

Benchmarks: RLBench (18 tasks, 249 variations) RLBench few-shot adaptation (6 new tasks) real-world Franka tasks

Metrics: success rate inference steps inference time (ms)

Baselines: RVT PerAct R3M CLIP DINO Hiveformer SAM→RVT

Builds on: Segment Anything Model RVT PerAct LoRA CLIP

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