Federated Learning with a Drone Orchestrator: Path Planning for Minimized Staleness

Research output: Contribution to journalJournal articleResearchpeer-review

2 Citations (Scopus)
21 Downloads (Pure)

Abstract

In this paper, we investigate the problem of scheduling transmissions for spatially scattered nodes that contribute to a collaborative federated learning (FL) algorithm via wireless links provided by a drone. In the considered system, the drone acts as an orchestrator, coordinating the transmissions and the learning schedule within a predefined deadline. The actual schedule is reflected in a planned path: as the drone traverses it, it controls the distance and thereby the data rate to each node. Hence, the model is structured such that the drone orchestrator uses the path (trajectory) as its only tool to achieve fairness in terms of learning staleness, which reflects the learning time discrepancy among the nodes. Using the number of learning epochs performed at each learner as a performance indicator, we combine the average number of epochs computed and staleness into a balanced optimization criterion that is agnostic to the underlying FL implementation. We consider two methods for solving the complex trajectory planning optimization problem for static nodes: (1) successive convex programming (SCP) and (2) deep reinforcement learning (RL). Considering the proposed criterion, both methods are compared in three specific scenarios with few nodes. The results show that drone-orchestrated FL outperforms an immobile deployment by providing improvements in the range of 57% to 87.7%. Additionally, RL-guided trajectories are generally superior to SCP provided ones for complex node arrangements.

Original languageEnglish
Article number9400376
JournalIEEE Open Journal of the Communications Society
Volume2
Pages (from-to)1000-1014
Number of pages15
ISSN2644-125X
DOIs
Publication statusPublished - May 2021

Keywords

  • Drone trajectory optimization
  • convex approximation
  • drone small cells
  • edge computing
  • federated learning
  • reinforcement learning
  • staleness minimization
  • unmanned aerial vehicles
  • wireless communications

Fingerprint

Dive into the research topics of 'Federated Learning with a Drone Orchestrator: Path Planning for Minimized Staleness'. Together they form a unique fingerprint.

Cite this