This repository is a simple pytorch version of the original repository in TF-2.

Please give credit to the original repository github and Paper: arxiv that is recently accepted by NDSS 2025.

Not implement:

• U-shape SL
• SDAR hetero client number

just install pytorch, matplotlib, imageio

pip install -y matplotlib imageio
# pytorch installation should do according to the pytorch official.

Not tinyimagenet dataset need to download. Cifar10, Cifar100, STL10 datasets will be placed in the data dir

mkdir data

cd data
wget http://cs231n.stanford.edu/tiny-imagenet-200.zip
unzip tiny-imagenet-200.zip

./run.sh