Federated Active Learning (F-AL): an Efficient Annotation Strategy for Federated Learning

Abstract

Federated learning (FL) has been intensively investigated in terms of communication efficiency, privacy, and fairness. However, efficient annotation, which is a pain point in real-world FL applications, is less studied. In this project, we propose to apply active learning (AL) and sampling strategy into the FL framework to reduce the annotation workload. We expect that the AL and FL can improve the performance of each other complementarily. In our proposed federated active learning (F-AL) method, the clients collaboratively implement the AL to obtain the instances which are considered as informative to FL in a distributed optimization manner. We compare the test accuracies of the global FL models using the conventional random sampling strategy, client-level separate AL (S-AL), and the proposed F-AL. We empirically demonstrate that the F-AL outperforms baseline methods in image classification tasks.

13 pages, 9 figures, submitted for conference publication

References in corpus (4)

• Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
• Ensemble Distillation for Robust Model Fusion in Federated Learning
• Deep Active Learning over the Long Tail
• Combining Federated and Active Learning for Communication-efficient Distributed Failure Prediction in Aeronautics