---
title: 'EfficientVMamba: Multi-Scale Visual SSMs'
url: https://www.emergentmind.com/topics/efficientvmamba
type: topic
---

# EfficientVMamba: Multi-Scale Visual SSMs

EfficientVMamba refers to a class of visual backbone architectures that combine state space models (SSMs) with algorithmic and architectural innovations to maximize computational efficiency, especially for vision tasks where large spatial resolution and global context are required. The term is most precisely associated with the “Multi-Scale Vision Mamba” (MSVMamba) approach, which implements a multi-scale selective scan mechanism to reduce compute and memory footprint relative to earlier SSM-based visual backbones, while achieving state-of-the-art accuracy on standard vision benchmarks [2405.14174]. Recent literature, including parallel lines such as “Fast Vision Mamba” [2502.00594], further extends this efficiency frontier through hardware-friendly and algorithmic modifications. This entry covers the mathematical foundation, architectural design, complexity analysis, implementation details, and benchmark results that define the EfficientVMamba family within the SSM vision modeling paradigm.

## 1. Mathematical Foundation: SSMs and the Visual Mamba Block

EfficientVMamba builds upon the Mamba SSM paradigm. The continuous-time SSM evolves a hidden state $h(t)\in\mathbb{R}^N$ under an input drive $x(t)$,
\[
\frac{dh}{dt} = A h(t) + B x(t), \qquad y(t) = C h(t) + D x(t)
\]
with matrices $A$, $B$, $C$, $D$. Discrete-time evolution with step $\Delta$ yields
\[
h_t = \bar{A} h_{t-1} + \bar{B} x_t, \qquad y_t = C h_t + D x_t,
\]
with $\bar{A} = \exp(\Delta A)$, $\bar{B} \approx \Delta B$. Eliminating the state generates a convolution kernel $K$:
\[
y_{1:L} = x_{1:L} * K, \quad K = [C \bar{B},\ C \bar{A} \bar{B},\dots, C \bar{A}^{L-1} \bar{B}] \in \mathbb{R}^L
\]
The "S6" or "Mamba" block replaces these parameters with data-dependent versions $\bar{A}_i$, $\bar{B}_i$, $C_i$, preserving $O(L)$ computation.

For images, VMamba [2401.10166] generalizes this to 2D by scanning a flattened feature map along four complementary spatial orders, computing one SSM per route, and aggregating outputs.

## 2. Multi-Scale 2D Selective Scanning and Architectural Design

MSVMamba/“EfficientVMamba” [2405.14174] innovates by replacing redundant multi-scan operations in VMamba with an explicit multi-scale design. The core steps are:

- **Feature Branches**: Start from input $Z_1 \in \mathbb{R}^{H\times W\times D}$; construct a downsampled map $Z_2 = \mathrm{DWConv}_{3\times3,\,s}(Z_1)$ with stride $s=2$.
- **Selective Scanning**: Define scan/order transforms $\sigma_k$ ($k=1\ldots4$). Apply the S6 block on $\sigma_1(Z_1)$ (full-res) and shared S6 weights to three scan orders of $Z_2$ (half-res).
- **Output Fusing**: The outputs $Y_i$ are reshaped and interpolated, then summed:
  \[
  Z' = \gamma_1(Y_1) + \mathrm{Interpolate}(\gamma_2(Y_2) + \gamma_3(Y_3) + \gamma_4(Y_4))
  \]
This hierarchy, with just one full-res and three half-res scans, preserves long-range dependency learning while lowering complexity by not scanning all directions at full resolution.

A lightweight Convolutional Feed-Forward Network (ConvFFN) addresses the deficiency of token-only mixing in SSMs, adding cross-channel mixing:
\[
\mathrm{ConvFFN}(X) = W_2\,\phi(\mathrm{DWConv}_{3\times3}(\phi(W_1 X)))
\]
where $W_1$, $W_2$ are $1\times1$ expansions and projections, and $\phi=$GELU.

## 3. Computational Complexity and Efficiency Gains

The complexity analysis crucially differentiates EfficientVMamba from its predecessors:
- **Single-scale VMamba** [2401.10166]: Four full-res scans incur $O(4 LDN)$ cost per block, with $L=H W$ (spatial tokens), $D$ (channels), $N$ (state dim).
- **Multi-scale MSVMamba** [2405.14174]: Only one full-res scan and three $s=2$ downsampled scans; the cost per block becomes:
  \[
  O\Bigl(D N [L + 3L/4]\Bigr) = O(1.75\,L D N)
  \]
  which represents a $2.3\times$ reduction in FLOPs over the original. Kernel memory requirements likewise drop.
- **ConvFFN** overhead is negligible compared to total block cost.

## 4. Implementation Details and Training Regimen

In the reference implementation [2405.14174], model scaling and training are as follows:
- **Model Sizes**: E.g., MSVMamba-Tiny: $33$M params, $4.6$G FLOPs, $224\times224$ images.
- **Regimen**: ImageNet-1K, 300 epochs, AdamW (lr=$1\mathrm{e}{-3}$, $\cos$ decay, $0.05$ weight decay), batch size $1024$, advanced augmentations, label smoothing $0.1$.
- **Inference**: Full-res scan in one order; three half-res scans along remaining orders, center crop for classification.
- **Codebase**: All models and detailed specs at https://github.com/YuHengsss/MSVMamba.

## 5. Empirical Results: Performance Benchmarks

EfficientVMamba achieves highly competitive or superior accuracy to all Vision Mamba variants under comparable or reduced computational cost:

| Model                | Params | FLOPs | ImageNet Top-1 |
|----------------------|--------|-------|----------------|
| EffVMamba-T          |   6M   | 0.8G  | 76.5%          |
| EffVMamba-S          |  11M   | 1.3G  | 78.7%          |
| EffVMamba-B          |  33M   | 4.0G  | 81.8%          |
| MSVMamba-Tiny        |  33M   | 4.6G  | 82.8%          |

On COCO Mask R-CNN (1×, $1280\times800$):
- EffVMamba-S: 39.3/36.7 (box/mask mAP, 31M params, 197G FLOPs)
- MSVMamba-Tiny: 46.9/42.2 (53M params, 252G FLOPs)

On ADE20K UPerNet (semantic segmentation):
- EffVMamba-S: 41.5mIoU/42.1mIoU (29M, 505G)
- MSVMamba-Tiny: 47.6/48.5 (65M, 942G)

MSVMamba-Tiny improves ImageNet top-1 by 0.6% versus VMamba-T for 17% fewer FLOPs, and similarly outperforms on COCO and ADE20K.

## 6. Extensions, Limitations, and Impact

EfficientVMamba demonstrates that multi-scale SSMs can supplant expensive multi-directional full-resolution scans without degrading receptive field or accuracy. The ConvFFN addition mitigates the channel mixing bottleneck present in pure SSM backbones. Remaining constraints involve further reducing constant factors in memory and FLOPs, optimizing hardware utilization, and adapting to dense prediction regimes requiring fine spatial detail.

The MSVMamba code and structural choices directly inform a series of follow-on work, including Fast Vision Mamba [2502.00594], which applies spatial pooling and parallel scan acceleration; MambaScope [2512.00647], which introduces coarse-to-fine adaptive tokenization; and various hybrid CNN/SSM architectures for device-critical deployment.

EfficientVMamba sets a new Pareto frontier for throughput-accuracy tradeoffs in visual SSMs, provides the substrate for real-time and large-scale applications, and serves as a crucial reference for the design of next-generation Mamba-like vision models. Its linear-time global modeling paradigm and compositional multi-scale design are now foundational in visual SSM efficiency research [2405.14174].

Source: https://www.emergentmind.com/topics/efficientvmamba