Joint Optimization for Volatile Federated Learning in Vehicular Edge Computing: A Deep Reinforcement Learning Approach

Yawei Li, Li Feng*, Muyu Mei, Amjad Ali*, Zubair Shah

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

2 Citations (Scopus)

Abstract

Federated learning (FL) is highly valued for its ability to reduce communication overhead and protectuser privacy. However, implementing FL in vehicular edge computing (VEC) presents various challenges, such as dropout effect, straggler effect and inefficient communication. Moreover, the dynamic and volatile nature of vehicles in VEC-enabled mobile volatile federated edge learning (VFEL) systems exacerbates these challenges. In this paper, we focus on the volatility in mobile VFEL systems, modeling the dropout problem with vehicles' local computation and communication volatility rates. We formulate the objective to optimize system reliability and learning cost, converting it into a Markov decision process considering environmental dynamics. To achieve the optimal vehicle selection and resource allocation scheme, we propose a reliability-aware twin delayed deep deterministic policy gradient (RA-TD3) scheme by combining the twin delayed deep deterministic policy gradient (TD3) algorithm and convex optimization. Our experimental results demonstrate that the proposed RA-TD3 scheme improves the success rate and reduces the learning cost while maintaining higher learning accuracy.

Original languageEnglish
Pages (from-to)14632-14644
Number of pages13
JournalIEEE Transactions on Vehicular Technology
Volume74
Issue number9
DOIs
Publication statusPublished - 2025

Keywords

  • Volatile federated learning
  • deep reinforcement learning
  • learning cost
  • resource allocation
  • system reliability
  • vehicular edge computing (VEC)

Fingerprint

Dive into the research topics of 'Joint Optimization for Volatile Federated Learning in Vehicular Edge Computing: A Deep Reinforcement Learning Approach'. Together they form a unique fingerprint.

Cite this