Can Graphs Help Vision SSMs See Better?
On the map: State-space sequence models (S4, Mamba) → Vision Mamba backbones
Abstract (arXiv)
Vision state space models inherit the efficiency and long-range modeling ability of Mamba-style selective scans. However, their performance depends critically on the representation of two-dimensional visual features as one-dimensional token sequences. Existing scan operators range from predefined geometric traversals to dynamic coordinate-based samplers that reroute tokens through predicted offsets and interpolation. While effective, these mechanisms primarily adapt paths or sampling locations, rather than explicitly modeling which local patches should exchange information before global state-space mixing. This motivates a simple question: can graphs help vision state space models see better? We introduce GraphScan, a graph-induced dynamic scanning operator for Vision SSMs. For each token, GraphScan constructs a spatially bounded local graph, learns feature-conditioned affinities with relative positional bias, and produces the output token by one-step message passing over its semantic neighborhood. The resulting tokens are locally grounded before being processed by the selective SSM for global aggregation. GraphScan preserves token count and linear scaling in image size, while replacing coordinate-conditioned interpolation with feature-conditioned semantic routing. Integrated into a hierarchical backbone, GraphScan-Mamba achieves state-of-the-art performance among Vision SSMs across image classification, object detection, instance segmentation, and semantic segmentation, with modest computational overhead. Our analysis further shows that GraphScan induces interpretable displacement fields over the token lattice, providing a semantic and spatially grounded view of dynamic scanning. These results suggest that future Vision SSMs should treat scanning not merely as geometric serialization, but as learned local semantic routing before global state-space modeling.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Vision Mamba backbones, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 66 of its 75 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
66 of 75 references.
Closest ancestors · 11
Same direction, newest first
- 2025Similarity-Aware Selective State-Space Modeling for Semantic Correspondence29 Sep
- 2025VCMamba: Bridging Convolutions with Multi-Directional Mamba for Efficient Visual Representation4 Sep
- 2025DefMamba: Deformable Visual State Space Model8 Apr
- 2024Spatial-Mamba: Effective Visual State Space Models via Structure-aware State Fusion19 Oct
- 2024QuadMamba: Learning Quadtree-based Selective Scan for Visual State Space Model9 Oct
- 2024VSSD: Vision Mamba with Non-Causal State Space Duality26 Jul
- 2024MambaVision: A Hybrid Mamba-Transformer Vision Backbone10 Jul
- 2024PlainMamba: Improving Non-Hierarchical Mamba in Visual Recognition26 Mar
- 2024EfficientVMamba: Atrous Selective Scan for Light Weight Visual Mamba15 Mar
- 2024LocalMamba: Visual State Space Model with Windowed Selective Scan14 Mar
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model17 Jan
Other references · 55
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Mask R-CNNcited by 7,277
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
38 more
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2014Going Deeper with Convolutionscited by 4,241
- 2022A ConvNet for the 2020scited by 3,579
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2017Non-local Neural Networkscited by 2,293
- 2021Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutionscited by 1,962
- 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,446
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,271
- 2024VMamba: Visual State Space Modelcited by 969
- 2020Designing Network Design Spacescited by 960
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
- 2021PVT v2: Improved Baselines with Pyramid Vision Transformercited by 751
- 2023ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoderscited by 712
- 2021CoAtNet: Marrying Convolution and Attention for All Data Sizescited by 569
- 2021Twins: Revisiting the Design of Spatial Attention in Vision Transformerscited by 565
- 2022Vision Transformer Adapter for Dense Predictionscited by 565
- 2021CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped Windowscited by 527
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
- 2022InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutionscited by 482
- 2022MaxViT: Multi-Axis Vision Transformercited by 476
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
- 2021Conditional Positional Encodings for Vision Transformerscited by 406
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 393
- 2021Pay Attention to MLPscited by 368
- 2021ResMLP: Feedforward networks for image classification with data-efficient trainingcited by 339
- 2022Vision Transformer with Deformable Attentioncited by 295
- 2021Focal Self-attention for Local-Global Interactions in Vision Transformerscited by 288
- 2022Neighborhood Attention Transformercited by 239
- 2022Vision GNN: An Image is Worth Graph of Nodescited by 186
- 2023BiFormer: Vision Transformer with Bi-Level Routing Attentioncited by 151
- 2022More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsitycited by 142
- 2023RepViT: Revisiting Mobile CNN From ViT Perspectivecited by 123
- 2026Mamba-3: Improved Sequence Modeling using State Space Principlescited by 89
- 2023InceptionNeXt: When Inception Meets ConvNeXtcited by 86
- 2024Mamba-ND: Selective State Space Modeling for Multi-Dimensional Datacited by 84
50 most cited of 55 All 55 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026OCTOPUS: Enhancing the Spatial-Awareness of Vision SSMs with Multi-Dimensional Scans and Traversal Selection
- 2024Mamba2D: A Natively Multi-Dimensional State-Space Model for Vision Tasks
- 2026MFil-Mamba: Multi-Filter Scanning for Spatial Redundancy-Aware Visual State Space Models
- 2026Partial Ring Scan: Revisiting Scan Order in Vision State Space Models
- 2024GG-SSMs: Graph-Generating State Space Models
- 2025Spectral State Space Model for Rotation-Invariant Visual Representation Learning
- 2024V"Mean"ba: Visual State Space Models only need 1 hidden dimension
- 2025Fast Vision Mamba: Pooling Spatial Dimensions for Accelerated Processing