arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.09412cs.LGcs.CV

SemiFed: Semi-supervised Federated Learning with Consistency and Pseudo-Labeling

University of Southern California(南加州大学);Emory University(埃默里大学);Xidian University(西安电子科技大学)

Haowen Lin, Jian Lou, Li Xiong, Cyrus Shahabi

英文摘要:

Federated learning enables multiple clients, such as mobile phones and organizations, to collaboratively learn a shared model for prediction while protecting local data privacy. However, most recent research and applications of federated learning assume that all clients have fully labeled data, which is impractical in real-world settings. In this work, we focus on a new scenario for cross-silo federated learning, where data samples of each client are partially labeled. We borrow ideas from semi-supervised learning methods where a large amount of unlabeled data is utilized to improve the model's accuracy despite limited access to labeled examples. We propose a new framework dubbed SemiFed that unifies two dominant approaches for semi-supervised learning: consistency regularization and pseudo-labeling. SemiFed first applies advanced data augmentation techniques to enforce consistency regularization and then generates pseudo-labels using the model's predictions during training. SemiFed takes advantage of the federation so that for a given image, the pseudo-label holds only if multiple models from different clients produce a high-confidence prediction and agree on the same label. Extensive experiments on two image benchmarks demonstrate the effectiveness of our approach under both homogeneous and heterogeneous data distribution settings

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑