arrow
返回

Ternary Compression for Communication-Efficient Federated Learning

delete2022-03-01
delete125
delete
OA
AI
J
Jinjin Xu
W
Wenli Du *
Y
Yaochu Jin *
W
Wangli He
R
Ran Cheng
DOI:10.1109/TNNLS.2020.3041185delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
Learning over massive data stored in different locations is essential in many real-world applications. However, sharing data is full of challenges due to the increasing demands of privacy and security with the growing use of smart mobile devices and Internet of thing (IoT) devices. Federated learning provides a potential solution to privacy-preserving and secure machine learning, by means of jointly training a global model without uploading data distributed on multiple devices to a central server. However, most existing work on federated learning adopts machine learning models with full-precision weights, and almost all these models contain a large number of redundant parameters that do not need to be transmitted to the server, consuming an excessive amount of communication costs. To address this issue, we propose a federated trained ternary quantization (FTTQ) algorithm, which optimizes the quantized networks on the clients through a self-learning quantization factor. Theoretical proofs of the convergence of quantization factors, unbiasedness of FTTQ, as well as a reduced weight divergence are given. On the basis of FTTQ, we propose a ternary federated averaging protocol (T-FedAvg) to reduce the upstream and downstream communication of federated learning systems. Empirical experiments are conducted to train widely used deep learning models on publicly available data sets, and our results demonstrate that the proposed T-FedAvg is effective in reducing communication costs and can even achieve slightly better performance on non-IID data in contrast to the canonical federated learning algorithms.
Keyword:
Communication efficiency
deep learning
federated learning
ternary coding
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

U
University of Surrey
学者数:
1.2W
论文数: 1.3W
被引数: 22
引用论文

引用论文

Measuring Highway Impacts on House Prices Using Spatial Regression
err2020-08-04
err0
errOAAI
errMarcus T. Allen; Grant W. Austin; Mushfiq Swaleheen
err分享
err收藏
Robust and Communication-Efficient Federated Learning From Non-i.i.d. Data
err2020-09-01
err1.0K
errOAAI
errSattler, Felix; Wiedemann, Simon; Mueller, Klaus-Robert; Samek, Wojciech
err分享
err收藏
Robust Student Network Learning
err2019-01-01
err48
errOAAI
errGuo, Tianyu; Xu, Chang; He, Shiyi; Shi, Boxin; Xu, Chao; Tao, Dacheng
err分享
err收藏
Simple rules can guide whether land- or ocean-based conservation will best benefit marine ecosystems
err2017-09-06
err0
errOAAI
errMegan I. Saunders; Michael Bode; Scott Atkinson; Carissa J. Klein; Anna Metaxas; Jutta Beher; Maria Beger; Morena Mills; Sylvaine Giakoumi; Vivitskaia Tulloch; Hugh P. Possingham
err分享
err收藏
学者 查看更多内容