Resource optimized federated learning-enabled cognitive internet of things for smart industries

Latif U. Khan, Madyan Alsenwi, Ibrar Yaqoob, Muhammad Imran, Zhu Han, Choong Seon Hong

Research output: Contribution to journalArticlepeer-review

40 Scopus citations

Abstract

Leveraging the cognitive Internet of things (C-IoT), emerging computing technologies, and machine learning schemes for industries can assist in streamlining manufacturing processes, revolutionizing operational analytics, and maintaining factory efficiency. However, further adoption of centralized machine learning in industries seems to be restricted due to data privacy issues. Federated learning has the potential to bring about predictive features in industrial systems without leaking private information. However, its implementation involves key challenges including resource optimization, robustness, and security. In this article, we propose a novel dispersed federated learning (DFL) framework to provide resource optimization, whereby distributed fashion of learning offers robustness. We formulate an integer linear optimization problem to minimize the overall federated learning cost for the DFL framework. To solve the formulated problem, first, we decompose it into two sub-problems: association and resource allocation problem. Second, we relax the association and resource allocation sub-problems to make them convex optimization problems. Later, we use the rounding technique to obtain binary association and resource allocation variables. Our proposed algorithm works in an iterative manner by fixing one problem variable (for example, association) and compute the other (for example, resource allocation). The iterative algorithm continues until convergence of the formulated cost optimization problem. Furthermore, we compare the proposed DFL with two schemes; namely, random resource allocation and random association. Numerical results show the superiority of the proposed DFL scheme.

Original languageBritish English
Pages (from-to)168854-168864
Number of pages11
JournalIEEE Access
Volume8
DOIs
StatePublished - 2020

Keywords

  • Cognitive Internet of Things
  • Convex optimization
  • Federated learning
  • Smart industry

Fingerprint

Dive into the research topics of 'Resource optimized federated learning-enabled cognitive internet of things for smart industries'. Together they form a unique fingerprint.

Cite this