Parallel Spiking Unit for Efficient Training of Spiking Neural Networks
文献类型:会议论文
作者 | Yang Li1,3,4![]() ![]() ![]() ![]() |
出版日期 | 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收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。