arrow
Return

Federated Learning Hyperparameter Tuning From a System Perspective

delete2023-08-15
delete6
delete
OA
AI
L
Lei Fu
M
Mi Zhang
胡
胡鹏飞 (Pengfei Hu) *
成
成秀珍 (Xiuzhen Cheng)
P
Prasant Mohapatra
X
Xin Liu
DOI:10.1109/JIOT.2023.3253813delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Federated learning (FL) is a distributed model training paradigm that preserves clients' data privacy. It has gained tremendous attention from both academia and industry. FL hyperparameters (e.g., the number of selected clients and the number of training passes) significantly affect the training overhead in terms of computation time, transmission time, computation load, and transmission load. However, the current practice of manually selecting FL hyperparameters imposes a heavy burden on FL practitioners because applications have different training preferences. In this article, we propose FedTune, an automatic FL hyperparameter tuning algorithm tailored to applications' diverse system requirements in FL training. FedTune iteratively adjusts FL hyperparameters during FL training and can be easily integrated into existing FL systems. Through extensive evaluations of FedTune for diverse applications and FL aggregation algorithms, we show that FedTune is lightweight and effective, achieving 8.48%-26.75% system overhead reduction compared to using fixed FL hyperparameters. This article assists FL practitioners in designing high-performance FL training solutions. The source code of FedTune is available at https://github.com/DataSysTech/FedTune.
Keywords:
Federated learning (FL)
hyperparameter tuning
optimization
system overhead
training preference

Journal

IEEE Internet of Things Journal cover
IEEE Internet of Things Journal
IF:
8.9
Papers:
1.4W
Citations:
7.8W

Organization

F
fudan university
Scholars:
11.8W
Papers: 7.7W
Citations: 121
O
Ohio State University
Scholars:
4.1W
Papers: 3.2W
Citations: 80
S
shandong university
Scholars:
9.5W
Papers: 6.4W
Citations: 94
University of California System cover
University of California System
Scholars:
37.7W
Papers: 33.8W
Citations: 6.6K
researcher View more organizations
Cited Papers

Cited Papers

Mental illness in China
err2009-09-01
err0
PREAI
errAdley Tsang
errShare
errSave
Total Synthesis of (–)‐Epothilone B: An Extension of the Suzuki Coupling Method and Insights into Structure–Activity Relationships of the Epothilones
err2003-12-22
err0
PREAI
errDai‐Shi Su; Dongfang Meng; Peter Bertinato; Aaron Balog; Erik J. Sorensen; Samuel J. Danishefsky; Yu‐Huang Zheng; Ting‐Chao Chou; Lifeng He; Susan B. Horwitz
errShare
errSave
Single-step genome-wide association studies and post-GWAS analyses for the number of oocytes and embryos in Gir cattle
err2023-07-12
err0
errOAAI
errRenata de Fátima Bretanha Rocha; Arielly Oliveira Garcia; Pamela Itajara Otto; Mateus Guimarães dos Santos; Marcos Vinícius Barbosa da Silva; Marta Fonseca Martins; Marco Antônio Machado; João Claudio do Carmo Panetto; Simone Eliza Facioni Guimarães
errShare
errSave
Machine Learning Applications for Precision Agriculture: A Comprehensive Review
err2021-01-01
err365
errOAAI
errSharma, Abhinav; Jain, Arpit; Gupta, Prateek; Chowdary, Vinay
errShare
errSave
A Smartphone-Enabled Fall Detection Framework for Elderly People in Connected Home Healthcare
err2019-11-01
err44
PREAI
errHassan, Mohammad Mehedi; Gumaei, Abdu; Aloi, Gianluca; Fortino, Giancarlo; Zhou, Mengchu
errShare
errSave
A Survey on Distributed Machine Learning
err2020-03-20
err430
errOAAI
errVerbraeken, Joost; Wolting, Matthijs; Katzy, Jonathan; Kloppenburg, Jeroen; Verbelen, Tim; Rellermeyer, Jan S.
errShare
errSave
researcher View more