中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Parallel Spiking Unit for Efficient Training of Spiking Neural Networks

文献类型:会议论文

作者Yang Li1,3,4; Yinqian Sun1,4; Xiang He1,3,4; Yiting Dong1,4,5; Dongcheng Zhao1,4; Yi Zeng1,2,3,4,5
出版日期2024
会议日期30 June - 5 July 2024
会议地点YOKOHAMA
英文摘要

Efficient parallel computing has become a pivotal element in advancing artificial intelligence. Yet, the deployment of Spiking Neural Networks (SNNs) in this domain is hampered by their inherent sequential computational dependency. This constraint arises from the need for each time step's processing to rely on the preceding step's outcomes, significantly impeding the adaptability of SNN models to massively parallel computing environments. Addressing this challenge, our paper introduces the innovative Parallel Spiking Unit (PSU) and its two derivatives, the Input-aware PSU (IPSU) and Reset-aware PSU (RPSU). These variants skillfully decouple the leaky integration and firing mechanisms in spiking neurons while probabilistically managing the reset process. By preserving the fundamental computational attributes of the spiking neuron model, our approach enables the concurrent computation of all membrane potential instances within the SNN, facilitating parallel spike output generation and substantially enhancing computational efficiency. Comprehensive testing across various datasets, including static and sequential images, Dynamic Vision Sensor (DVS) data, and speech datasets, demonstrates that the PSU and its variants not only significantly boost performance and simulation speed but also augment the energy efficiency of SNNs through enhanced sparsity in neural activity. These advancements underscore the potential of our method in revolutionizing SNN deployment for high-performance parallel computing applications.

语种英语
源URL[http://ir.ia.ac.cn/handle/173211/57075]  
专题类脑智能研究中心_类脑认知计算
通讯作者Yi Zeng
作者单位1.Brain-inspired Cognitive Intelligence Lab, Institute of Automation, Chinese Academy of Sciences
2.Key Laboratory of Brain Cognition and Brain-inspired Intelligence Technology, Chinese Academy of Sciences
3.School of Artificial Intelligence, University of Chinese Academy of Sciences
4.Center for Long-term Artificial Intelligence
5.School of Future Technology, University of Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Yang Li,Yinqian Sun,Xiang He,et al. Parallel Spiking Unit for Efficient Training of Spiking Neural Networks[C]. 见:. YOKOHAMA. 30 June - 5 July 2024.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。