摘要
arXiv:2503.20485v1 Announce Type: cross
摘要:水下图像增强(UIE)对于海洋应用,包括自主基于视觉的导航至关重要。使用卷积神经网络(CNN)和视觉变换器的深度学习方法提高了UIE的性能。最近,脉冲神经网络(SNN)由于其轻量级设计、能源效率和可扩展性而引起了关注。本文介绍了UIE-SNN,这是第一个基于SNN的UIE算法,用于提高水下图像的可见度。UIE-SNN是一个带有跳连接的19层卷积脉冲编码-解码框架,直接使用潜梯度替代的反向时间传播(BPTT)策略进行训练。我们探讨并验证了训练数据集对能量降低的影响,这是UIE-SNN架构的一个独特优势,与传统的基于学习的架构相比,能量消耗取决于模型。UIE-SNN在潜在空间表示中优化损失函数以重建清晰的水下图像。我们的算法在减少的时间步长($T=5$)和能量消耗($85\%$)下,在PSNR和结构相似性指数(SSIM)方面与非脉冲版本的算法表现相当。该算法在UIEB和EUVP两个公开的基准数据集上进行训练,并在UIEB、EUVP、LSUI、U45和我们自定义的UIE数据集上对未见过的图像进行测试。UIE-SNN算法在UIEB上实现了PSNR为\(17.7801~dB\)和SSIM为\(0.7454\),在EUVP上实现了PSNR为\(23.1725~dB\)和SSIM为\(0.7890\)。与非脉冲版本的算法相比,UIE-SNN以更少的操作数(\(147.49\) GSOPs)和更少的能量(\(0.1327~J\))实现了这一算法性能。与现有最先进的UIE方法相比,UIE-SNN在能量效率方面平均提高了\(6.5\times\)。源代码可在\href{https://github.com/vidya-rejul/UIE-SNN.git}{UIE-SNN}获取。