Robust communication strategy for federated learning by incorporating self-attention

Yawen Xu; Xiaojun Li; Zeyu Yang; HengJie Song

doi:10.1117/12.2581491

10 November 2020 Robust communication strategy for federated learning by incorporating self-attention

Yawen Xu, Xiaojun Li, Zeyu Yang, HengJie Song

Proceedings Volume 11584, 2020 International Conference on Image, Video Processing and Artificial Intelligence; 115841F (2020) https://doi.org/10.1117/12.2581491
Event: Third International Conference on Image, Video Processing and Artificial Intelligence, 2020, Shanghai, China

Abstract

Federated learning is an emerging machine learning setting, which can train a shared model on large amounts of decentralized data while protecting data privacy. However, the communication cost of federated learning is heavy, especially for mobile devices with higher latency and lower throughput. Although several algorithms have been proposed to reduce the communication cost, they are extremely sensitive to data distribution, even inapplicable to the real client Non-IID data. In this paper, we propose an effective communication strategy for federated learning called FedSAA, which increases the testing performance on Non-IID data by introducing self- attention mechanism. Two major innovations of our paper are presented here. Firstly, we utilize self-attention mechanism to optimize both the server-to-client and the client-to-client parameter divergence during the model aggregation process so as to improve the model robustness for Non-IID data. Secondly, we adopt the sign compression operator to help data transmission between nodes. The experimental results demonstrate that the model accuracy of our communication-efficient strategy for federated learning with Non-IID data is superior to other communication-efficient algorithms.

Citation Download Citation

Yawen Xu, Xiaojun Li, Zeyu Yang, and HengJie Song "Robust communication strategy for federated learning by incorporating self-attention", Proc. SPIE 11584, 2020 International Conference on Image, Video Processing and Artificial Intelligence, 115841F (10 November 2020); https://doi.org/10.1117/12.2581491

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available