首页 | 本学科首页   官方微博 | 高级检索  
     

高速车联网场景下分簇式无线联邦学习算法
引用本文:王家瑞,谭国平,周思源.高速车联网场景下分簇式无线联邦学习算法[J].计算机应用,2021,41(6):1546-1550.
作者姓名:王家瑞  谭国平  周思源
作者单位:1. 河海大学 计算机与信息学院 南京 211100;2. 江苏智能交通及智能驾驶研究院 南京 210019
基金项目:国家自然科学基金资助项目(61701168,61832005,61571303);中国博士后科学基金资助项目(2019M651546);江苏省交通技术改造项目(2018Y45)。
摘    要:现有无线联邦学习框架缺乏对实际的分布式高速车联网(IoV)场景的有效支持。针对该场景下的分布式学习问题,提出了一种基于随机网络拓扑模型的分布式训练算法——分簇式无线联邦学习算法(C-WFLA)。首先,该算法基于高速公路场景下的车辆分布情况设计网络模型;其次,该算法考虑了用户端进行上行数据传输时的路径衰落、瑞利衰落等因素;最后,该算法设计了基于分簇式训练的无线联邦学习方法。利用所提算法对手写体识别模型进行了训练与测试,仿真结果表明:在信道状态较好、用户发射功率受限较小的情况下,传统无线联邦学习算法与C-WFLA在相同的训练条件下损失函数均能收敛至相近的数值,且C-WFLA收敛更快;而在信道状态较差、用户发射功率受限较大的情况下,C-WFLA损失函数收敛值相较于传统的集中式算法可以降低10%~50%。可见,C-WFLA更有助于高速IoV场景下的模型训练。

关 键 词:无线联邦学习  随机拓扑  车联网  分布式学习  分簇算法  
收稿时间:2020-11-04
修稿时间:2021-03-31

Clustered wireless federated learning algorithm in high-speed internet of vehicles scenes
WANG Jiarui,TAN Guoping,ZHOU Siyuan.Clustered wireless federated learning algorithm in high-speed internet of vehicles scenes[J].journal of Computer Applications,2021,41(6):1546-1550.
Authors:WANG Jiarui  TAN Guoping  ZHOU Siyuan
Affiliation:1. School of Computer and Information, Hohai University, Nanjing Jiangsu 211100, China;2. Jiangsu Intelligent Transportation and Intelligent Driving Research Institute, Nanjing Jiangsu 210019, China
Abstract:Existing wireless federated learning frameworks lack the effective support for the actual distributed high-speed Internet of Vehicles (IoV) scenes. Aiming at the distributed learning problem in such scenes, a distributed training algorithm based on the random network topology model named Clustered-Wireless Federated Learning Algorithm (C-WFLA) was proposed. In this algorithm, firstly, a network model was designed on the basis of the distribution situation of vehicles in the highway scene. Secondly, the path fading, Rayleigh fading and other factors during the uplink data transmission of the users were considered. Finally, a wireless federated learning method based on clustered training was designed. The proposed algorithm was used to train and test the handwriting recognition model. The simulation results show that under the situations of good channel state and little user transmit power limit, the loss functions of traditional wireless federated learning algorithm and C-WFLA can converge to similar values under the same training condition, but C-WFLA converges faster; under the situations of poor channel state and much user transmit power limit, C-WFLA can reduce the convergence value of loss function by 10% to 50% compared with the traditional centralized algorithm. It can be seen that C-WFLA is more helpful for model training in high-speed IoV scenes.
Keywords:wireless federated learning  random topology  Internet of Vehicles (IoV)  distributed learning  clustering algorithm  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机应用》浏览原始摘要信息
点击此处可从《计算机应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号