arXiv:2410.10316cs.CV2024-10被引 2

用频域重构图像序列,让视觉Mamba更懂全局结构

GlobalMamba: Global Image Serialization for Vision Mamba

  • 将图像转为频域,按频率分组重排成有因果关系的序列
  • 在ImageNet-1K上达到85.3%准确率,优于现有方法
  • 适合需要捕捉图像全局语义的任务,如目标检测与分割

视觉Mamba凭借线性复杂度在图像处理中表现优异,其效率源于对图像标记的顺序处理。然而,现有方法多采用基于补丁的图像标记化,并将其展平为一维序列进行因果处理,忽略了图像内在的二维结构相关性,也难以通过局部补丁的顺序处理提取全局信息。本文提出一种全局图像序列化方法,将图像转换为包含2D全局信息的因果标记序列。首先利用离散余弦变换(DCT)将图像从空间域转换到频率域,再按对应频率范围排列像素;随后将同一频率带内的像素组重新映射回空间域,生成一系列图像,再进行标记化。我们构建了基于该全局序列化输入的视觉Mamba模型GlobalMamba,可更好利用图像序列间的因果关系。大量实验验证了其有效性,涵盖ImageNet-1K上的图像分类、COCO上的目标检测以及ADE20K上的语义分割任务。

原文摘要 · Abstract (English)

Vision mambas have demonstrated strong performance with linear complexity to the number of vision tokens. Their efficiency results from processing image tokens sequentially. However, most existing methods employ patch-based image tokenization and then flatten them into 1D sequences for causal processing, which ignore the intrinsic 2D structural correlations of images. It is also difficult to extract global information by sequential processing of local patches. In this paper, we propose a global image serialization method to transform the image into a sequence of causal tokens, which contain global information of the 2D image. We first convert the image from the spatial domain to the frequency domain using Discrete Cosine Transform (DCT) and then arrange the pixels with corresponding frequency ranges. We further transform each set within the same frequency band back to the spatial domain to obtain a series of images before tokenization. We construct a vision mamba model, GlobalMamba, with a causal input format based on the proposed global image serialization, which can better exploit the causal relations among image sequences. Extensive experiments demonstrate the effectiveness of our GlobalMamba, including image classification on ImageNet-1K, object detection on COCO, and semantic segmentation on ADE20K.

视觉Mamba图像序列化频域处理全局建模

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