English

Enhancing Federated Quadruplet Learning: Stochastic Client Selection and Embedding Stability Analysis

Machine Learning 2026-05-11 v1 Computer Vision and Pattern Recognition

Abstract

Federated Learning (FL) enables decentralised model training across distributed clients without requiring data centralisation. However, the generalisation performance of the global model is usually degraded by data heterogeneity across clients, particularly under limited data availability and class imbalance. To address this challenge, we propose FedQuad, a novel method that explicitly enforces minimising intra-class representations while enabling inter-class splits across clients. By jointly minimising distances between positive pairs and maximising distances between negative pairs, the proposed approach mitigates representation misalignment introduced during model aggregation. We evaluate our method on CIFAR-10, CIFAR-100, and Tiny-ImageNet under diverse non-IID settings and varying numbers of clients, demonstrating consistent improvements over existing baselines. Additionally, we provide a comprehensive analysis of metric learning-based approaches in both centralised and federated environments, highlighting their effectiveness in alleviating representation collapse under heterogeneous data distributions.

Keywords

Cite

@article{arxiv.2605.07888,
  title  = {Enhancing Federated Quadruplet Learning: Stochastic Client Selection and Embedding Stability Analysis},
  author = {Ozgu Goksu and Nicolas Pugeault},
  journal= {arXiv preprint arXiv:2605.07888},
  year   = {2026}
}

Comments

arXiv admin note: substantial text overlap with arXiv:2509.04107

R2 v1 2026-07-01T12:58:00.977Z