Standing Firm in 5G: A Single-Round, Dropout-Resilient Secure Aggregation for Federated Learning
Yiwei Zhang, Rouzbeh Behnia, Imtiaz Karim, Attila A. Yavuz, Elisa Bertino
TL;DR
The paper tackles privacy-preserving federated learning in 5G, where high device churn and dropouts hinder secure aggregation. It introduces a single-round secure aggregation protocol that leverages base-station assisted unmasking, precomputation, key-homomorphic pseudorandom functions (KHPRFs), and $t$-out-of-$k$ secret sharing to tolerate dropouts and reduce overhead. A thorough security analysis under a realistic threat model is complemented by experimental results showing lower computation and communication costs, along with robust resilience to UE and BS dropouts. The approach enables scalable, private FL deployments in real-world 5G networks, with potential impact on edge analytics and network optimization.
Abstract
Federated learning (FL) is well-suited to 5G networks, where many mobile devices generate sensitive edge data. Secure aggregation protocols enhance privacy in FL by ensuring that individual user updates reveal no information about the underlying client data. However, the dynamic and large-scale nature of 5G-marked by high mobility and frequent dropouts-poses significant challenges to the effective adoption of these protocols. Existing protocols often require multi-round communication or rely on fixed infrastructure, limiting their practicality. We propose a lightweight, single-round secure aggregation protocol designed for 5G environments. By leveraging base stations for assisted computation and incorporating precomputation, key-homomorphic pseudorandom functions, and t-out-of-k secret sharing, our protocol ensures efficiency, robustness, and privacy. Experiments show strong security guarantees and significant gains in communication and computation efficiency, making the approach well-suited for real-world 5G FL deployments.
