Saved in:
Bibliographic Details
Main Authors: Tu, Haoyu, Chen, Lin, Li, Zuguang, Chen, Xiaopei, Wu, Wen
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.10451
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916439689003008
author Tu, Haoyu
Chen, Lin
Li, Zuguang
Chen, Xiaopei
Wu, Wen
author_facet Tu, Haoyu
Chen, Lin
Li, Zuguang
Chen, Xiaopei
Wu, Wen
contents In this paper, we study a vehicle selection problem for federated learning (FL) over vehicular networks. Specifically, we design a mobility-aware vehicular federated learning (MAVFL) scheme in which vehicles drive through a road segment to perform FL. Some vehicles may drive out of the segment which leads to unsuccessful training. In the proposed scheme, the real-time successful training participation ratio is utilized to implement vehicle selection. We conduct the convergence analysis to indicate the influence of vehicle mobility on training loss. Furthermore, we propose a multi-armed bandit-based vehicle selection algorithm to minimize the utility function considering training loss and delay. The simulation results show that compared with baselines, the proposed algorithm can achieve better training performance with approximately 28\% faster convergence.
format Preprint
id arxiv_https___arxiv_org_abs_2410_10451
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mobility-Aware Federated Learning: Multi-Armed Bandit Based Selection in Vehicular Network
Tu, Haoyu
Chen, Lin
Li, Zuguang
Chen, Xiaopei
Wu, Wen
Machine Learning
Artificial Intelligence
In this paper, we study a vehicle selection problem for federated learning (FL) over vehicular networks. Specifically, we design a mobility-aware vehicular federated learning (MAVFL) scheme in which vehicles drive through a road segment to perform FL. Some vehicles may drive out of the segment which leads to unsuccessful training. In the proposed scheme, the real-time successful training participation ratio is utilized to implement vehicle selection. We conduct the convergence analysis to indicate the influence of vehicle mobility on training loss. Furthermore, we propose a multi-armed bandit-based vehicle selection algorithm to minimize the utility function considering training loss and delay. The simulation results show that compared with baselines, the proposed algorithm can achieve better training performance with approximately 28\% faster convergence.
title Mobility-Aware Federated Learning: Multi-Armed Bandit Based Selection in Vehicular Network
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.10451