arrow
返回

Deep multi-threshold spiking-UNet for image processing

delete2024-06-01
delete0
delete
OA
AI
H
Hebei Li
Y
Yueyi Zhang *
熊
熊志伟 (Zhiwei Xiong)
X
Xiaoyan Sun
DOI:10.1016/j.neucom.2024.127653delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
U -Net, known for its simple yet efficient architecture, is widely utilized for image processing tasks and is particularly suitable for deployment on neuromorphic chips. This paper introduces the novel concept of Spiking-UNet for image processing, which combines the power of Spiking Neural Networks (SNNs) with the U -Net architecture. To achieve an efficient Spiking-UNet, we face two primary challenges: ensuring highfidelity information propagation through the network via spikes and formulating an effective training strategy. To address the issue of information loss, we introduce multi -threshold spiking neurons, which improve the efficiency of information transmission within the Spiking-UNet. For the training strategy, we adopt a conversion and fine-tuning pipeline that leverage pre -trained U -Net models. During the conversion process, significant variability in data distribution across different parts is observed when utilizing skip connections. Therefore, we propose a connection -wise normalization method to prevent inaccurate firing rates. Furthermore, we adopt a flow -based training method to fine-tune the converted models, reducing time steps while preserving performance. Experimental results show that, on image segmentation and denoising, our Spiking-UNet achieves comparable performance to its non -spiking counterpart, surpassing existing SNN methods. Compared with the converted Spiking-UNet without fine-tuning, our Spiking-UNet reduces inference time by approximately 90%. This research broadens the application scope of SNNs in image processing and is expected to inspire further exploration in the field of neuromorphic engineering. The code for our Spiking-UNet implementation is available at https://github.com/SNNresearch/Spiking-UNet.
Keyword:
U-Net
Spiking neural network
Multi-threshold spiking neuron
Weight normalization
Image segmentation
Image denoising
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
引用论文

引用论文

Hierarchical U-Shape Attention Network for Salient Object Detection用于显著目标检测的分层U形注意网络
err2020-01-01
err21
PREAI
errZhou, Sanping; Wang, Jinjun; Zhang, Jimuyang; Wang, Le; Huang, Dong; Du, Shaoyi; Zheng, Nanning
err分享
err收藏
Optical flow estimation from event-based cameras and spiking neural networks来自基于事件的相机和尖峰神经网络的光流估计
err2023-05-11
err21
errOAAI
errCuadrado, Javier; Rancon, Ulysse; Cottereau, Benoit R.; Barranco, Francisco; Masquelier, Timothee
err分享
err收藏
StereoSpike: Depth Learning With a Spiking Neural NetworkStereoSpike: 使用尖峰神经网络进行深度学习
err2022-01-01
err19
errOAAI
errRancon, Ulysse; Cuadrado-Anibarro, Javier; Cottereau, Benoit R.; Masquelier, Timothee
err分享
err收藏
err分享
err收藏
Dynamic Image Quantization Using Leaky Integrate-and-Fire Neurons
err2021-01-01
err10
errOAAI
errDoutsi, Effrosyni; Fillatre, Lionel; Antonini, Marc; Tsakalides, Panagiotis
err分享
err收藏
Conversion of Continuous-Valued Deep Networks to Efficient Event-Driven Networks for Image Classification
err2017-12-07
err771
errOAAI
errRueckauer, Bodo; Lungu, Iulia-Alexandra; Hu, Yuhuang; Pfeiffer, Michael; Liu, Shih-Chii
err分享
err收藏
学者 查看更多内容