English

FedSQ: Optimized Weight Averaging via Fixed Gating

Machine Learning 2026-04-06 v1 Artificial Intelligence Distributed, Parallel, and Cluster Computing

Abstract

Federated learning (FL) enables collaborative training across organizations without sharing raw data, but it is hindered by statistical heterogeneity (non-i.i.d.\ client data) and by instability of naive weight averaging under client drift. In many cross-silo deployments, FL is warm-started from a strong pretrained backbone (e.g., ImageNet-1K) and then adapted to local domains. Motivated by recent evidence that ReLU-like gating regimes (structural knowledge) stabilize earlier than the remaining parameter values (quantitative knowledge), we propose FedSQ (Federated Structural-Quantitative learning), a transfer-initialized neural federated procedure based on a DualCopy, piecewise-linear view of deep networks. FedSQ freezes a structural copy of the pretrained model to induce fixed binary gating masks during federated fine-tuning, while only a quantitative copy is optimized locally and aggregated across rounds. Fixing the gating reduces learning to within-regime affine refinements, which stabilizes aggregation under heterogeneous partitions. Experiments on two convolutional neural network backbones under i.i.d.\ and Dirichlet splits show that FedSQ improves robustness and can reduce rounds-to-best validation performance relative to standard baselines while preserving accuracy in the transfer setting.

Keywords

Cite

@article{arxiv.2604.02990,
  title  = {FedSQ: Optimized Weight Averaging via Fixed Gating},
  author = {Cristian Pérez-Corral and Jose I. Mestre and Alberto Fernández-Hernández and Manuel F. Dolz and José Duato and Enrique S. Quintana-Ortí},
  journal= {arXiv preprint arXiv:2604.02990},
  year   = {2026}
}
R2 v1 2026-07-01T11:52:46.511Z