# Mamba2D: A Natively Multi-Dimensional State-Space Model for Vision Tasks

Source: [Mamba2D: A Natively Multi-Dimensional State-Space Model for Vision Tasks](<https://www.sota2.com/research/paper/68fa76961e209131cea6de89>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2026-03-20T05:00:12.244Z

Authors: Enis Baty, Alejandro Hern\\'andez D\\'iaz, Rebecca Davidson, Chris Bridges, Simon Hadfield

Published: 2024-12-20T18:50:36.000Z

## Abstract

State-Space Models (SSMs) have emerged as an efficient alternative to transformers, yet existing visual SSMs retain deeply ingrained biases from their origins in natural language processing. In this paper, we address these limitations by introducing M2D-SSM, a ground-up re-derivation of selective state-space techniques for multidimensional data. Unlike prior works that apply 1D SSMs directly to images through arbitrary rasterised scanning, our M2D-SSM employs a single 2D scan that factors in both spatial dimensions natively. On ImageNet-1K classification, M2D-T achieves 84.0% top-1 accuracy with only 27M parameters, surpassing all prior SSM-based vision models at that size. M2D-S further achieves 85.3%, establishing state-of-the-art results among SSM-based architectures. Across downstream tasks, Mamba2D achieves 52.2 box AP on MS-COCO object detection (3$\\times$ schedule) and 51.7 mIoU on ADE20K segmentation, demonstrating strong generalisation and efficiency at scale. Source code is available at https://github.com/cocoalex00/Mamba2D.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2412.16146>)

[PDF](<https://arxiv.org/pdf/2412.16146.pdf>)

[Implementation](<https://github.com/cocoalex00/Mamba2D>)
