RAFL: A Robust and Adaptive Federated Meta-Learning Framework Against Adversaries

Yu, Zhengxin and Lu, Yang and Suri, Neeraj (2023) RAFL: A Robust and Adaptive Federated Meta-Learning Framework Against Adversaries. In: 2023 IEEE 20th International Conference on Mobile Ad Hoc and Smart Systems (MASS) :. IEEE. ISBN 9798350324341

Text (Robust_FL_short_version_2_final)
Robust_FL_short_version_2_final.pdf - Accepted Version
Available under License Creative Commons Attribution.
Download (719kB)

Abstract

With the emergence of data silos and increasing privacy awareness, traditional centralized machine learning provides limited support. Federated learning (FL), as a promising alternative machine learning approach, is capable of leveraging distributed personalized datasets from multiple clients to train a shared global model in a privacy-preserving manner. However, FL systems are vulnerable to attacker-controlled adversarial clients that potentially conduct adversarial attacks by uploading unreliable model updates or clients unintentionally uploading low-quality models leading to degraded FL performance and reduced resilience to attacks. In this paper, we propose RAFL: a new robust-by-design federated meta learning framework capable of mitigating adversarial model updates on non-IID data. RAFL leverages 1) a residual rule-based detection method and a Variational AutoEncoder (VAE) learning based detection method combined to distinguish adversarial clients from benign clients. 2) a similarity-based model aggregation method to reduce the likelihood of uploading adversarial models from adversarial clients. 3) multiple learning loops to collaboratively train multiple personalized detection models against adversaries effectively. Experimental results demonstrate that our proposed FL framework is robust by design and outperforms other defensive methods against adversaries in terms of model accuracy and efficiency.