atlas

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

RoboMamba: Efficient Vision-Language-Action Model for Robotic Reasoning and Manipulation

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

Introduces RoboMamba, an efficient VLA model pairing a CLIP vision encoder with the Mamba 2.8B/1.4B LLM, plus a tiny frozen-backbone policy head for end-effector pose prediction.

Problem

Existing MLLM-based robotic VLA models have limited robotic reasoning and are computationally expensive to fine-tune and run due to attention-based LLMs.

Approach

Replace the attention LLM with the linear-complexity Mamba SSM in a VLA, train general and robotic reasoning via alignment pre-training and co-training, then learn manipulation by freezing the whole model and training only a small MLP policy head (3.7M params, ~0.1%) on Mamba's output tokens.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Mamba-2.7B instead of RWKV-3B (ablation)RoboMamba reasoning model · General and robotic-related reasoning benchmarksreasoning performanceHelped
Add RoboVQA robotic instruction data during co-training (ablation)RoboMamba with LLaVA 1.5 instruction data · RoboVQA validationBLEU-4: 26.5 -> 42.8Helped
Add RoboVQA robotic instruction data during co-training (ablation)RoboMamba with LLaVA 1.5 instruction data · POPEscore: 85.6 -> 86.3Helped
Add RoboVQA robotic instruction data during co-training (ablation)RoboMamba with ShareGPT4V-SFT · GQAscore: 64.3 -> 62.1Hurt
Add RoboVQA robotic instruction data during co-training (ablation)RoboMamba with ShareGPT4V-SFT · RoboVQA validationBLEU-4: 26.7 -> 42.5Helped
Add RoboVQA robotic instruction data during co-training (ablation)RoboMamba with LLaVA-Next · RoboVQA validationBLEU-4: 25.1 -> 43.0Helped
Replace LLaVA 1.5 instructions with ShareGPT4V-SFT (ablation)RoboMamba without RoboVQA co-training · GQAscore: 65.3 -> 64.3No clear effect
Replace LLaVA 1.5 instructions with LLaVA-Next (ablation)RoboMamba without RoboVQA co-training · RoboVQA validationBLEU-4: 26.5 -> 25.1No clear effect
Replace the CLIP encoder with XCiT (ablation)RoboMamba · RoboVQA validationBLEU-4: 35.2 -> 42.8No clear effect
Change CLIP image resolution from 224 x 224 to 336 x 336 (ablation)RoboMamba · RoboVQA validationBLEU-4: 35.2 -> 41.8No clear effect
Replace CLIP with SigLIP and use 384 x 384 resolution (ablation)RoboMamba · RoboVQA validationBLEU-4: 35.2 -> 40.6No clear effect
Use one shared MLP head instead of separate position and direction heads (ablation)RoboMamba manipulation policy · SAPIEN manipulation, seenaccuracy: 63.7% -> 62.1%No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: LLaVA-LCS 558K LLaVA-v1.5 655K ShareGPT4V-SFT LLaVA-Next RoboVQA PartNet-Mobility SAPIEN simulation data

Benchmarks: OKVQA VQAv2 GQA VizWiz POPE MME MMBench MM-Vet RoboVQA SAPIEN manipulation benchmark

Metrics: accuracy BLEU-1 BLEU-2 BLEU-3 BLEU-4 success rate inference speed

Baselines: UMPNet FlowBot3D RoboFlamingo ManipLLM LLaMA-AdapterV2 TinyLLaVA RWKV-3B

Builds on: Mamba LLaVA ManipLLM RoboFlamingo Cobra

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