Incremental Unsupervised Adversarial Domain Adaptation for Federated Learning in IoT Networks

Abstract

Federated learning, as an effective machine learning paradigm, can collaboratively training an efficient global model by exchanging the network parameters between edge nodes and the cloud server without sacrificing data privacy. Unfortunately, the obtained global model cannot generalize to newly collected unlabeled data since the unlabeled data collected by different edge devices are diverse. Furthermore, the distributions of collected labeled data and unlabeled data are also different for edge devices. In this paper, we propose a method named Incremental Unsupervised Adversarial Domain Adaptation (IUADA) for federated learning, which aims to reduce the domain shift between the labeled data and unlabeled data in the edge nodes and enhance the performance of the personalized target domain models based on the local unlabeled data. Finally, we evaluate the performance of the proposed method by using three real-world datasets. Extensive experimental results demonstrate that the proposed method is efficient to solve the problem of domain shift and achieves a better performance for unlabeled data for federated learning.