HFL-GAN: scalable hierarchical federated learning GAN for high quantity heterogeneous clients

Abstract

This paper introduces a novel approach for training generative adversarial networks using federated machine learning. Generative adversarial networks have gained plenty of attention in the research community especially with their abilities to produce high quality synthetic data for a variety of use-cases. Yet, when combined with federated learning, those models suffer from degradation in both training time and quality of results. To address this challenge, this paper introduces a novel approach that uses hierarchical learning techniques to enable the efficient training of federated GAN models. The proposed approach introduces an innovative mechanism that dynamically clusters participant clients to edge servers as well as a novel multi-generator GAN architecture that utilizes non-identical model aggregation stages. The proposed approach has been evaluated on a number of benchmark datasets to measure its performance on higher numbers of participating clients. The results show that HFL-GAN outperforms other comparative state-of-the-art approaches in the training of GAN models in complex non-IID federated learning settings.