arrow
返回

Communication Efficient Federated Learning With Quantization-Aware Training Design

delete2026-01-01
delete1
PRE
AI
X
Xiang Fang
L
Li Chen *
H
Huarui Yin
X
Xiaohui Chen
W
Weidong Wang
DOI:10.1109/TMLCN.2025.3635050delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
模型量化是提高联邦学习(FL)通信效率的有效方法。现有的FL量化协议几乎停留在后训练量化(PTQ)水平,这会导致较大的量化损失,尤其是在低比特量化设置下。本文提出了一种FL量化训练策略以减少量化对模型质量的影响。具体而言,我们首先将量化感知训练(QAT)应用于FL(QAT-FL),通过向模型添加假量化模块来减少量化失真,使模型能够在训练过程中感知未来的量化。在特定假设下建立了QAT-FL算法的收敛性保证。在QAT-FL算法的基础上,我们扩展了对非均匀量化和自适应算法的讨论,使模型能够自适应地调整参数分布和量化比特数以减少训练中的通信量。基于MNIST、CIFAR-10和FEMNIST数据集的实验结果表明,QAT-FL在训练损失和模型推理准确率方面具有优势,且QAT-FL的自适应比特量化也显著提高了通信效率。
Keyword:
Quantization (signal)
Adaptation models
Computational modeling
Training
Convergence
Analytical models
Vectors
Servers
Federated learning
Distortion
quantization-aware learning
non-uniform quantization
post-training quantization
adaptive-bits quantization

期刊

I
IEEE Transactions on Machine Learning in Communications and Networking
IF:
0
论文数:
42
被引数:
0

机构

C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
引用论文

引用论文

err分享
err收藏
err分享
err收藏
Quantization-Aware Training for Mixed Precision Photonic Neural Networks
err2023-09-17
err0
PREAI
errManos Kirtas; Nikolaos Passalis; Athina Oikonomou; Miltos Moralis-Pegios; George Giamourgiannis; Apostolos Tsakyridis; George Mourgias-Alexandris; Nikolaos Pleros; Anastasios Tefas
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容