摘要
脉冲神经网络 (SNN) 作为一种超低功耗计算范式受到了广泛关注。最近的研究集中于提高 SNN 的特征提取能力,但它们存在推理效率低和性能次优的问题。本文提出了一种简单而有效的时空反转训练 (TRT) 方法来优化 SNN 的时空性能并规避这些问题。我们通过时间反转来扰动输入时间数据,促使 SNN 产生原始-反转一致的输出,并学习扰动不变的表示。对于没有时间维度的静态数据,我们通过利用 SNN 的固有时间特性进行脉冲特征时间反转来推广这一策略。此外,我们利用轻量级的“星型运算”(逐元素乘法)来混合原始和时间反转的脉冲放电率并扩展隐式维度,这作为时空正则化来进一步增强 SNN 的泛化能力。我们的方法仅在训练过程中涉及时间反转运算和逐元素乘法,因此训练开销可忽略不计,并且完全不会影响推理效率。在静态/神经形态目标/动作识别和 3D 点云分类任务上的大量实验结果证明了该方法的有效性和泛化能力。特别是,仅使用两个时间步长,我们的方法在 ImageNet 和 ModelNet40 上分别达到了 74.77% 和 90.57% 的准确率。