ARTFEED — Contemporary Art Intelligence

Data-Free Federated Learning Framework Addresses Class Imbalance

ai-technology · 2026-05-20

A new framework called CELM (class-wise contribution estimation and aggregation via logit maximization) enables data-free client contribution estimation in federated learning (FL). FL allows collaborative training of computer vision models without centralizing data, but practical deployments often suffer from class imbalance and label skew, causing standard aggregation to overfit dominant clients. CELM does not require raw data, client metadata, or auxiliary public datasets. The server probes client updates to obtain class-wise evidence scores, assembles a cross-client evidence matrix quantifying per-class competence and coverage, and computes contribution weights that upweight clients providing strong evidence for underrepresented classes. The resulting aggregation is stable. The paper is available on arXiv with ID 2605.18892.

Key facts

  • CELM framework addresses class imbalance in federated learning
  • No raw data, client metadata, or auxiliary datasets required
  • Uses logit maximization for contribution estimation
  • Server probes client updates for class-wise evidence scores
  • Cross-client evidence matrix quantifies per-class competence and coverage
  • Upweights clients with strong evidence for underrepresented classes
  • Aims to improve minority-class performance in FL
  • Paper available on arXiv with ID 2605.18892

Entities

Institutions

  • arXiv

Sources