LLM2D
SnnTrans-DHZ:一种用于 underwater image dehazing 的轻量级脉冲神经网络架构
snnTrans-DHZ: A Lightweight Spiking Neural Network Architecture for Underwater Image Dehazing
作者: Vidya Sudevan, Fakhreddine Zayer, Rizwana Kausar, Sajid Javed, Hamad Karki, Giulia De Masi, Jorge Dias
发布日期: 4/17/2025
arXiv ID: oai:arXiv.org:2504.11482v1

摘要

arXiv:2504.11482v1 跨领域公告类型 摘要:基于视觉的海洋作业中,水下图像去雾是非常关键的,因为光线散射和吸收会严重影响能见度。本文介绍了一种名为snnTrans-DHZ的轻量级脉冲神经网络(SNN),该网络专门设计用于水下去雾。通过利用SNN的时间动态特性,snnTrans-DHZ能够高效处理时间依赖的原始图像序列,同时保持低功耗。首先,静态水下图像通过在用户定义的时间步长中重复输入同一张图像,被转换成时间依赖的序列。然后,这些RGB序列被转换到LAB颜色空间表示,并同时进行处理。该架构包含三个关键模块:(i)一个K估计器,用于从多种颜色空间表示中提取特征;(ii)一个背景光估计器,能够从RGB-LAB图像中联合推断背景光成分;(iii)一个软图像重建模块,用于生成无雾、能见度增强的输出。snnTrans-DHZ模型直接使用基于替代梯度的时间反向传播(BPTT)策略和一种新的综合损失函数进行训练。在UIEB基准测试上,snnTrans-DHZ实现了21.68 dB的PSNR和0.8795的SSIM;在EUVP数据集上,其实现了23.46 dB的PSNR和0.8439的SSIM。该模型仅有0.5670百万个网络参数,仅需7.42 GSOPs和0.0151焦耳的能量,其表现明显优于现有最先进的方法,在效率方面具有显著优势。这些特性使得snnTrans-DHZ非常适合部署在水下机器人、海洋探索和环境监测中。