Enhancing Federated Learning in Fog-Aided IoT by CPU Frequency and Wireless Power Control

Jingjing Yao, Nirwan Ansari

Research output: Contribution to journalArticlepeer-review

Abstract

Machine learning models have been built in fog nodes in fog-aided Internet-of-Things (IoT) networks to provision future events prediction and image classification by training data collected from IoT devices. However, sending massive data from all devices to a fog node incurs huge network traffic in wireless links in between. Federated learning is proposed to address the challenge by training models locally in IoT devices and only sharing model parameters in the fog node. In this article, we investigate both the CPU frequency control and wireless transmission power control of all IoT devices to balance the tradeoff between the device energy consumption and federated learning time (consisting of both the computation and communication latencies) in fog-aided IoT networks. We formulate the joint optimization of CPU and power control as a nonlinear programming (NLP) problem with the objective to minimize the energy consumption of all IoT devices constrained by the federated learning time requirement. An alternative direction algorithm, which alternatively optimizes the CPU frequency and wireless transmission power until convergence, is hence designed to solve this problem and its performance is demonstrated via extensive simulations.

Original languageEnglish (US)
Article number9187798
Pages (from-to)3438-3445
Number of pages8
JournalIEEE Internet of Things Journal
Volume8
Issue number5
DOIs
StatePublished - Mar 1 2021

All Science Journal Classification (ASJC) codes

  • Signal Processing
  • Information Systems
  • Hardware and Architecture
  • Computer Science Applications
  • Computer Networks and Communications

Keywords

  • CPU frequency control
  • federated learning
  • fog computing
  • Internet of Things (IoT)
  • power control

Fingerprint Dive into the research topics of 'Enhancing Federated Learning in Fog-Aided IoT by CPU Frequency and Wireless Power Control'. Together they form a unique fingerprint.

Cite this