Byzantine-Resilient Federated Machine Learning via Over-the-Air Computation

Yuanming Shi, Yong Zhou, Ting Wang, Shaoming Huang

Publication date: 23 May 2021

Abstract: Federated learning (FL) is recognized as a key enabling technology to provide intelligent services for future wireless networks and industrial systems with delay and privacy guarantees. However, the performance of wireless FL can be significantly degraded by Byzantine attack, such as data poisoning attack, model poisoning attack and free-riding attack. To design the Byzantine-resilient FL paradigm in wireless networks with limited radio resources, we propose a novel communication-efficient robust model aggregation scheme via over-the-air computation (AirComp). This is achieved by applying the Weiszfeld algorithm to obtain the smoothed geometric median aggregation against Byzantine attack. The additive structure of the Weiszfeld algorithm is further leveraged to match the signal superposition property of multiple-access channels via AirComp, thereby expediting the communication-efficient secure aggregation process of FL. Numerical results demonstrate the robustness against Byzantine devices and good learning performance of the proposed approach.

Has companion code repository: https://github.com/goldenBill/Byzantine_AirComp

This page was built for publication: Byzantine-Resilient Federated Machine Learning via Over-the-Air Computation

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q6368322)