arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.12507cs.CV

面向低延迟事件驱动视觉识别的混合逐步蒸馏脉冲神经网络

Towards Low-latency Event-based Visual Recognition with Hybrid Step-wise Distillation Spiking Neural Networks

Wuhan University of Technology(武汉理工大学);Nanyang Technological University(南洋理工大学);Hubei University(湖北大学);Peking University(北京大学)

Xian Zhong, Shengwang Hu, Wenxuan Liu, Wenxin Huang, Jianhao Ding, Zhaofei Yu, Tiejun Huang

AI总结:

本文提出混合逐步蒸馏(HSD)方法,通过解耦事件帧与时间步长依赖并引入逐步知识蒸馏,在神经形态数据集上实现低延迟分类,兼顾准确性与推理效率。

摘要:

脉冲神经网络(SNNs)因其低功耗和高生物可解释性而受到广泛关注。其丰富的时空信息处理能力和事件驱动特性使其非常适合神经形态数据集。然而,当前的SNNs在分类这些数据集时难以平衡准确性和延迟。在本文中,我们提出了一种针对神经形态数据集的混合逐步蒸馏(HSD)方法,以缓解在较低时间步长下性能显著下降的问题。我们的工作解耦了事件帧数量与SNNs时间步长之间的依赖关系,在训练阶段利用更多的事件帧来提高性能,而在推理阶段使用更少的事件帧来降低延迟。然而,SNNs在所有时间步长上的平均输出容易受到具有异常输出的单个时间步长的影响,尤其是在极低时间步长下。为了解决这个问题,我们实现了逐步知识蒸馏(SKD)模块,该模块考虑了SNNs在每个时间步长上输出分布的变化。实验证据表明,我们的方法在神经形态数据集的分类任务中取得了有竞争力的性能,尤其是在较低时间步长下。我们的代码将在{https://github.com/hsw0929/HSD}上提供。

英文摘要:

Spiking neural networks (SNNs) have garnered significant attention for their low power consumption and high biological interpretability. Their rich spatio-temporal information processing capability and event-driven nature make them ideally well-suited for neuromorphic datasets. However, current SNNs struggle to balance accuracy and latency in classifying these datasets. In this paper, we propose Hybrid Step-wise Distillation (HSD) method, tailored for neuromorphic datasets, to mitigate the notable decline in performance at lower time steps. Our work disentangles the dependency between the number of event frames and the time steps of SNNs, utilizing more event frames during the training stage to improve performance, while using fewer event frames during the inference stage to reduce latency. Nevertheless, the average output of SNNs across all time steps is susceptible to individual time step with abnormal outputs, particularly at extremely low time steps. To tackle this issue, we implement Step-wise Knowledge Distillation (SKD) module that considers variations in the output distribution of SNNs at each time step. Empirical evidence demonstrates that our method yields competitive performance in classification tasks on neuromorphic datasets, especially at lower time steps. Our code will be available at: {https://github.com/hsw0929/HSD}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑