NVIDIA FLARE로 Non-IID CIFAR-10에서 FedAvg와 FedProx 연합 학습을 구축하고 비교하는 단계별 가이드
Step-by-Step Guide to Build and Compare FedAvg and FedProx Federated Learning on Non-IID CIFAR-10 with NVIDIA FLARE
TL;DR AI
1분핵심 요약
이 튜토리얼은 NVIDIA FLARE를 사용해 비-IID CIFAR-10 환경에서 FedAvg와 FedProx를 비교하는 방법을 보여준다.
Dirichlet 분할로 CIFAR-10을 여러 시뮬레이션 클라이언트에 나눠 현실적인 데이터 불균형을 재현한다.
NVIDIA FLARE의 Job API와 Client API로 여러 라운드에 걸쳐 모델을 학습, 교환, 집계한다.
통신 라운드별 전역 정확도를 측정해 비-IID 조건에서 두 알고리즘의 차이를 비교한다.