arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.17672cs.LGcs.AI

纵向联邦学习中的脉冲神经网络:性能权衡

Spiking Neural Networks in Vertical Federated Learning: Performance Trade-offs

University of Texas at San Antonio(德克萨斯大学圣安东尼奥分校);University of Oklahoma(俄克拉荷马大学)

Maryam Abbasihafshejani, Anindya Maiti, Murtuza Jadliwala

AI总结:

本文首次研究脉冲神经网络(SNNs)在纵向联邦学习(VFL)中的适用性,实现带/不带模型拆分的两类联邦架构,在CIFAR数据集上验证发现SNN准确率与传统ANN相当且能效更高。

摘要:

联邦机器学习支持在多客户端间开展模型训练,同时保障数据隐私。纵向联邦学习(Vertical Federated Learning,VFL)专门针对客户端持有相同样本的不同特征集的场景。随着联邦学习模型致力于提升效率与适配性,脉冲神经网络(Spiking Neural Networks,SNNs)这类创新神经网络架构正被用于实现边缘端的快速精准处理。SNNs 以其优于人工神经网络(Artificial Neural Networks,ANNs)的效率著称,但迄今为止尚未有研究分析其在 VFL 中的适用性。本文研究了在纵向联邦学习场景下使用 SNN 模型的优势与权衡。我们实现了两种不同的联邦学习架构——带模型拆分与不带模型拆分的架构,二者在隐私性与性能影响上存在差异。我们采用 CIFAR-10 和 CIFAR-100 基准数据集,结合 VGG9 与 ResNET 分类模型的 SNN 实现对该设置进行评估。对比评估表明,在 VFL 应用中,SNN 模型的准确率与传统 ANNs 相当,但能效显著更高。

英文摘要:

Federated machine learning enables model training across multiple clients while maintaining data privacy. Vertical Federated Learning (VFL) specifically deals with instances where the clients have different feature sets of the same samples. As federated learning models aim to improve efficiency and adaptability, innovative neural network architectures like Spiking Neural Networks (SNNs) are being leveraged to enable fast and accurate processing at the edge. SNNs, known for their efficiency over Artificial Neural Networks (ANNs), have not been analyzed for their applicability in VFL, thus far. In this paper, we investigate the benefits and trade-offs of using SNN models in a vertical federated learning setting. We implement two different federated learning architectures -- with model splitting and without model splitting -- that have different privacy and performance implications. We evaluate the setup using CIFAR-10 and CIFAR-100 benchmark datasets along with SNN implementations of VGG9 and ResNET classification models. Comparative evaluations demonstrate that the accuracy of SNN models is comparable to that of traditional ANNs for VFL applications, albeit significantly more energy efficient.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑