학술논문

A Practical Recipe for Federated Learning Under Statistical Heterogeneity Experimental Design
Document Type
Periodical
Source
IEEE Transactions on Artificial Intelligence IEEE Trans. Artif. Intell. Artificial Intelligence, IEEE Transactions on. 5(4):1708-1717 Apr, 2024
Subject
Computing and Processing
Federated learning
Benchmark testing
Convergence
Artificial intelligence
Training
Mathematical models
Data models
Benchmark
data heterogeneity
experimental design
federated learning (FL)
machine learning
Non -IID data
Language
ISSN
2691-4581
Abstract
Federated learning (FL) has been an area of active research in recent years. There have been numerous studies in FL to make it more successful in the presence of data heterogeneity. However, despite the existence of many publications, the state of progress in the field is unknown. Many of the works use inconsistent experimental settings and there are no comprehensive studies on the effect of FL-specific experimental variables on the results and practical insights for a more comparable and consistent FL experimental setup. Furthermore, the existence of several benchmarks and confounding variables has further complicated the issue of inconsistency and ambiguity. In this work, we present the first comprehensive study on the effect of FL-specific experimental variables in relation to each other and performance results, bringing several insights and recommendations for designing a meaningful and well-incentivized FL experimental setup. We further aid the community by releasing FedZoo-Bench, an open-source library based on PyTorch with pre-implementation of 22 state-of-the-art methods, and a broad set of standardized and customizable features. We also provide a comprehensive comparison of several SOTA ethods to better understand the current state of the field and existing limitations.