cs.LGApr 3, 2026

FedSQ: Optimized Weight Averaging via Fixed Gating

Authors: Cristian Pérez-Corral, Jose I. Mestre, Alberto Fernández-Hernández, Manuel F. Dolz, José Duato, Enrique S. Quintana-Ortí

Organizations: Universitat Politècnica de València Valencia, Spain · Universitat Jaume I Castelló de la Plana, Spain · Openchip & Software Technologies Barcelona, Spain

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.

Figures & tables

Explore similar work

CardsList
  1. FedHQ: Hybrid Runtime Quantization for Federated Learning

    May 17, 2025Zihao Zheng, Ziyao Wang, Jiawei Yang +7Federated LearningAdaptive Quantization

  2. Federated Lightweight Fine-Tuning

    Jul 20, 2026Radhakrishna Achanta, Will ReedFederated Learning AggregationFederated Fine-Tuning