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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Goksu, Ozgu, Pugeault, Nicolas
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909026923577344
author Goksu, Ozgu
Pugeault, Nicolas
author_facet Goksu, Ozgu
Pugeault, Nicolas
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2605_07888
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Enhancing Federated Quadruplet Learning: Stochastic Client Selection and Embedding Stability Analysis
Goksu, Ozgu
Pugeault, Nicolas
Machine Learning
Computer Vision and Pattern Recognition
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.
title Enhancing Federated Quadruplet Learning: Stochastic Client Selection and Embedding Stability Analysis
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2605.07888