TallyTrain: Communication-Efficient Federated Distillation
TallyTrain: 通信高效的联邦蒸馏
机构 * Cisco Systems Inc(思科系统公司)
AI总结 针对联邦学习中模型大小和类别数导致的通信瓶颈,提出TallyTrain方法,通过传输argmax类别索引将每个探针的通信量压缩至⌈log₂C⌉比特,在非独立同分布训练下优于软标签蒸馏,并在标准基准上匹配或超越软标签蒸馏,通信量降低三个数量级。
Comments 27 pages, 7 figures, 12 tables