mlsanityData to 7 Oct 2026

MergeSFL: Split Federated Learning with Feature Merging and Batch Size Regulation

arXiv 2311.13348 ↗First posted 22 Nov 2023cs.LG cs.DC cs.NIcited by 13

Not on the map. Why?

Abstract (arXiv)

Recently, federated learning (FL) has emerged as a popular technique for edge AI to mine valuable knowledge in edge computing (EC) systems. To mitigate the computing/communication burden on resource-constrained workers and protect model privacy, split federated learning (SFL) has been released by integrating both data and model parallelism. Despite resource limitations, SFL still faces two other critical challenges in EC, i.e., statistical heterogeneity and system heterogeneity. To address these challenges, we propose a novel SFL framework, termed MergeSFL, by incorporating feature merging and batch size regulation in SFL. Concretely, feature merging aims to merge the features from workers into a mixed feature sequence, which is approximately equivalent to the features derived from IID data and is employed to promote model accuracy. While batch size regulation aims to assign diverse and suitable batch sizes for heterogeneous workers to improve training efficiency. Moreover, MergeSFL explores to jointly optimize these two strategies upon their coupled relationship to better enhance the performance of SFL. Extensive experiments are conducted on a physical platform with 80 NVIDIA Jetson edge devices, and the experimental results show that MergeSFL can improve the final model accuracy by 5.82% to 26.22%, with a speedup by about 1.74x to 4.14x, compared to the baselines.

Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 34 of its 83 references that are in the map. First posted on arXiv 22 Nov 2023.

Builds on

34 of 83 references.

References · 34

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
  3. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
  4. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
  5. 2019Advances and Open Problems in Federated Learningcited by 3,021
  6. 2018Federated Learning with Non-IID Datacited by 1,099
  7. 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 836
  8. 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 741
  9. 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 667
  10. 2018Federated Learning for Mobile Keyboard Predictioncited by 565
  11. 2018Adaptive Federated Learning in Resource Constrained Edge Computing Systemscited by 408
  12. 2019Federated Learning in Mobile Edge Networks: A Comprehensive Surveycited by 407
22 more
  1. 2018Split learning for health: Distributed deep learning without sharing raw patient datacited by 391
  2. 2019Bayesian Nonparametric Federated Learning of Neural Networkscited by 385
  3. 2020SplitFed: When Federated Learning Meets Split Learningcited by 308
  4. 2018Applied Federated Learning: Improving Google Keyboard Query Suggestionscited by 202
  5. 2020FetchSGD: Communication-Efficient Federated Learning with Sketchingcited by 178
  6. 2021Deep Learning with Label Differential Privacycited by 113
  7. 2018Federated Learning for Keyword Spottingcited by 95
  8. 2022Divergence-aware Federated Self-Supervised Learningcited by 71
  9. 2021Tackling System and Statistical Heterogeneity for Federated Learning with Adaptive Client Samplingcited by 66
  10. 2020Communication-Efficient and Distributed Learning Over Wireless Networks: Principles and Applicationscited by 62
  11. 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 57
  12. 2022FederatedScope: A Flexible Federated Learning Platform for Heterogeneitycited by 53
  13. 2022FedBalancer: Data and Pace Control for Efficient Federated Learning on Heterogeneous Clientscited by 46
  14. 2020Advancements of federated learning towards privacy preservation: from federated learning to split learningcited by 34
  15. 2022Data Heterogeneity-Robust Federated Learning via Group Client Selection in Industrial IoTcited by 25
  16. 2020FedSL: Federated Split Learning on Distributed Sequential Data in Recurrent Neural Networkscited by 23
  17. 2021Server-Side Local Gradient Averaging and Learning Rate Acceleration for Scalable Split Learningcited by 22
  18. 2021LegoDNN: Block-grained Scaling of Deep Neural Networks for Mobile Visioncited by 20
  19. 2021SplitAVG: A heterogeneity-aware federated deep learning method for medical imagingcited by 19
  20. 2021Auto-weighted Robust Federated Learning with Corrupted Data Sourcescited by 12
  21. 2023Taming Resource Heterogeneity In Distributed ML Training With Dynamic Batchingcited by 12
  22. 2022Adaptive Configuration for Heterogeneous Participants in Decentralized Federated Learningcited by 11

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 4; 2025: 6; 2026: 3.

13 papers.

Sort: