Three-Stage Stackelberg Game Enabled Clustered Federated Learning in Heterogeneous UAV Swarms | IEEE Journals & Magazine | IEEE Xplore

Three-Stage Stackelberg Game Enabled Clustered Federated Learning in Heterogeneous UAV Swarms


Abstract:

In the past decade, the unmanned aerial vehicles (UAVs) swarm has become a disruptive force reshaping our lives and work. In particular, advances in artificial intelligen...Show More

Abstract:

In the past decade, the unmanned aerial vehicles (UAVs) swarm has become a disruptive force reshaping our lives and work. In particular, advances in artificial intelligence have allowed multiple UAVs to coordinate their operations and work together to accomplish various complex tasks, one of which is Federated Learning (FL). As a promising distributed learning paradigm, FL can be adopted well with the limited resources and dynamic network topology of UAV swarms. However, the current FL's training process relies on homogeneous data paradigms, which require distributed UAVs to hold the same structure data. This ideal hypothesis can not apply to the heterogeneous UAV swarms. To tackle this problem, in this paper, we design a clustered federated learning (CFL) architecture, in which we cluster UAV swarms based on the similarities between the participants' optimization directions. Then, we formulate the model trading among model owners, cluster heads, and UAV workers as a three-stage Stackelberg game to optimize the allocation of the limited resources. We design a hierarchical reinforcement learning algorithm to search for the Stackelberg equilibrium under the clustered federated learning system. The performance evaluation demonstrates the uniqueness and stability of the proposed three-stage leader-follower game under the clustered framework, as well as the convergence and effectiveness of the reinforcement learning algorithm.
Published in: IEEE Transactions on Vehicular Technology ( Volume: 72, Issue: 7, July 2023)
Page(s): 9366 - 9380
Date of Publication: 23 February 2023

ISSN Information:

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.