FedRE: A Representation Entanglement Framework for Model-Heterogeneous Federated Learning

View PDF HTML (experimental)

Abstract:Federated learning (FL) enables collaborative training across clients while preserving privacy. While most existing FL methods assume homogeneous model architectures, client heterogeneity in both data and resources makes this assumption impractical, thus motivating model-heterogeneous FL. To address this problem, we propose Federated Representation Entanglement (FedRE), a framework built upon a novel form of client knowledge termed entangled representation. Specifically, each client aggregates its local representations into a single entangled representation using normalized random weights, and then applies the same weights to integrate the corresponding one-hot label encodings into an entangled-label encoding. Both are subsequently uploaded to the server to train a global classifier. During training, each entangled representation is supervised across categories via its entangled-label encoding, while random weights are re-sampled at each round to introduce diversity, alleviating overconfidence in the global classifier and yielding smoother decision boundaries. Moreover, each client uploads a single entangled representation along with its entangled-label encoding, mitigating the risk of representation inversion attacks and reducing communication overhead. Extensive experiments demonstrate that FedRE achieves an effective trade-off among model performance, privacy protection, and communication overhead. The codes are available at this https URL.

Comments: Accepted by CVPR 2026

Subjects:

Machine Learning (cs.LG)

Cite as: arXiv:2511.22265 [cs.LG]

(or arXiv:2511.22265v2 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2511.22265

arXiv-issued DOI via DataCite

Submission history

From: Yuan Yao [view email] [v1] Thu, 27 Nov 2025 09:42:22 UTC (5,254 KB) [v2] Fri, 27 Mar 2026 02:12:09 UTC (5,238 KB)