A Hybrid Federated Learning Framework for Task-Oriented Semantic Communication

Haofeng Sun, Wanli Ni, Hui Tian, Jingheng Zheng, Gaofeng Nie, Dusit Niyato

Abstract: In existing deep learning-based semantic communication systems, centralized training of semantic models brings a risk of privacy leakage, whereas distributed training imposes a huge computational burden on user equipments (UEs). To address these challenges, we propose a hybrid federated learning (Hybrid-FL) framework to alleviate the computational burden on UEs while protecting the user privacy. Specifically, each UE uploads local gradients and semantic symbols to the base station for the collaborative training of global and local semantic models. Furthermore, we propose a joint communication and computation scheme for supporting the model aggregation and semantics transmission. To gain deep insights, we expose the joint impact of communication and computation on the convergence behavior of Hybrid-FL by deriving an upper bound. Then, we formulate a mixed-integer nonlinear programming problem to improve the convergence performance of Hybrid-FL, which is then effectively solved by using our proposed algorithm that developed based on alternating and matching theory. Experimental results demonstrate that Hybrid-FL outperforms conventional FL by achieving a 20% accuracy gain and a 80% latency reduction.

External IDs:dblp:journals/iotj/SunNTZNN25