FedIDE: Federated Semi-Supervised Learning with Instance Discrimination & LocalEMA

Zhipeng Gao,Shaolong Niu,Chen Zhao,Yang
DOI: https://doi.org/10.1109/icc51166.2024.10622945
2024-01-01
Abstract:Federated Learning is a promising paradigm, of-fering advantages such as access to extensive datasets and robust data privacy preservation. However, a notable challenge arises from the predominant reliance of most federated learning algorithms on labeled data for model training, leaving a limited presence of algorithms capable of effectively utilizing unlabeled data. In real-world scenarios, individuals using electronic devices inadvertently generate copious volumes of unlabeled data, often accompanied by a small fraction of labeled data. This wealth of unlabeled data harbors valuable information, underscoring the importance of developing high-quality federated semi-supervised learning algorithms adept at harnessing both labeled and unlabeled data. This paper introduces FedIDE, an advanced federated semi-supervised learning algorithm. Our approach integrates the principles of pseudo-labeling and instance discrimination, drawing inspiration from contrastive learning, to unlock the potential of unlabeled data. Concurrently, supervised learning is applied to a labeled dataset to enhance model performance. Additionally, we design the LocalEMA local model update algorithm, which amalgamates local and global models during local training, yielding a hybrid model. FedIDE undergoes extensive testing across multiple datasets, surpassing state-of-the-art baselines.
What problem does this paper attempt to address?