Papers › Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion

Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion

19 Oct 2024arXiv:2410.15091archive 2025-07-28

Chaodong Xiao, Minghan Li, Zhengqiang Zhang, Deyu Meng, Lei Zhang

Selective state space models (SSMs), such as Mamba, highly excel at capturing long-range dependencies in 1D sequential data, while their applications to 2D vision tasks still face challenges. Current visual SSMs often convert images into 1D sequences and employ various scanning patterns to incorporate local spatial dependencies. However, these methods are limited in effectively capturing the complex image spatial structures and the increased computational cost caused by the lengthened scanning paths. To address these limitations, we propose Spatial-Mamba, a novel approach that establishes neighborhood connectivity directly in the state space. Instead of relying solely on sequential state transitions, we introduce a structure-aware state fusion equation, which leverages dilated convolutions to capture image spatial structural dependencies, significantly enhancing the flow of visual contextual information. Spatial-Mamba proceeds in three stages: initial state computation in a unidirectional scan, spatial context acquisition through structure-aware state fusion, and final state computation using the observation equation. Our theoretical analysis shows that Spatial-Mamba unifies the original Mamba and linear attention under the same matrix multiplication framework, providing a deeper understanding of our method. Experimental results demonstrate that Spatial-Mamba, even with a single scan, attains or surpasses the state-of-the-art SSM-based models in image classification, detection and segmentation. Source codes and trained models can be found at $\href{https://github.com/EdwardChasel/Spatial-Mamba}{\text{this https URL}}$.

PaperPDFCodeCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2410.15091")

Code

Syntology Ran 2 of 14 code samples harvested from 1 repository linked to this paper; 12 have no recorded run. Of those that ran: 1 ran · honoured contract; 1 ran · our draft was wrong.

By repository: official repository: 14 samples from 1 repository, 2 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

edwardchasel/spatial-mamba officialmentioned in papermentioned on GitHubpytorchApache-2.0 report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

14 samples harvested; 2 ran; 1 honoured the contract we drafted; 12 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

1ran · honoured contract
1ran · our draft was wrong
12unverified

Licence: 0 of the 14 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from edwardchasel/spatial-mamba. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

get_grad_norm edwardchasel/spatial-mamba/classification/utils/utils.py official repository ran · honoured contract Apache-2.0 (permissive) · eec1e7cba51d5e8e · report
import_abspy EdwardChasel/Spatial-Mamba/detection/model.py official repository ran · our draft was wrong Apache-2.0 (permissive) · 5315806d8b42f938 · report
auto_resume_helper edwardchasel/spatial-mamba/classification/utils/utils.py official repository unverified Apache-2.0 (permissive) · 1f9f75e583e2169e · report
build_optimizer edwardchasel/spatial-mamba/classification/utils/optimizer.py official repository unverified Apache-2.0 (permissive) · 00fc7f0336e999a0 · report
build_scheduler edwardchasel/spatial-mamba/classification/utils/lr_scheduler.py official repository unverified Apache-2.0 (permissive) · 0fbdf3257aea9bcc · report
check_keywords_in_name edwardchasel/spatial-mamba/classification/utils/optimizer.py official repository unverified Apache-2.0 (permissive) · b0a5beb34716d5a6 · report
create_logger edwardchasel/spatial-mamba/classification/utils/logger.py official repository unverified Apache-2.0 (permissive) · 80450600f7f09b0a · report
flops_selective_scan_fn edwardchasel/spatial-mamba/classification/models/utils.py official repository unverified Apache-2.0 (permissive) · d1a00af9c7447534 · report
get_config edwardchasel/spatial-mamba/classification/config.py official repository unverified Apache-2.0 (permissive) · 3760f0a84c81308d · report
load_checkpoint_ema edwardchasel/spatial-mamba/classification/utils/utils.py official repository unverified Apache-2.0 (permissive) · c3565ec7e9c7f4df · report
selective_scan_flop_jit edwardchasel/spatial-mamba/classification/models/utils.py official repository unverified Apache-2.0 (permissive) · 994608c8957905d6 · report
selective_scan_state_flop_jit edwardchasel/spatial-mamba/classification/models/utils.py official repository unverified Apache-2.0 (permissive) · 2cbc0f8f150c8aae · report
set_weight_decay edwardchasel/spatial-mamba/classification/utils/optimizer.py official repository unverified Apache-2.0 (permissive) · 2d5473f673967c35 · report
update_config edwardchasel/spatial-mamba/classification/config.py official repository unverified Apache-2.0 (permissive) · 13920ae73055d5cf · report

Tasks

Image ClassificationMambaState Space Modelsimage-classification

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

AttentionMambaSoftmax

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections