GPFL: A Gradient Projection-Based Client Selection Framework for Efficient Federated Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Na, Shijie, Liang, Yuzhi, Yiu, Siu-Ming
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909210844856320
author Na, Shijie
Liang, Yuzhi
Yiu, Siu-Ming
author_facet Na, Shijie
Liang, Yuzhi
Yiu, Siu-Ming
contents Federated learning client selection is crucial for determining participant clients while balancing model accuracy and communication efficiency. Existing methods have limitations in handling data heterogeneity, computational burdens, and independent client treatment. To address these challenges, we propose GPFL, which measures client value by comparing local and global descent directions. We also employ an Exploit-Explore mechanism to enhance performance. Experimental results on FEMINST and CIFAR-10 datasets demonstrate that GPFL outperforms baselines in Non-IID scenarios, achieving over 9\% improvement in FEMINST test accuracy. Moreover, GPFL exhibits shorter computation times through pre-selection and parameter reuse in federated learning.
format Preprint
id arxiv_https___arxiv_org_abs_2403_17833
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GPFL: A Gradient Projection-Based Client Selection Framework for Efficient Federated Learning
Na, Shijie
Liang, Yuzhi
Yiu, Siu-Ming
Machine Learning
Distributed, Parallel, and Cluster Computing
Federated learning client selection is crucial for determining participant clients while balancing model accuracy and communication efficiency. Existing methods have limitations in handling data heterogeneity, computational burdens, and independent client treatment. To address these challenges, we propose GPFL, which measures client value by comparing local and global descent directions. We also employ an Exploit-Explore mechanism to enhance performance. Experimental results on FEMINST and CIFAR-10 datasets demonstrate that GPFL outperforms baselines in Non-IID scenarios, achieving over 9\% improvement in FEMINST test accuracy. Moreover, GPFL exhibits shorter computation times through pre-selection and parameter reuse in federated learning.
title GPFL: A Gradient Projection-Based Client Selection Framework for Efficient Federated Learning
topic Machine Learning
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2403.17833