Equitable Federated Learning with Activation Clustering

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Upadhyay, Antesh, Hashemi, Abolfazl
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913569379975168
author Upadhyay, Antesh
Hashemi, Abolfazl
author_facet Upadhyay, Antesh
Hashemi, Abolfazl
contents Federated learning is a prominent distributed learning paradigm that incorporates collaboration among diverse clients, promotes data locality, and thus ensures privacy. These clients have their own technological, cultural, and other biases in the process of data generation. However, the present standard often ignores this bias/heterogeneity, perpetuating bias against certain groups rather than mitigating it. In response to this concern, we propose an equitable clustering-based framework where the clients are categorized/clustered based on how similar they are to each other. We propose a unique way to construct the similarity matrix that uses activation vectors. Furthermore, we propose a client weighing mechanism to ensure that each cluster receives equal importance and establish $O(1/\sqrt{K})$ rate of convergence to reach an $ε-$stationary solution. We assess the effectiveness of our proposed strategy against common baselines, demonstrating its efficacy in terms of reducing the bias existing amongst various client clusters and consequently ameliorating algorithmic bias against specific groups.
format Preprint
id arxiv_https___arxiv_org_abs_2410_19207
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Equitable Federated Learning with Activation Clustering
Upadhyay, Antesh
Hashemi, Abolfazl
Machine Learning
Artificial Intelligence
Signal Processing
Federated learning is a prominent distributed learning paradigm that incorporates collaboration among diverse clients, promotes data locality, and thus ensures privacy. These clients have their own technological, cultural, and other biases in the process of data generation. However, the present standard often ignores this bias/heterogeneity, perpetuating bias against certain groups rather than mitigating it. In response to this concern, we propose an equitable clustering-based framework where the clients are categorized/clustered based on how similar they are to each other. We propose a unique way to construct the similarity matrix that uses activation vectors. Furthermore, we propose a client weighing mechanism to ensure that each cluster receives equal importance and establish $O(1/\sqrt{K})$ rate of convergence to reach an $ε-$stationary solution. We assess the effectiveness of our proposed strategy against common baselines, demonstrating its efficacy in terms of reducing the bias existing amongst various client clusters and consequently ameliorating algorithmic bias against specific groups.
title Equitable Federated Learning with Activation Clustering
topic Machine Learning
Artificial Intelligence
Signal Processing
url https://arxiv.org/abs/2410.19207