Feb, 2024

脉冲中心网络:基于蒸馏提升的脉冲神经网络于目标检测

TL;DR在 AI 边缘计算、自动驾驶汽车和气候变化的时代,需要节能、小型、嵌入式 AI。脉冲神经网络 (SNNs) 是应对这一挑战的有前途的方法,具有事件驱动的信息流和稀疏激活。我们提出了适用于事件数据的脉冲 CenterNet 用于目标检测。它结合了一个 SNN CenterNet 的适应性和一个基于高效 M2U-Net 的解码器。我们的模型在 Prophesee 公司具有挑战性的 GEN1 汽车检测数据集上的性能明显优于可比较的先前工作,同时能使用不到一半的能量。将非脉冲教师的知识蒸馏到我们的 SNN 中进一步提高了性能。据我们所知,我们的工作是脉冲目标检测领域第一个利用知识蒸馏的方法。