摘要
arXiv:2407.20708v4 宣布类型: 替换
摘要: 受脑启发的突触神经网络(SNNs)在生物可行性和低功耗方面优于人工神经网络(ANNs)。目前,SNNs的应用仅限于简单的分类任务,这是由于其性能较差。在这项工作中,我们专注于在对象检测方面弥合ANNs和SNNs之间的性能差距。我们的设计围绕网络架构和突触神经元展开。首先,过于复杂的模块设计导致将YOLO系列转换为其对应的突触版本时突触退化。我们设计了一个SpikeYOLO架构来解决这个问题,通过简化现有的YOLO并结合元SNN块来简化其结构。其次,对象检测对突触神经元在膜电位转换为二元突触时的量化误差更为敏感。为了解决这一挑战,我们设计了一种新的突触神经元,在训练期间激活整数值,同时在推理期间通过扩展虚拟时间步长来保持由突触驱动的特性。所提出的方法在静态和神经形态对象检测数据集上进行了验证。在静态COCO数据集上,我们获得了66.2%的mAP@50和48.9%的mAP@50:95,分别比之前的最先进SNN高出15.0%和18.7%。在神经形态Gen1数据集上,我们达到了67.2%的mAP@50,比具有同等架构的人工神经网络高2.5%,同时能量效率提高了5.7*。代码: https://github.com/BICLab/SpikeYOLO