计算机应用 ›› 2021, Vol. 41 ›› Issue (6): 1546-1550.DOI: 10.11772/j.issn.1001-9081.2020121912

所属专题: 2020年全国开放式分布与并行计算学术年会(DPCS 2020)

• 2020年全国开放式分布与并行计算学术年会(DPCS 2020) • 上一篇    下一篇

高速车联网场景下分簇式无线联邦学习算法

王家瑞1,2, 谭国平1,2, 周思源1,2   

  1. 1. 河海大学 计算机与信息学院 南京 211100;
    2. 江苏智能交通及智能驾驶研究院 南京 210019
  • 收稿日期:2020-11-04 修回日期:2021-03-31 出版日期:2021-06-10 发布日期:2021-06-23
  • 通讯作者: 谭国平
  • 作者简介:王家瑞(1998-),男,山东威海人,硕士研究生,主要研究方向:无线网络;谭国平(1975-),男,湖南澧县人,教授,博士,CCF会员,主要研究方向:无线通信系统与网络;周思源(1985-),男,江苏南京人,副教授,博士,CCF会员,主要研究方向:无线通信。
  • 基金资助:
    国家自然科学基金资助项目(61701168,61832005,61571303);中国博士后科学基金资助项目(2019M651546);江苏省交通技术改造项目(2018Y45)。

Clustered wireless federated learning algorithm in high-speed internet of vehicles scenes

WANG Jiarui1,2, TAN Guoping1,2, ZHOU Siyuan1,2   

  1. 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
  • Received:2020-11-04 Revised:2021-03-31 Online:2021-06-10 Published:2021-06-23
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (61701168, 61832005, 61571303), the China Postdoctoral Science Foundation (2019M651546), the Jiangsu Province Transportation Technology Transformation Project (2018Y45).

摘要: 现有无线联邦学习框架缺乏对实际的分布式高速车联网(IoV)场景的有效支持。针对该场景下的分布式学习问题,提出了一种基于随机网络拓扑模型的分布式训练算法——分簇式无线联邦学习算法(C-WFLA)。首先,该算法基于高速公路场景下的车辆分布情况设计网络模型;其次,该算法考虑了用户端进行上行数据传输时的路径衰落、瑞利衰落等因素;最后,该算法设计了基于分簇式训练的无线联邦学习方法。利用所提算法对手写体识别模型进行了训练与测试,仿真结果表明:在信道状态较好、用户发射功率受限较小的情况下,传统无线联邦学习算法与C-WFLA在相同的训练条件下损失函数均能收敛至相近的数值,且C-WFLA收敛更快;而在信道状态较差、用户发射功率受限较大的情况下,C-WFLA损失函数收敛值相较于传统的集中式算法可以降低10%~50%。可见,C-WFLA更有助于高速IoV场景下的模型训练。

关键词: 无线联邦学习, 随机拓扑, 车联网, 分布式学习, 分簇算法

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.

Key words: wireless federated learning, random topology, Internet of Vehicles (IoV), distributed learning, clustering algorithm

中图分类号: