arXiv:2503.24389cs.CVcs.NE2025-03被引 44

用脉冲神经网络实现低功耗水下目标检测,精度高且能耗极低。

SU-YOLO: Spiking Neural Network for Efficient Underwater Object Detection

  • 基于脉冲神经网络设计,用整数加法实现图像去噪,计算开销小。
  • 在URPC2019数据集上达到78.8% mAP,参数仅697万,能耗2.98mJ。
  • 适合资源受限的水下设备,如无人潜航器和海洋监测系统。

水下目标检测对海洋研究和工业安全至关重要。然而,复杂的光学环境与水下设备的有限资源给高精度、低功耗检测带来挑战。为此,我们提出脉冲水下YOLO(SU-YOLO),一种脉冲神经网络(SNN)模型。利用SNN轻量级与低功耗特性,SU-YOLO引入一种仅基于整数加法的新型脉冲式水下图像去噪方法,显著提升特征图质量且计算开销极小。同时,提出分离式批量归一化(SeBN),独立处理多时间步特征图,优化融合残差结构以更好捕捉SNN的时间动态。重新设计的脉冲残差块结合跨阶段部分网络(CSPNet)与YOLO架构,缓解脉冲退化问题,增强特征提取能力。在URPC2019水下数据集上的实验表明,SU-YOLO达到78.8% mAP,参数量为6.97M,能耗仅2.98mJ,优于主流SNN模型,在检测精度与计算效率上均具优势。代码已开源。

原文摘要 · Abstract (English)

Underwater object detection is critical for oceanic research and industrial safety inspections. However, the complex optical environment and the limited resources of underwater equipment pose significant challenges to achieving high accuracy and low power consumption. To address these issues, we propose Spiking Underwater YOLO (SU-YOLO), a Spiking Neural Network (SNN) model. Leveraging the lightweight and energy-efficient properties of SNNs, SU-YOLO incorporates a novel spike-based underwater image denoising method based solely on integer addition, which enhances the quality of feature maps with minimal computational overhead. In addition, we introduce Separated Batch Normalization (SeBN), a technique that normalizes feature maps independently across multiple time steps and is optimized for integration with residual structures to capture the temporal dynamics of SNNs more effectively. The redesigned spiking residual blocks integrate the Cross Stage Partial Network (CSPNet) with the YOLO architecture to mitigate spike degradation and enhance the model's feature extraction capabilities. Experimental results on URPC2019 underwater dataset demonstrate that SU-YOLO achieves mAP of 78.8% with 6.97M parameters and an energy consumption of 2.98 mJ, surpassing mainstream SNN models in both detection accuracy and computational efficiency. These results underscore the potential of SNNs for engineering applications. The code is available in https://github.com/lwxfight/snn-underwater.

水下检测脉冲神经网络低功耗

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