May, 2024

灵活高效的代理梯度建模与前向梯度注入

TL;DR本文介绍了一种替代方法,即前向梯度注入(FGI),用于在前向传递过程中将任意梯度形状直接注入计算图,证明了使用 FGI 相对于自定义向后传递方法在 SNNs 中使用 TorchScript 可以显著提高模型性能,并通过在 TorchScript 和 torch.compile 中对循环 SNNs 进行的一般性能研究结果补充,揭示了与纯 PyTorch 相比,训练速度可提高 7 倍以上,推理速度可提高 16 倍以上的潜力。