中文

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

机器学习 2024-08-15 v2 人工智能

摘要

联邦机器学习使模型能够在多个客户端之间进行训练,同时保持数据隐私。垂直联邦学习(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.

关键词

引用

@article{arxiv.2407.17672,
  title  = {Spiking Neural Networks in Vertical Federated Learning: Performance Trade-offs},
  author = {Maryam Abbasihafshejani and Anindya Maiti and Murtuza Jadliwala},
  journal= {arXiv preprint arXiv:2407.17672},
  year   = {2024}
}