Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Multimodal Mamba: Decoder-only Multimodal State Space Model via Quadratic to Linear Distillation

About

Recent Multimodal Large Language Models (MLLMs) have achieved remarkable performance but face deployment challenges due to their quadratic computational complexity, growing Key-Value cache requirements, and reliance on separate vision encoders. We propose mmMamba, a framework for developing linear-complexity native multimodal state space models through progressive distillation from existing MLLMs using moderate academic computational resources. Our approach enables the direct conversion of trained decoder-only MLLMs to linear-complexity architectures without requiring pre-trained RNN-based LLM or vision encoders. We propose an seeding strategy to carve Mamba from trained Transformer and a three-stage distillation recipe, which can effectively transfer the knowledge from Transformer to Mamba while preserving multimodal capabilities. Our method also supports flexible hybrid architectures that combine Transformer and Mamba layers for customizable efficiency-performance trade-offs. Distilled from the Transformer-based decoder-only HoVLE, mmMamba-linear achieves competitive performance against existing linear and quadratic-complexity VLMs, while mmMamba-hybrid further improves performance significantly, approaching HoVLE's capabilities. At 103K tokens, mmMamba-linear demonstrates 20.6$\times$ speedup and 75.8% GPU memory reduction compared to HoVLE, while mmMamba-hybrid achieves 13.5$\times$ speedup and 60.2% memory savings. Code and models are released at https://github.com/hustvl/mmMamba

Bencheng Liao, Hongyuan Tao, Qian Zhang, Tianheng Cheng, Yingyue Li, Haoran Yin, Wenyu Liu, Xinggang Wang• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMathVista
Accuracy74.1
382
Visual Question AnsweringTextVQA (val)
VQA Score55.1
365
Visual Question AnsweringGQA (test-dev)
Accuracy59.3
236
Object Hallucination EvaluationPOPE (test)
Accuracy86.7
107
ReasoningMMLU-Pro
MMLU-Pro Reasoning Score54.4
36
Math ReasoningMathVista
Score75.4
30
General ReasoningMMBench
Accuracy82.9
15
ReasoningMMMU-Pro
Accuracy (Reasoning on MMMU-Pro)53.3
13
Fine-Grained PerceptionOCRBench v2
Score59.5
10
Fine-Grained PerceptionChartQA
Score84.2
10
Showing 10 of 19 rows

Other info

Follow for update