Skip to content
Open access

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

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.

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

Federated learning (FL) enables distributed optimization and learning across decentralized edge devices while preserving data privacy, but its performance is fundamentally constrained by heterogeneous data distributions, limited communication resources, and energy availability. In practical wireless networks, mobile devices (MDs) often exhibit diverse data and learning objectives, naturally forming clusters of users with jointly trainable models. When devices rely on energy harvesting (EH), stochastic energy arrivals further complicate participation and scheduling under communication constraints. In this work, we study over-the-air (OTA) FL with EH MDs under heterogeneous data distributions, and investigate two closely related learning objectives within a unified framework: one aiming for a more representative global model by reducing data bias, and the other learning more personalized cluster-specific models by exploiting this bias. In the global training mode, cluster information guides energy- and diversity-aware scheduling, ensuring that the scheduled active users provide a more representative aggregate update. In the personalization mode, the same cluster structure defines cluster-level learning objectives and OTA recovery targets, enabling the parameter server to train multiple cluster-specific models through simultaneous transmissions over the wireless multiple-access channel. Numerical results demonstrate that 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

The proliferation of Internet of Things (IoT) devices has generated unprecedented volumes of data, rendering traditional cloud-centric processing paradigms inadequate due to latency constraints, bandwidth limitations, and privacy concerns. This paper presents a comprehensive mathematical framework for distributed intelligence at the edge, enabling decentralized learning across heterogeneous IoT networks. We propose 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. The framework addresses critical challenges including device heterogeneity, non-independent and identically distributed (non-i.i.d.) data distributions, resource constraints, and communication efficiency. We formalize the decentralized learning problem, derive convergence bounds under heterogeneous conditions, and introduce a multi-worker selection mechanism optimized through swarm learning principles. Experimental validation using real-world IoT datasets demonstrates that our approach achieves 30-50% reduction in training latency and 35-55% decrease in energy consumption compared to conventional federated averaging methods, while maintaining competitive accuracy of 92.86% on classification tasks . The proposed framework offers 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
Open access Jul 2026

Federated Learning over 5G/6G Networks: Dynamic Client Selection and Resource Allocation for Heterogeneous Edge Environments

Federated learning (FL) has emerged as a promising paradigm for privacy-preserving edge intelligence because it enables geographically distributed devices to collaboratively train a shared model without transferring raw data to a central cloud. This capability is particularly valuable for 5G and emerging 6G networks, where edge-native services are required to satisfy stringent latency, bandwidth, and privacy constraints while operating on highly heterogeneous devices and time-varying wireless channels. In practice, however, synchronous FL is often constrained by straggling clients with limited computation capability or unfavorable communication conditions, which increases round latency and reduces overall resource efficiency. To address this challenge, this study develops a rigorously structured framework for dynamic client selection and radio resource allocation in heterogeneous wireless edge environments. Each FL round is formulated as a latency-aware scheduling problem that jointly captures local computation time, uplink transmission time, minimum participation constraints, and resource block assignment. On this basis, we propose a Dynamic Client Selection and Resource Allocation (DCS-RA) method that integrates computation-aware, channel-aware, and fairness-aware scoring with greedy resource block allocation guided by marginal completion time reduction. The study further provides a clear methodological structure, workflow visualization, literature-grounded justification, dataset documentation, and uncertainty-aware result reporting. Under the reported simulation setting with 100 clients and 20 resource blocks, DCS-RA reduces the average round completion time from 1.92 s to 1.55 s on MNIST and from 2.02 s to 1.57 s on CIFAR-10, corresponding to improvements of 19.39% and 22.47%, respectively. Standard deviation reductions of 70.59% and 80.77% further indicate improved round-to-round stability and more reliable training behavior. These results support the central conclusion that lightweight joint scheduling can materially improve wall-clock FL efficiency in heterogeneous 5G/6G edge networks.

A. Al-Karawi, R. Akdeniz · 0 citations