论文标题

通过尖峰神经网络有效的联合学习,以进行交通标志识别

Efficient Federated Learning with Spike Neural Networks for Traffic Sign Recognition

论文作者

Xie, Kan, Zhang, Zhe, Li, Bo, Kang, Jiawen, Niyato, Dusit, Xie, Shengli, Wu, Yi

论文摘要

随着自动驾驶的逐渐普及,对于车辆巧妙地做出正确的驾驶决策并自主遵守交通标志,可以自主遵守交通规则变得越来越重要。但是,对于基于机器学习的交通标志识别(IOV),需要在集中式服务器中收集大量分布式车辆的交通标志数据进行模型培训,这会带来严重的隐私泄漏风险,因为包含大量位置隐私信息的交通符号数据。为了解决这个问题,我们首先利用保护隐私的联合学习,以对准确的识别模型进行协作培训,而无需共享原始的流量标志数据。然而,由于大多数设备的计算和能源有限,车辆很难连续执行复杂的人工智能任务。因此,我们将功能强大的尖峰神经网络(SNN)引入到节能和快速模型训练的交通标志识别中,这是下一代神经网络的实用性,并且非常适合IOV场景。此外,我们设计了一种基于神经元接收场的SNN的新颖编码方案,以从流量标志的像素和空间维度中提取信息,以实现高智能训练。数值结果表明,在准确性,噪声免疫和能源效率方面,提议的联合SNN胜过传统的联邦卷积神经网络。

With the gradual popularization of self-driving, it is becoming increasingly important for vehicles to smartly make the right driving decisions and autonomously obey traffic rules by correctly recognizing traffic signs. However, for machine learning-based traffic sign recognition on the Internet of Vehicles (IoV), a large amount of traffic sign data from distributed vehicles is needed to be gathered in a centralized server for model training, which brings serious privacy leakage risk because of traffic sign data containing lots of location privacy information. To address this issue, we first exploit privacy-preserving federated learning to perform collaborative training for accurate recognition models without sharing raw traffic sign data. Nevertheless, due to the limited computing and energy resources of most devices, it is hard for vehicles to continuously undertake complex artificial intelligence tasks. Therefore, we introduce powerful Spike Neural Networks (SNNs) into traffic sign recognition for energy-efficient and fast model training, which is the next generation of neural networks and is practical and well-fitted to IoV scenarios. Furthermore, we design a novel encoding scheme for SNNs based on neuron receptive fields to extract information from the pixel and spatial dimensions of traffic signs to achieve high-accuracy training. Numerical results indicate that the proposed federated SNN outperforms traditional federated convolutional neural networks in terms of accuracy, noise immunity, and energy efficiency as well.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源