arXiv:2410.07599cs.CV2024-10CVPR被引 10

用序列化图像+因果建模,实现高分辨率图像的高效视觉识别

Adventurer: Optimizing Vision Mamba Architecture Designs for Efficiency

  • 将图像转为补丁序列,用单向语言模型实现线性复杂度计算
  • Adventurer-Base在ImageNet-1k上达84.3%准确率,训练速度是Vim的6.2倍
  • 适合追求高分辨率图像处理效率的研究者和应用开发者

本文提出Adventurer系列模型,将图像视为补丁序列,采用单向语言模型学习视觉表征。该建模范式以线性复杂度处理图像,有效缓解高分辨率、细粒度图像带来的内存与计算爆炸问题。我们设计了两种简单结构:在序列首加入全局池化标记,以及层间交替翻转操作,无缝融入因果推理框架。大量实验表明,相比DeiT和Vim等现有架构,Adventurer在效率与精度间取得更优平衡。例如,Adventurer-Base在标准ImageNet-1k基准上达到84.3%测试准确率,训练吞吐量达216张/秒,比Vim快3.8倍,比DeiT快6.2倍。由于其计算与内存效率高且可线性扩展,该架构有望推动高分辨率或细粒度图像长序列建模的未来发展。代码已开源。

原文摘要 · Abstract (English)

In this work, we introduce the Adventurer series models where we treat images as sequences of patch tokens and employ uni-directional language models to learn visual representations. This modeling paradigm allows us to process images in a recurrent formulation with linear complexity relative to the sequence length, which can effectively address the memory and computation explosion issues posed by high-resolution and fine-grained images. In detail, we introduce two simple designs that seamlessly integrate image inputs into the causal inference framework: a global pooling token placed at the beginning of the sequence and a flipping operation between every two layers. Extensive empirical studies highlight that compared with the existing plain architectures such as DeiT and Vim, Adventurer offers an optimal efficiency-accuracy trade-off. For example, our Adventurer-Base attains a competitive test accuracy of 84.3% on the standard ImageNet-1k benchmark with 216 images/s training throughput, which is 3.8 and 6.2 times faster than Vim and DeiT to achieve the same result. As Adventurer offers great computation and memory efficiency and allows scaling with linear complexity, we hope this architecture can benefit future explorations in modeling long sequences for high-resolution or fine-grained images. Code is available at https://github.com/wangf3014/Adventurer.

视觉模型序列建模高效计算图像识别

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