May, 2023

Joint A-SNN: 通过自蒸馏和权重分解联合训练人工神经网络和脉冲神经网络

TL;DR本文提出了一种人工神经网络和脉冲神经网络之间联合训练的框架,其中人工神经网络可以引导脉冲神经网络的优化。我们通过多个网络分支将人工神经网络中的知识提炼到脉冲神经网络中,并限制了两者的参数,分享部分参数和学习不同的独立权重。在多项广泛的网络结构实验中,我们的方法始终表现出色,超越了许多其他最先进的训练方法。例如,在 CIFAR100 分类任务中,我们的方法训练的脉冲 ResNet-18 模型只需要 4 个时间步,就可以达到 77.39%的 top-1 准确率。