FedMDR: Federated Model Distillation with Robust Aggregation

Yuxi Mi,Yutong Mu,Shuigeng Zhou,Jihong Guan
DOI: https://doi.org/10.1007/978-3-030-85899-5_2
2021-01-01
Abstract:This paper presents FedMDR, a federated model distillation framework with a novel, robust aggregation mechanism that exploits transfer learning and knowledge distillation. FedMDR adopts a weighted geometric-median-based aggregation with trimmed prediction accuracy on the server-side, which orchestrates communication-efficient training on both heterogeneous model architectures and non-i.i.d. data. The aggregation provides resilience to sharp accuracy drop of corrupted models. We also extend FedMDR to support differential privacy by adding Gaussian noise to the aggregated consensus. Results show that FedMDR achieves significant robustness gain and satisfactory accuracy, and outperforms the existing techniques.
What problem does this paper attempt to address?