arXiv:2501.06786cs.CV2025-01被引 4

用脉冲网络和三维小波分解实现低功耗高效视觉数据检索

Temporal-Aware Spiking Transformer Hashing Based on 3D-DWT

  • 基于3D-DWT分解时空特征,分层融合频域信息
  • 脉冲自注意力捕获全局时空依赖,参数少能耗低
  • 动态软相似性损失提升分类差异表达,适合嵌入式部署

随着动态视觉传感器(DVS)数据的快速增长,构建低功耗、高效的检索系统成为迫切需求。哈希学习是重要技术之一,可保持哈希码与DVS数据间距离一致。脉冲神经网络(SNN)通过脉冲编码信息,具备显著能效优势。本文提出一种新型有监督哈希方法Spikinghash,采用分层轻量结构:浅层使用脉冲波形混合器(SWM),通过多级3D离散小波变换(3D-DWT)将时空特征分解为不同频段成分,并实现高效频域特征融合,有效捕捉时序依赖与局部空间特征;深层采用脉冲自注意力(SSA)进一步提取全局时空信息。设计基于SNN二值特性的哈希层,整合多时间步信息生成最终哈希码。此外,提出一种新的动态软相似性损失,利用膜电位构建可学习的相似性矩阵作为软标签,充分捕捉类别间差异,补偿SNN中信息损失,从而提升检索性能。在多个数据集上的实验表明,Spikinghash在保持低能耗与少参数的同时达到当前最优性能。

原文摘要 · Abstract (English)

With the rapid growth of dynamic vision sensor (DVS) data, constructing a low-energy, efficient data retrieval system has become an urgent task. Hash learning is one of the most important retrieval technologies which can keep the distance between hash codes consistent with the distance between DVS data. As spiking neural networks (SNNs) can encode information through spikes, they demonstrate great potential in promoting energy efficiency. Based on the binary characteristics of SNNs, we first propose a novel supervised hashing method named Spikinghash with a hierarchical lightweight structure. Spiking WaveMixer (SWM) is deployed in shallow layers, utilizing a multilevel 3D discrete wavelet transform (3D-DWT) to decouple spatiotemporal features into various low-frequency and high frequency components, and then employing efficient spectral feature fusion. SWM can effectively capture the temporal dependencies and local spatial features. Spiking Self-Attention (SSA) is deployed in deeper layers to further extract global spatiotemporal information. We also design a hash layer utilizing binary characteristic of SNNs, which integrates information over multiple time steps to generate final hash codes. Furthermore, we propose a new dynamic soft similarity loss for SNNs, which utilizes membrane potentials to construct a learnable similarity matrix as soft labels to fully capture the similarity differences between classes and compensate information loss in SNNs, thereby improving retrieval performance. Experiments on multiple datasets demonstrate that Spikinghash can achieve state-of-the-art results with low energy consumption and fewer parameters.

脉冲神经网络哈希检索低功耗时空建模

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