摘要
arXiv:2409.12769v1 公告类型: 交叉 摘要: 脉冲神经网络 (SNNs) 近年来在嵌入式设备上的片上学习中引起了极大的兴趣,并作为传统人工神经网络 (ANNs) 的节能替代方案出现。然而,要将 SNNs 扩展到涉及协作模型训练的联邦学习 (FL) 环境中,本地设备与远程服务器之间的通信仍然是瓶颈,这通常受到限制且成本高昂。在本文中,我们首先探讨了 SNNs 在 FL 中噪声通信下的固有鲁棒性。在此基础上,我们提出了一种新的基于 Top-K 稀疏化的联邦学习 (FLTS) 算法,以减少 FL 训练的带宽使用。我们发现,与 ANNs 相比,所提出的 SNNs 方案在不损害模型准确性的情况下允许更多的带宽节省。此外,需要通信的参数数量可以减少到原始模型大小的 6% 以下。我们进一步通过在模型训练期间启用动态参数压缩来提高通信效率。广泛的实验结果表明,我们提出的算法在通信成本和模型准确性方面显著优于基线,并且对于具有 SNNs 的实用网络高效 FL 具有前景。