Traffic Sign Recognition Using Optimized Federated Learning in Internet of Vehicles

Lian, Zhuotao and Zeng, Qingkui and Wang, Weizheng and Xu, Dequan and Meng, Weizhi and Su, Chunhua (2024) Traffic Sign Recognition Using Optimized Federated Learning in Internet of Vehicles. IEEE Internet of Things Journal, 11 (4). pp. 6722-6729. ISSN 2327-4662

Abstract

Traffic sign recognition (TSR) is vital for vehicle safety and navigation, especially in the era of autonomous cars. Internet of Vehicles (IoV) provide a promising infrastructure for vehicular networks due to their agility and interoperability. However, privacy concerns and network restrictions hinder the collection of massive data from distributed automotive sensors in IoV. To address these challenges, this article proposes the application of federated learning (FL) and model sparsification to optimize traffic sign recognition (TSR) in autonomous vehicles. FL enables decentralized learning while preserving data privacy, and model sparsification significantly reduces communication costs. Furthermore, we incorporate the Adam optimizer for local training, ensuring efficient model optimization on each vehicle. Experimental results demonstrate the effectiveness of our approach, with improved TSR performance while mitigating privacy risks and enhancing communication efficiency. This research contributes to the advancement of TSR in IoV by introducing FL, model sparsification, and the use of the Adam optimizer for local training, facilitating efficient and privacy-preserving vehicular network learning.