This repo is the implementation of FedSkel: Efficient Federated Learning on Heterogeneous Systems with Skeleton Gradients Update [CIKM'21].
FedSkel: Efficient Federated Learning on Heterogeneous Systems with Skeleton Gradients Update
Doi: 10.1145/3459637.3482107
We verified that FedSkel will not affect FL accuracy with PyTorch Framework. The core codes are in ./acc-pytorch. We adapted the FL framework of LG-FedAvg for FedSkel.
We measure the speedups of FedSkel on mobile devices with Caffe framework. The core codes are in ./speedup-caffe. We modified CONV layers of Caffe to enable gradients' pruning in the SetSkel and UpdateSkel processes during training.