bamsumit/slayerPytorch

PyTorch implementation of SLAYER for training Spiking Neural Networks

解决的问题

解决了多层脉冲神经网络(SNN)训练的挑战,通过反向传播实现突触权重和轴突延迟参数的学习。

工作原理

这是 SLAYER(Spike Layer Error Reassignment in Time)框架的 PyTorch 实现。系统原生支持每层多个脉冲,并通过层间反向传播误差来优化网络参数。

适用人群

希望使用 PyTorch 和支持 CUDA 的 GPU 实现并训练这些模型的研究人员和开发者。

主要亮点

  • SLAYER 框架的 PyTorch 实现。
  • 支持突触权重和轴突延迟的联合学习。
  • 原生支持每层多个脉冲。
  • 包含多层感知机(MLP)和卷积神经网络(CNN)的示例实现。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目