Skip to content
Log in
Open access

Context-Aware Online Model Splitting and Device Association for Semi-Decentralized Federated Learning in Internet of Things

Bo Xu Shuang Wang Xiaoyu Tang
Jun 2026 · Italian National Conference on Sensors · Vol 26 · 0 citations · 42 references
Medicine

Abstract

As a distributed approach to Artificial Intelligence (AI) model construction over wireless networks, federated learning (FL) based on multi-device collaborative training can protect data privacy, as well as increase the computing load of local model updates. In contrast, split learning (SL) with proper model splitting can adapt to the computation and transmission capabilities among devices. In this paper, while taking advantage of FL and SL, we concentrate on a semi-decentralized hybrid federated split learning (SD-HFSL) framework, in which we surpass the limitations of a single central server and allow the shared split models to be aggregated among multiple edge servers. To verify the importance of latency optimization for training efficiency, we analyze the convergence performance of SD-HFSL while jointly considering the limited computation and communication resources. Then, aiming at maximizing the long-term training efficiency, we propose an online optimization problem that includes local model splitting and device association. Considering that the training latency is unknown to the system a priori, a context-aware online training algorithm with sublinear regret is proposed based on the framework of contextual multi-armed bandit (CMAB), where the edge servers can observe the context information of device sites for latency estimation, followed by the iterative optimization based on the evaluated information in different contexts. Experiments on several neural network models show that the proposed algorithm reduces training latency and improves test accuracy compared with the selected benchmarks.

DOI PubMed Semantic Scholar
Read PDF

Similar papers

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
Preprint Aug 2026

Cluster-Aware Over-the-Air Federated Learning with Energy-Harvesting Devices: From Global Training to Model Personalization

Over-the-air FL with EH MDs under heterogeneous data distributions under heterogeneous data distributions is studied, and the proposed unified framework improves fairness or personalization, depending on the operating mode, while reducing communication overhead.

F. Bagci, Busra Tegin, Mohammad Kazemi et al. · 0 citations
Open access Jul 2026

DISTRIBUTED INTELLIGENCE AT THE EDGE: A MATHEMATICAL FRAMEWORK FOR DECENTRALIZED LEARNING IN IOT NETWORKS

A novel Federated Edge Learning (FEL) architecture that integrates software-defined networking principles with gossip-based communication protocols to facilitate collaborative model training while preserving data locality is proposed, offering a scalable, privacy-preserving solution for deploying artificial intelligence at the network edge.

N.Durga, A. Mary Posonia, Selvakumar et al. · 0 citations
Open access Jul 2026

Digital Twin-Enabled Dynamic Aggregation for Efficient Federated Learning

Simulation results demonstrate that the proposed digital twin (DT)-based dynamic FL aggregation method not only accelerates model convergence and improves accuracy but also significantly reduces training latency and energy consumption costs compared with baseline FL algorithms.

Wenqin Zhuang, Yuao Wang, Guocheng Wang · 0 citations