This is the source code for implementing the paper of “Communication-efficient federated edge learning via optimal probabilistic device scheduling”, in this paper, we proposed an optimized scheduling policy for total communication time minimization.
For any reproduce, further research or development, please kindly cite our TWC journal paper:
M. Zhang, G. Zhu, S. Wang, J. Jiang, Q. Liao, C. Zhong, and S. Cui, "Communication-Efficient Federated Edge Learning via Optimal Probabilistic Device Scheduling," in IEEE Transactions on Wireless Communications, vol. 21, no. 10, pp. 8536-8551, Oct. 2022, doi: 10.1109/TWC.2022.3166941. |
One experiments in the paper is included.
- CNN on MNIST dataset
#Requirements python = 3.6.5
pytorch
#Commands for running the experiments:
python main_fed_multiuser.py --dataset mnist --num_channels 1 --model cnn --num_users 30 --epochs 1000 --lr 1 --nu 100 --frac 1.0 --local_ep 1 --momentum 0.0 --local_bs 50 --bs 50 --mode 'schedule' --schedule_policy 'CTM' --schedule_user_num 1 --lr_schedule --differ_label '_v10' --gpu 1 --rho 0.5
#Note
-
The output of the experiment is a .log file including all the training results, e.g. training loss and test accuracy;
-
The configurable parameters in the code are defined in the file
option.py
.