LLM2D
基于脉冲神经网络的节能行人属性识别
SNN-PAR: Energy Efficient Pedestrian Attribute Recognition via Spiking Neural Networks
作者: Haiyang Wang, Qian Zhu, Mowen She, Yabo Li, Haoyu Song, Minghe Xu, Xiao Wang
发布日期: 10/11/2024
arXiv ID: oai:arXiv.org:2410.07857v1

摘要

近年来,基于人工神经网络的行人属性识别 (PAR) 受到广泛研究,尽管取得了许多进展,但其能耗仍然很高。为了解决这个问题,本文提出了一种基于脉冲神经网络 (SNN) 的框架,用于实现节能的属性识别。具体来说,我们首先采用脉冲标记器模块将给定的行人图像转换为脉冲特征表示。然后,将输出输入到脉冲 Transformer 主干网络中,以实现节能的特征提取。我们将增强后的脉冲特征输入到一组前馈网络中,用于行人属性识别。除了广泛使用的二元交叉熵损失函数外,我们还利用从人工神经网络到脉冲 Transformer 网络的知识蒸馏,以实现更准确的属性识别。在三个广泛使用的 PAR 基准数据集上进行的大量实验充分验证了我们提出的 SNN-PAR 框架的有效性。本文的源代码已发布在 \url{https://github.com/Event-AHU/OpenPAR} 上。