Decentralized Federated Learning for Heterogeneous Multi-Task Semantic Communication
Authors: Lin Yin, Tiejun Lv, Weicai Li, Xi Yu, Xiaoyu He
Organizations: School of Information and Communication Engineering, Beijing University of Posts and Telecommunications (BUPT), Beijing 100876, China · Center for Target Cognition Information Processing Science and Technology, and the Key Laboratory of Modern Measurement and Control Technology, Ministry of Education, both at Beijing Information Science and Technology University, Beijing, China
Collaborative training in distributed semantic communication (DSC) networks typically relies on decentralized federated learning (DFL). However, pushing topology-agnostic aggregation into heterogeneous, multi-task environments creates a fundamental bottleneck: it drives negative transfer and overconsensus bias (OCB). This paper introduces a personalized DSC framework that cuts off this cross-task interference. At the node level, a policy-driven multi-path routing mechanism separates task-specific features from shared representations to preserve local fidelity. Across the network, we deploy a "communicationwhile- aggregation" protocol. It calibrates a column-stochastic consensus matrix using task affinities. This limits the system to absorbing complementary knowledge while actively blocking mismatched parameter updates. To bound the convergence, we derive a unified Lyapunov drift analysis. We reveal a strict Ushaped trade-off: deeper topological mixing reduces variance but amplifies structural OCB. Resolving this tension yields a closed-form expression for the optimal aggregation depth. We evaluate the proposed framework on NYU-v2, where the results reveal a clear trade-off between insufficient aggregation and excessive topological mixing. At the analytically derived optimal aggregation depth, our method achieves a 4.77% global relative improvement over the no-aggregation baseline and outperforms decentralized FedAvg, FedAMP, and heuristic max aggregation. We further evaluate the framework on Taskonomy and imperfect wireless links to examine the effects of network-size variation and wireless-link reliability.
Figures & tables
Notation
Definition
i,N
Client index, and set of clients in the network.
t,T
Task index, and set of global semantic tasks.
Ti
Personalized task set of client i .
n,Ni
One-hop neighbor index, and set of neighbors of client i .
ut
Learnable binary policy vector for task t .
st,t′
Semantic similarity between task t and task t′ .
TABLE 1: Summary of Key Notations
Fig. 1 : The proposed distributed semantic communication framework. The distinct colors indicate different tasks and semantic information.
Fig. 2 : Schematic of the policy-driven multi-path semantic encoding and task-specific decoding architecture for heterogeneous multi-task execution.
Fig. 3 : Workflow of the proposed “communication-while-aggregation” protocol for the distributed personalized SC system.
Method
Seg
Sn
Dep
Δall (%)
mIoU
PAcc
ΔSeg
Mean ↓
Median ↓
11.25∘
22.5∘
30∘
ΔSn
Abs ↓
Rel ↓
1.25
1.25 2
1.25 3
ΔDep
No aggregation
28.46
59.21
-
16.54
13.05
44.11
73.12
83.14
-
0.565
0.223
62.38
88.72
96.71
-
-
FedAvg( J =1)
28.89
59.72
1.19
16.29
12.83
44.80
73.62
83.66
1.21
0.557
0.219
63.30
89.16
97.14
1.12
1.17
max( J =1)
29.31
60.31
2.42
16.30
12.66
45.36
73.78
84.60
1.99
0.570
0.220
61.44
88.81
97.06
− 0.12
1.43
FedAMP( J =1)
29.14
59.77
1.67
16.23
12.74
45.16
73.80
84.67
1.88
0.553
0.215
63.42
89.48
97.32
1.77
1.77
Similarity( J =1)
29.26
59.92
2.01
16.22
12.67
45.33
73.89
84.84
2.14
0.551
0.212
63.49
89.67
97.37
2.19
2.11
TABLE 2: Comparison of Different Aggregation Methods.
Fig. 4 : Task Correlation Heatmap.
Fig. 5 : Convergence behavior of different tasks (Seg, Sn, Dep) over communication rounds.
Fig. 6 : Relative improvement versus aggregation depth J for different task groups.
Fig. 8 : Overall relative improvement with varying network size. The curves show the mean over task-assignment realizations, and the error bars indicate the standard deviation.
Fig. 9 : Effect of regional communication quality on packet loss probability and communication latency. Here, N10 , N20 , and N30 denote networks with 10 , 20 , and 30 clients, respectively. S and L denote the small and large backbone settings, corresponding to ResNet-18 and ResNet-34. The curves show the mean over random topology realizations, and the shaded regions indicate the 10th–90th percentile range.
Decentralized learning systems aim to collaboratively train models across multiple clients without relying on a central coordinator. While decentralization improves scalability, privacy, and robustness, it also exacerbates three fundamental challenges: statistical heterogeneity across clients, fairness in client-level performance, and stringent communication constraints. This raises a natural question: \emph{how fair can decentralized learning be under limited communication?} We address this question by presenting a unified framework for decentralized learning under communication constraints, bringing together graph-based personalization, agnostic fairness, and compressed event-triggered communication. Specifically, we propose a new algorithm DMFL-SQ, a decentralized multi-task learning algorithm that couples personalized model training over a communication graph with an agnostic mixture fairness objective, while reducing communication through sparsification, quantization, and event-triggered synchronization. We establish convergence guarantees for general non-convex objectives and show that DMFL-SQ achieves an O(T−1/2) rate in expected squared Moreau-envelope stationarity despite sparse, quantized, and event-triggered communication. We further derive PAC-Bayes generalization guarantees for the fairness-aware mixture objective. Experiments on CIFAR-10 and the real heterogeneous MUSMET EEG dataset demonstrate that DMFL-SQ substantially reduces communication while maintaining predictive performance and improving fairness across clients. Together, our theoretical and empirical results show that personalization, fairness, and communication efficiency can be jointly achieved in decentralized learning while preserving the dominant convergence rate.
Krishnendu S. Tharakan, Carlo Fischione
School of Electrical Engineering and Computer Science, KTH Royal Institute of Technology, Stockholm, Sweden
Decentralized federated learning (DFL) has emerged as a transformative server-free paradigm that enables collaborative learning over large-scale heterogeneous networks. However, it continues to face fundamental challenges, including data heterogeneity, restrictive assumptions for theoretical analysis, and degraded convergence when standard communication- or privacyenhancing techniques are applied. To overcome these drawbacks, this paper develops a novel algorithm, PaME (DFL by Partial Message Exchange). The central principle is to allow only randomly selected sparse coordinates to be exchanged between two neighbor nodes. Consequently, PaME achieves substantial reductions in communication costs while still preserving a high level of privacy, without sacrificing accuracy. Moreover, grounded in rigorous analysis, the algorithm is shown to converge at a linear rate under the gradient to be locally Lipschitz continuous and the communication matrix to be doubly stochastic. These two mild assumptions not only dispense with many restrictive conditions commonly imposed by existing DFL methods but also enables PaME to effectively address data heterogeneity. Furthermore, comprehensive numerical experiments demonstrate its superior performance compared with several representative decentralized learning algorithms.
Shan Sha, Shenglong Zhou, Xin Wang +2
School of Mathematics and Statistics, Beijing Jiaotong University, Beijing, China · Department of Electrical and Electronic Engineering, Faculty of Engineering, Imperial College London, London, U.K.
Decentralized intelligence systems with heterogeneous devices and limited coordination increasingly rely on decentralized federated learning (DFL). However, DFL suffers from convergence inefficiency under data heterogeneity due to the use of a uniform learning rate (LR) that ignores layer-specific optimization needs. Foundational layers are responsible for maintaining network consensus, while specialized layers adapt to local data characteristics, leading to conflicting gradients and degraded performance under non-IID conditions. To address this fundamental tension, this work introduces FedA2L, a method that dynamically adjusts layer-wise LRs based on model divergence signals. By leveraging local update intensity and network consensus constraints, FedA2L seamlessly integrates into existing DFL protocols without additional communication or coordination. Extensive evaluations across DFL algorithms, various model architectures, and datasets demonstrate that FedA2L achieves up to 4.94 times faster convergence than vanilla DFL and reduces communication rounds by up to 59% compared to scheduler-based baselines. Furthermore, FedA2L exhibits resilience to severe data heterogeneity, larger network sizes, and sparse topologies, reducing communication overhead and establishing it as a versatile optimization tool for resource-constrained or large-scale distributed learning in edge and IoT deployments. The code is released at https://github.com/nclabteam/FedA2L.
Van Truong Vo, Khoa Nguyen, Taehong Kim
School of Information and Communication Engineering, Chungbuk National University, Cheongju, South Korea