提出稀疏跨层连接SparX,提升Mamba与Transformer视觉模型的特征利用效率。
SparX: A Sparse Cross-Layer Connection Mechanism for Hierarchical Vision Mamba and Transformer Networks
- 模仿视网膜神经节细胞设计稀疏跨层连接,分高密度与低密度层交替布局。
- 参数更少时,SparX-Mamba-T准确率提升至83.5%,优于原模型1个百分点。
- 适合追求高效高精度视觉模型的开发者,尤其适配Mamba类架构。
由于状态空间模型(SSMs)能以线性时间复杂度捕捉长程依赖,Mamba在自然语言处理任务中表现优异,推动了基于Mamba的视觉模型快速发展,并在视觉识别任务中取得良好效果。然而,这类模型难以通过特征聚合、交互与选择实现跨层特征提炼。现有针对CNN或ViT设计的跨层聚合方法在Mamba模型中因计算开销过高而不适用。为此,本文提出一种高效的视觉骨干网络跨层特征聚合机制。受人类视觉系统中视网膜神经节细胞(RGCs)启发,提出名为SparX的稀疏跨层连接机制,有效增强跨层特征交互与复用。具体地,构建两种不同结构的网络层:高连通性且复杂度高的神经节层,支持输入相关性的多层特征聚合与交互;低连通性且复杂度低的普通层。通过交替排列这两类层,设计出新型稀疏交叉连接的视觉骨干网络,在模型规模、计算成本、内存消耗与准确率之间取得良好平衡。例如,参数更少的SparX-Mamba-T将VMamba-T的Top-1准确率从82.5%提升至83.5%,SparX-Swin-T相较Swin-T提升1.3%。大量实验表明,该机制在多种视觉任务中兼具优越性能与泛化能力。
原文摘要 · Abstract (English)
Due to the capability of dynamic state space models (SSMs) in capturing long-range dependencies with linear-time computational complexity, Mamba has shown notable performance in NLP tasks. This has inspired the rapid development of Mamba-based vision models, resulting in promising results in visual recognition tasks. However, such models are not capable of distilling features across layers through feature aggregation, interaction, and selection. Moreover, existing cross-layer feature aggregation methods designed for CNNs or ViTs are not practical in Mamba-based models due to high computational costs. Therefore, this paper aims to introduce an efficient cross-layer feature aggregation mechanism for vision backbone networks. Inspired by the Retinal Ganglion Cells (RGCs) in the human visual system, we propose a new sparse cross-layer connection mechanism termed SparX to effectively improve cross-layer feature interaction and reuse. Specifically, we build two different types of network layers: ganglion layers and normal layers. The former has higher connectivity and complexity, enabling multi-layer feature aggregation and interaction in an input-dependent manner. In contrast, the latter has lower connectivity and complexity. By interleaving these two types of layers, we design a new family of vision backbone networks with sparsely cross-connected layers, achieving an excellent trade-off among model size, computational cost, memory cost, and accuracy in comparison to its counterparts. For instance, with fewer parameters, SparX-Mamba-T improves the top-1 accuracy of VMamba-T from 82.5\% to 83.5\%, while SparX-Swin-T achieves a 1.3\% increase in top-1 accuracy compared to Swin-T. Extensive experimental results demonstrate that our new connection mechanism possesses both superior performance and generalization capabilities on various vision tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。