mlsanityData to 7 Oct 2026

RefProtoFL: Communication-Efficient Federated Learning via External-Referenced Prototype Alignment

arXiv 2601.14746 ↗First posted 21 Jan 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Federated learning (FL) enables collaborative model training without sharing raw data in edge environments, but is constrained by limited communication bandwidth and heterogeneous client data distributions. Prototype-based FL mitigates this issue by exchanging class-wise feature prototypes instead of full model parameters; however, existing methods still suffer from suboptimal generalization under severe communication constraints. In this paper, we propose RefProtoFL, a communication-efficient FL framework that integrates External-Referenced Prototype Alignment (ERPA) for representation consistency with Adaptive Probabilistic Update Dropping (APUD) for communication efficiency. Specifically, we decompose the model into a private backbone and a lightweight shared adapter, and restrict federated communication to the adapter parameters only. To further reduce uplink cost, APUD performs magnitude-aware Top-K sparsification, transmitting only the most significant adapter updates for server-side aggregation. To address representation inconsistency across heterogeneous clients, ERPA leverages a small server-held public dataset to construct external reference prototypes that serve as shared semantic anchors. For classes covered by public data, clients directly align local representations to public-induced prototypes, whereas for uncovered classes, alignment relies on server-aggregated global reference prototypes via weighted averaging. Extensive experiments on standard benchmarks demonstrate that RefProtoFL attains higher classification accuracy than state-of-the-art prototype-based FL baselines.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 24 references that are in the map. First posted on arXiv 21 Jan 2026.

Builds on

17 of 24 references.

References · 17

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
  3. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
  4. 2018Federated Optimization in Heterogeneous Networkscited by 2,971
  5. 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
  6. 2021FedProto: Federated Prototype Learning across Heterogeneous Clientscited by 349
  7. 2021Federated Learning on Non-IID Data: A Surveycited by 345
  8. 2022Federated Learning from Pre-Trained Models: A Contrastive Learning Approachcited by 141
  9. 2021FedProc: Prototypical Contrastive Federated Learning on Non-IID datacited by 80
  10. 2023FedGH: Heterogeneous Federated Learning with Generalized Global Headercited by 56
  11. 2022Virtual Homogeneity Learning: Defending against Data Heterogeneity in Federated Learningcited by 52
  12. 2022Federated Learning Under Intermittent Client Availability and Time-Varying Communication Constraintscited by 30
5 more
  1. 2021FEDZIP: A Compression Framework for Communication-Efficient Federated Learningcited by 27
  2. 2024Federated Learning under Periodic Client Participation and Heterogeneous Data: A New Communication-Efficient Algorithm and Analysiscited by 8
  3. 2025FedSA: A Unified Representation Learning via Semantic Anchors for Prototype-based Federated Learningcited by 8
  4. 2025Communication-Efficient Federated Learning by Quantized Variance Reduction for Heterogeneous Wireless Edge Networkscited by 2
  5. 2024A Resource-Adaptive Approach for Federated Learning under Resource-Constrained Environmentscited by 1

Built on by

Not cited yet.