Federated Long-Tailed Learning by Retraining the Biased Classifier with Prototypes

Yang Li, Kan Li*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

1 Citation (Scopus)

Abstract

Federated learning is a privacy-preserving framework that collaboratively trains the global model without sharing raw data among clients. However, one significant issue encountered in federated learning is that biased classifiers affect the classification performance of the global model, especially when training on long-tailed data. Retraining the classifier on balanced datasets requires sharing the client’s information and poses the risk of privacy leakage. We propose a method for retraining the biased classifier using prototypes, that leverage the comparison of distances between local and global prototypes to guide the local training process. We conduct experiments on CIFAR-10-LT and CIFAR-100-LT, and our approach outperforms the accuracy of baseline methods, with accuracy improvements of up to 10%.

Original languageEnglish
Title of host publicationFrontiers in Cyber Security - 6th International Conference, FCS 2023, Revised Selected Papers
EditorsHaomiao Yang, Rongxing Lu
PublisherSpringer Science and Business Media Deutschland GmbH
Pages575-585
Number of pages11
ISBN (Print)9789819993307
DOIs
Publication statusPublished - 2024
Event6th International Conference on Frontiers in Cyber Security, FCS 2023 - Chengdu, China
Duration: 21 Aug 202323 Aug 2023

Publication series

NameCommunications in Computer and Information Science
Volume1992
ISSN (Print)1865-0929
ISSN (Electronic)1865-0937

Conference

Conference6th International Conference on Frontiers in Cyber Security, FCS 2023
Country/TerritoryChina
CityChengdu
Period21/08/2323/08/23

Keywords

  • Federated learning
  • Long-tailed data
  • Privacy protection
  • Prototype learning

Fingerprint

Dive into the research topics of 'Federated Long-Tailed Learning by Retraining the Biased Classifier with Prototypes'. Together they form a unique fingerprint.

Cite this