FTBC:用于优化 ANN-SNN 转换的前向时间偏差校正

💡 原文中文,约1600字,阅读约需4分钟。
📝

内容提要

本文提出了一种新的人工神经网络(ANN)转脉冲神经网络(SNN)框架,结合细粒度正则化和时空反向传播(STBP)训练方法,实现低延迟、低能耗和高准确性。该框架在多个数据集上表现优异,推动了类脑计算的发展。

🎯

关键要点

  • 提出了一种新的人工神经网络(ANN)转脉冲神经网络(SNN)框架,结合细粒度正则化和时空反向传播(STBP)训练方法。
  • 该框架实现了低延迟、低计算能耗和高测试准确性,在ImageNet数据集上仅需4个时间步即可达到73.30%的准确率。
  • 通过结合空间域和时间相关的时域,提出了一种适合于梯度下降训练的迭代LIF模型,提升了多层感知器在多个数据集上的性能。
  • 研究表明,基于偏移尖峰的ANN-SNN转换优化策略能够减少转换误差,并在CIFAR-10、CIFAR-100和ImageNet数据集上取得最先进的性能。
  • 提出的SNN在线训练方法(OTTT)能够在前馈和递归条件下提供与基于脉冲表示的梯度相似的下降方向,避免了BPTT对GPU训练的巨大内存成本。
  • 基于深度脉冲神经网络的计算高效训练方法显著降低了训练复杂度,减少了处理输入所需的时间步骤。
  • 新方法通过门限平衡和软重置机制将权重转移至目标SNN,几乎不出现精度损失,适合嵌入式平台实现。
  • 时域批量归一化技术(BNTT)允许神经元控制脉冲率,实现低延迟和低能耗的训练,首次在复杂数据集上训练深层SNN。
  • 提出的基于阈值相关批归一化和空间时间反向传播的方法在神经形态硬件上实现了高性能的深度SNN训练。
  • 通过ANN-to-SNN转换算法实现对深度SNN的鲁棒性训练,展示了具有低延迟的可扩展高鲁棒性深度SNN的最新解决方案。

延伸问答

FTBC框架的主要优势是什么?

FTBC框架实现了低延迟、低计算能耗和高测试准确性,尤其在ImageNet数据集上仅需4个时间步即可达到73.30%的准确率。

如何优化ANN到SNN的转换过程?

通过基于偏移尖峰的优化策略和初始膜电位的移动来减少转换误差,从而优化ANN到SNN的转换过程。

什么是时空反向传播(STBP)训练方法?

STBP是一种结合空间域和时间相关的时域的训练方法,适合于梯度下降训练,提升多层感知器的性能。

OTTT方法的主要特点是什么?

OTTT是一种SNN在线训练方法,能够在前馈和递归条件下提供与基于脉冲表示的梯度相似的下降方向,避免了BPTT的高内存成本。

如何实现深度脉冲神经网络的高效训练?

通过使用经过转换的SNN和脉冲反向传播技术,逐步执行基于时序的反向传播,从而显著降低训练复杂度。

BNTT技术在SNN训练中的作用是什么?

BNTT技术允许神经元控制脉冲率,实现低延迟和低能耗的训练,首次在复杂数据集上训练深层SNN。

➡️

继续阅读