English

V"Mean"ba: Visual State Space Models only need 1 hidden dimension

Computer Vision and Pattern Recognition 2024-12-24 v1 Artificial Intelligence

Abstract

Vision transformers dominate image processing tasks due to their superior performance. However, the quadratic complexity of self-attention limits the scalability of these systems and their deployment on resource-constrained devices. State Space Models (SSMs) have emerged as a solution by introducing a linear recurrence mechanism, which reduces the complexity of sequence modeling from quadratic to linear. Recently, SSMs have been extended to high-resolution vision tasks. Nonetheless, the linear recurrence mechanism struggles to fully utilize matrix multiplication units on modern hardware, resulting in a computational bottleneck. We address this issue by introducing \textit{VMeanba}, a training-free compression method that eliminates the channel dimension in SSMs using mean operations. Our key observation is that the output activations of SSM blocks exhibit low variances across channels. Our \textit{VMeanba} leverages this property to optimize computation by averaging activation maps across the channel to reduce the computational overhead without compromising accuracy. Evaluations on image classification and semantic segmentation tasks demonstrate that \textit{VMeanba} achieves up to a 1.12x speedup with less than a 3\% accuracy loss. When combined with 40\% unstructured pruning, the accuracy drop remains under 3\%.

Keywords

Cite

@article{arxiv.2412.16602,
  title  = {V"Mean"ba: Visual State Space Models only need 1 hidden dimension},
  author = {Tien-Yu Chi and Hung-Yueh Chiang and Chi-Chih Chang and Ning-Chi Huang and Kai-Chiang Wu},
  journal= {arXiv preprint arXiv:2412.16602},
  year   = {2024}
}

Comments

Accepted by NeurIPS 2024 Machine Learning for Systems workshop

R2 v1 2026-06-28T20:44:54.886Z