seyoungahn/KDFL-torch

Improving generalization of federated learning in heterogeneous IoT Edge Networks

Todo list

1. Related works 조사

• Federated distillation관련 paper survey
• Knowledge distillation관련 paper survey
• unbalanced data in federated learning 관련 paper survey

2. Baseline performance 실험

• FL, KD 성능 각각 확보
• ResNet-20, AlexNet, VGGNet, DenseNet

3. NB-IoT simulator를 활용한 heterogeneous data environment 구성

• Dataset shaping
• Communication delay 측정
• 전체 학습 시간 측정
• KD + FL 통합

Hyperparameter JSON file structure