English

FedAIoT: A Federated Learning Benchmark for Artificial Intelligence of Things

Machine Learning 2024-08-23 v3 Distributed, Parallel, and Cluster Computing Digital Libraries

Abstract

There is a significant relevance of federated learning (FL) in the realm of Artificial Intelligence of Things (AIoT). However, most existing FL works do not use datasets collected from authentic IoT devices and thus do not capture unique modalities and inherent challenges of IoT data. To fill this critical gap, in this work, we introduce FedAIoT, an FL benchmark for AIoT. FedAIoT includes eight datasets collected from a wide range of IoT devices. These datasets cover unique IoT modalities and target representative applications of AIoT. FedAIoT also includes a unified end-to-end FL framework for AIoT that simplifies benchmarking the performance of the datasets. Our benchmark results shed light on the opportunities and challenges of FL for AIoT. We hope FedAIoT could serve as an invaluable resource to foster advancements in the important field of FL for AIoT. The repository of FedAIoT is maintained at https://github.com/AIoT-MLSys-Lab/FedAIoT.

Keywords

Cite

@article{arxiv.2310.00109,
  title  = {FedAIoT: A Federated Learning Benchmark for Artificial Intelligence of Things},
  author = {Samiul Alam and Tuo Zhang and Tiantian Feng and Hui Shen and Zhichao Cao and Dong Zhao and JeongGil Ko and Kiran Somasundaram and Shrikanth S. Narayanan and Salman Avestimehr and Mi Zhang},
  journal= {arXiv preprint arXiv:2310.00109},
  year   = {2024}
}

Comments

Camera-ready version of the Journal of Data-centric Machine Learning Research (DMLR)