English

Soft-Label Caching and Sharpening for Communication-Efficient Federated Distillation

Machine Learning 2026-02-02 v4

Abstract

Federated Learning (FL) enables collaborative model training across decentralized clients, enhancing privacy by keeping data local. Yet conventional FL, relying on frequent parameter-sharing, suffers from high communication overhead and limited model heterogeneity. Distillation-based FL approaches address these issues by sharing predictions (soft-labels, i.e., normalized probability distributions) instead, but they often involve redundant transmissions across communication rounds, reducing efficiency. We propose SCARLET, a novel framework integrating synchronized soft-label caching and an enhanced Entropy Reduction Aggregation (Enhanced ERA) mechanism. SCARLET minimizes redundant communication by reusing cached soft-labels, achieving up to 50% reduction in communication costs compared to existing methods while maintaining competitive accuracy. Enhanced ERA resolves the fundamental instability of conventional temperature-based aggregation, ensuring robust control and high performance in diverse client scenarios. Experimental evaluations demonstrate that SCARLET consistently outperforms state-of-the-art distillation-based FL methods in terms of accuracy and communication efficiency. The implementation of SCARLET is publicly available at https://github.com/kitsuyaazuma/SCARLET.

Keywords

Cite

@article{arxiv.2504.19602,
  title  = {Soft-Label Caching and Sharpening for Communication-Efficient Federated Distillation},
  author = {Kitsuya Azuma and Takayuki Nishio and Yuichi Kitagawa and Wakako Nakano and Takahito Tanimura},
  journal= {arXiv preprint arXiv:2504.19602},
  year   = {2026}
}

Comments

23 pages, 18 figures. Accepted to IEEE Transactions on Mobile Computing (TMC)

R2 v1 2026-06-28T23:13:28.628Z