arXiv:2502.12627cs.CV2025-02NeurIPS被引 20

动态自适应扫描让视觉状态空间模型更灵活高效

DAMamba: Vision State Space Model with Dynamic Adaptive Scan

  • 用数据驱动方式自适应调整图像扫描顺序和区域
  • 在图像分类等任务上超越现有SOTA视觉Mamba模型
  • 保持线性计算复杂度,适合大图和高精度场景

状态空间模型(SSMs)在计算机视觉领域备受关注。然而,由于图像数据的特性,将自然语言处理中的SSMs直接迁移到视觉任务,尚未超越最先进的卷积神经网络(CNNs)和视觉Transformer(ViTs)。现有视觉SSMs主要依赖人工设计的扫描方式,将图像块局部或全局展平为序列,这破坏了图像原有的语义空间邻近关系,且缺乏灵活性,难以捕捉复杂图像结构。为此,我们提出动态自适应扫描(DAS),一种数据驱动的方法,可自适应分配扫描顺序与区域,提升建模灵活性,同时保持线性计算复杂度和全局建模能力。基于DAS,我们进一步提出视觉主干网络DAMamba,在图像分类、目标检测、实例分割和语义分割等任务上显著优于当前SOTA视觉Mamba模型,甚至超越部分最新CNN和ViT。代码将开源于https://github.com/ltzovo/DAMamba。

原文摘要 · Abstract (English)

State space models (SSMs) have recently garnered significant attention in computer vision. However, due to the unique characteristics of image data, adapting SSMs from natural language processing to computer vision has not outperformed the state-of-the-art convolutional neural networks (CNNs) and Vision Transformers (ViTs). Existing vision SSMs primarily leverage manually designed scans to flatten image patches into sequences locally or globally. This approach disrupts the original semantic spatial adjacency of the image and lacks flexibility, making it difficult to capture complex image structures. To address this limitation, we propose Dynamic Adaptive Scan (DAS), a data-driven method that adaptively allocates scanning orders and regions. This enables more flexible modeling capabilities while maintaining linear computational complexity and global modeling capacity. Based on DAS, we further propose the vision backbone DAMamba, which significantly outperforms current state-of-the-art vision Mamba models in vision tasks such as image classification, object detection, instance segmentation, and semantic segmentation. Notably, it surpasses some of the latest state-of-the-art CNNs and ViTs. Code will be available at https://github.com/ltzovo/DAMamba.

视觉模型状态空间动态扫描Mamba

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。