arXiv:2506.16353cs.CV2025-06中稿 · ICMR2025被引 6

用Mamba架构提升图像哈希效率,实现大规模图像检索

MambaHash: Visual State Space Deep Hashing Model for Large-Scale Image Retrieval

  • 采用分阶段Mamba结构,多方向扫描建模局部与全局特征
  • 在CIFAR-10、NUS-WIDE、ImageNet上均优于现有方法
  • 适合追求高效图像检索的工程与研究应用

深度图像哈希通过深度神经网络将输入图像映射为简洁的二进制哈希码,以实现高效的大型图像检索。近期,具有线性时间复杂度的Vision Mamba在多种计算机任务中表现出色,引起广泛关注。然而,其在大规模图像检索任务中的适用性仍有待探索。为此,我们提出一种视觉状态空间哈希模型MambaHash。具体而言,设计了一种分阶段架构的主干网络,引入分组Mamba操作,利用Mamba沿不同通道组进行多方向扫描,以建模局部与全局信息;随后,提出通道交互注意力模块,增强跨通道信息通信;最后,精心设计自适应特征增强模块,提升特征多样性,强化模型的视觉表示能力。我们在CIFAR-10、NUS-WIDE和ImageNet三个常用数据集上进行了全面实验。结果表明,相较于当前最优的深度哈希方法,MambaHash在效率和性能上均表现优异,能有效完成大规模图像检索任务。源代码已公开:https://github.com/shuaichaochao/MambaHash.git

原文摘要 · Abstract (English)

Deep image hashing aims to enable effective large-scale image retrieval by mapping the input images into simple binary hash codes through deep neural networks. More recently, Vision Mamba with linear time complexity has attracted extensive attention from researchers by achieving outstanding performance on various computer tasks. Nevertheless, the suitability of Mamba for large-scale image retrieval tasks still needs to be explored. Towards this end, we propose a visual state space hashing model, called MambaHash. Concretely, we propose a backbone network with stage-wise architecture, in which grouped Mamba operation is introduced to model local and global information by utilizing Mamba to perform multi-directional scanning along different groups of the channel. Subsequently, the proposed channel interaction attention module is used to enhance information communication across channels. Finally, we meticulously design an adaptive feature enhancement module to increase feature diversity and enhance the visual representation capability of the model. We have conducted comprehensive experiments on three widely used datasets: CIFAR-10, NUS-WIDE and IMAGENET. The experimental results demonstrate that compared with the state-of-the-art deep hashing methods, our proposed MambaHash has well efficiency and superior performance to effectively accomplish large-scale image retrieval tasks. Source code is available https://github.com/shuaichaochao/MambaHash.git

图像哈希Mamba检索深度学习

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