CoeF-SFL: Preserving Collaborative Server-Client Learning with Enhanced Communication Efficiency
cs.AI, stat.ML
Submitted: 2026-09-28
Updated: 2026-09-28
Comments: Submitted to a conference
License: http://creativecommons.org/licenses/by/4.0/
The gist: Split Federated Learning (SFL) enables resource-constrained clients to participate in collaborative training, but vanilla SFL exchanges smashed data and gradients at every batch, which incurs
Terminology
Abstract
Split Federated Learning (SFL) enables resource-constrained clients to participate in collaborative training, but vanilla SFL exchanges smashed data and gradients at every batch, which incurs significant communication overhead. Recent methods reduce this overhead with an auxiliary network at the client-side cut layer. However, we identify that this approach makes the client optimize a local objective that differs from the end-to-end objective, which fundamentally limits the collaborative training between the client and the server. We propose Compensated Feedback based SFL (CoeF-SFL), a communication-efficient framework that retains the end-to-end objective without any auxiliary network. In CoeF-SFL, the client and the server exchange the smashed data and the gradients once per round and reuse them during local training. Since this reuse makes the gradients stale on the client side, we compensate them with a curvature-based correction in the activation space and develop two variants. CoeF-D approximates the Hessian with a diagonal gradient outer product, while CoeF-J exploits the tractable Jacobian-based Hessian of a surrogate loss that upper-bounds the true loss. We provide the theoretical background of each method, characterizing its compensation. Across vision and language tasks, model capacities, cut layers, and data distributions, CoeF-SFL significantly outperforms auxiliary-network-based methods under the same communication frequency, and the improvement is most substantial on vision tasks. Code is available at https://anonymous.4open.science/r/CoeF-SFL-2686/README.md
Sources
- Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
- Online Meta-Learning
- Neural Tangent Kernel: Convergence and Generalization in Neural Networks
- Towards Straggler-Resilient Split Federated Learning: An Unbalanced Update Approach
- RoBERTa: A Robustly Optimized BERT Pretraining Approach
- Communication and Storage Efficient Federated Split Learning
- FSL-SAGE: Accelerating Federated Split Learning via Smashed Activation Gradient Estimation
- Opening the Black Box of Deep Neural Networks via Information
- Split learning for health: Distributed deep learning without sharing raw patient data
- GAS: Generative Activation-Aided Asynchronous Split Federated Learning
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems