arXiv:2411.15241cs.CV2024-11CVPR被引 72

提出高效视觉马尔可夫模型,用压缩状态混合提升全局建模速度与精度。

EfficientViM: Efficient Vision Mamba with Hidden State Mixer based State Space Duality

  • 基于隐藏状态混合的双态空间结构,压缩计算瓶颈中的通道混合操作。
  • 在ImageNet-1k上达到新最优速度-精度平衡,比次优模型快且高0.7%准确率。
  • 适合资源受限场景部署,尤其在大图输入和蒸馏训练下表现显著提升。

为在资源受限环境部署神经网络,已有工作采用卷积和注意力分别捕捉局部与全局依赖。近期状态空间模型(SSM)因其在序列长度上的线性计算开销,成为高效建模全局交互的新选择。本文提出Efficient Vision Mamba(EfficientViM),基于隐藏状态混合的双态空间结构(HSM-SSD),以更低计算成本高效捕捉全局依赖。观察到原双态空间层运行时间主要由输入序列上的线性投影决定,因此重新设计该层,在压缩的隐藏状态中执行通道混合。同时提出多阶段隐藏状态融合机制,增强状态表示能力,并缓解内存密集型操作带来的瓶颈。结果表明,EfficientViM系列在ImageNet-1k上实现新的速度-精度权衡,相比次优模型SHViT性能提升0.7%,且推理更快。进一步在图像缩放或蒸馏训练下,显著提升吞吐量与准确率。代码已开源:https://github.com/mlvlab/EfficientViM。

原文摘要 · Abstract (English)

For the deployment of neural networks in resource-constrained environments, prior works have built lightweight architectures with convolution and attention for capturing local and global dependencies, respectively. Recently, the state space model (SSM) has emerged as an effective operation for global interaction with its favorable linear computational cost in the number of tokens. To harness the efficacy of SSM, we introduce Efficient Vision Mamba (EfficientViM), a novel architecture built on hidden state mixer-based state space duality (HSM-SSD) that efficiently captures global dependencies with further reduced computational cost. With the observation that the runtime of the SSD layer is driven by the linear projections on the input sequences, we redesign the original SSD layer to perform the channel mixing operation within compressed hidden states in the HSM-SSD layer. Additionally, we propose multi-stage hidden state fusion to reinforce the representation power of hidden states and provide the design to alleviate the bottleneck caused by the memory-bound operations. As a result, the EfficientViM family achieves a new state-of-the-art speed-accuracy trade-off on ImageNet-1k, offering up to a 0.7% performance improvement over the second-best model SHViT with faster speed. Further, we observe significant improvements in throughput and accuracy compared to prior works, when scaling images or employing distillation training. Code is available at https://github.com/mlvlab/EfficientViM.

视觉模型状态空间轻量化高效架构

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