Skip to content

Similar papers

Conference Open access Jun 2026

Drift-Aware Straggler Mitigation for Heterogeneous Split Federated Learning via Proactive Cut-Layer and Asynchronous Server Updates

Results indicate that jointly optimizing cut-layer adaptation and asynchronous aggregation is an effective direction for improving the robustness and efficiency of SFL in heterogeneous environments.

Junqi Zhang, Bingxu Chen, Yihang Tang et al. · 3 citations
2026

Distribution Deviation-Aware Split Federated Learning in Resource-Limited Wireless Networks

The escalating complexity of deep neural networks introduces substantial challenges to deploying federated learning (FL) in resource-limited edge environments. To address these limitations, split federated learning (SFL) has emerged as a promising paradigm, alleviating client-side computational and communication burdens via strategic model splitting, and periodically aggregating client-side and server-side models consistent with the principles of FL. Nevertheless, existing SFL frameworks encounter significant performance degradation arising from data heterogeneity and imbalance, client heterogeneity, as well as constrained wireless resources. To overcome these issues, this paper introduces a novel data distribution deviation-aware split federated learning (DA-SFL) framework. DA-SFL dynamically adjusts aggregation weights according to the deviation of clients’ data distributions from a global distribution, effectively mitigating biases induced by data imbalance and heterogeneity. Furthermore, we theoretically establish the convergence bound of DA-SFL under a non-convex loss function setting, demonstrating that minimizing the data deviation in each training round enhances learning efficacy. Motivated by this, we formulate a mixed-integer nonlinear programming to optimize learning performance under long-term energy constraints. Leveraging the Lyapunov optimization framework, we decompose the problem into a series of tractable subproblems in each learning round, and propose efficient algorithms to find the client scheduling, adaptive cut layer selection, bandwidth allocation, and aggregation weighting policies. Extensive experimental evaluations conducted on Fashion-MNIST, CIFAR-10, and CINIC-10 datasets across diverse scenarios of data heterogeneity and imbalance demonstrate that DA-SFL significantly outperforms baselines regarding test accuracy, time and energy efficiency, while exhibiting notable robustness and scalability.

Chunfeng Xie, Zhixiong Chen, Wenqiang Yi et al. · 1 citation
Conference Jul 2026

FedEdge-Adapt: Adaptive Federated Learning for Heterogeneous Edge AI Systems

Federated learning (FL) on heterogeneous edge networks faces a fundamental tension: standard aggregation protocols assume client homogeneity, yet real-world edge deployments span device tiers with $7 \times$ compute and $\mathbf{1 0} \times$ bandwidth disparities. Slow clients become stragglers that stall synchronous rounds, while uniform gradient compression degrades accuracy on bandwidth-constrained devices. This paper presents FedEdge-Adapt, a novel adaptive federated learning framework that jointly addresses device heterogeneity, straggler mitigation, and communication efficiency without sacrificing model quality. FedEdge-Adapt introduces three tightly coupled mechanisms: (1) tier-aware gradient compression that applies device-class-specific sparsification ratios, (2) drift-corrected aggregation that reweights client updates based on staleness and data heterogeneity, and (3) predictive client selection that anticipates dropout-prone devices using a lightweight resource oracle. We evaluate FedEdge-Adapt on a 30-node heterogeneous edge network over 150 communication rounds using the CIFAR-10 dataset under non-IID distributions ($\alpha=0.5$ Dirichlet) and compare against FedAvg, FedProx, and SCAFFOLD baselines. FedEdge-Adapt achieves 85.44% global accuracy, a 6.85 percentage-point improvement over FedAvg, while simultaneously reducing round latency by $\mathbf{6 7. 4 \%}$, communication overhead by $\mathbf{3 4. 0 \%}$, and client dropout rate by $\mathbf{5 4. 0 \%}$. Convergence is reached in 18 rounds versus 31 for FedAvg. Extended experiments across 100+ rounds confirm long-term stability with no late-stage divergence.

Saher Elsayed, Mohamed Ali, Samer Abubaker et al. · 0 citations
Preprint Jul 2026

Robust Federated Learning Under Real-World Client Churn

FeLiX is presented, an FL orchestration framework that minimizes wall-clock time-to-target accuracy on live interaction streams and achieves near-oracular performance in real-world settings.

Dhruv Garg, Neha Lakhani, Debopam Sanyal et al. · 0 citations
2026

When Split Federated Learning Meets Prototype Learning: A Communication-Efficient Approach in Wireless Networks

Nowadays, split federated learning (SFL) has emerged as an effective paradigm for enabling privacy-preserving collaborative intelligence across heterogeneous devices with limited computation. However, SFL incurs significant communication overhead in wireless networks due to the uplink transmission of high-dimensional smashed data, which degrades network efficiency. To mitigate the communication bottleneck, we propose a prototype-based SFL framework ProtoSFL. Specifically, each selected client computes local prototypes for observed classes and uploads them to the server. Based on the received prototypes, the server derives global prototypes and optimizes a weighted objective that combines classification loss with prototype alignment loss. The server then updates the model accordingly and returns personalized prototype gradients to the clients. Simulation results verify the effectiveness of ProtoSFL in reducing communication overhead, achieving a substantial reduction in uplink communication, while maintaining competitive testing accuracy under various heterogeneous data settings compared with SFL baselines.

Xinran Zhang, Xianke Qiang, Weilong Chen et al. · 0 citations