Knowledge Caching for Federated Learning in Wireless Cellular Networks

Xin Ying Zheng, Ming Chun Lee*, Kai Chieh Hsu, Y. W.Peter Hong

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

This work examines a novel wireless knowledge caching framework where machine learning models (i.e., knowledge) are cached at local small cell base-stations (SBSs) to facilitate both federated training and access of the models by users. We first consider a single-SBS scenario, where the caching decision, user selection, and wireless resource allocation are jointly determined by minimizing a training error bound subject to constraints on the cache capacity, the communication and computation latency, and the energy consumption. The solution is obtained by first computing the minimum achievable training loss for each model, followed by the optimization of the binary caching variables, which reduces to a 0-1 knapsack problem. The proposed framework is then extended to the multiple-SBS scenario where the user association among SBSs is further examined. We adopt a dual-ascent method where Lagrange multipliers are introduced and updated in each iteration to regularize the dependence among user selection and association. Given the Lagrange multipliers, the caching decision, user selection, resource allocation and user association variables are optimized in turn using a block coordinate descent algorithm. Simulation results show that the proposed scheme can achieve a training error bound that is lower than preference-only and random caching policies in both scenarios.

Original languageEnglish
Pages (from-to)9235-9250
Number of pages16
JournalIEEE Transactions on Wireless Communications
Volume23
Issue number8
DOIs
StatePublished - 2024

Keywords

  • Wireless caching networks
  • federated learning
  • multi-cell multi-user networks
  • resource optimization

Fingerprint

Dive into the research topics of 'Knowledge Caching for Federated Learning in Wireless Cellular Networks'. Together they form a unique fingerprint.

Cite this