Instruct: Identify the main and secondary category of Arxiv papers based on the titles and abstracts
Query: Particle Swarm Optimized Federated Learning For Industrial IoT and Smart
  City Services   Most of the research on Federated Learning (FL) has focused on analyzing
global optimization, privacy, and communication, with limited attention
focusing on analyzing the critical matter of performing efficient local
training and inference at the edge devices. One of the main challenges for
successful and efficient training and inference on edge devices is the careful
selection of parameters to build local Machine Learning (ML) models. To this
aim, we propose a Particle Swarm Optimization (PSO)-based technique to optimize
the hyperparameter settings for the local ML models in an FL environment. We
evaluate the performance of our proposed technique using two case studies.
First, we consider smart city services and use an experimental transportation
dataset for traffic prediction as a proxy for this setting. Second, we consider
Industrial IoT (IIoT) services and use the real-time telemetry dataset to
predict the probability that a machine will fail shortly due to component
failures. Our experiments indicate that PSO provides an efficient approach for
tuning the hyperparameters of deep Long short-term memory (LSTM) models when
compared to the grid search method. Our experiments illustrate that the number
of clients-server communication rounds to explore the landscape of
configurations to find the near-optimal parameters are greatly reduced (roughly
by two orders of magnitude needing only 2%--4% of the rounds compared to state
of the art non-PSO-based approaches). We also demonstrate that utilizing the
proposed PSO-based technique to find the near-optimal configurations for FL and
centralized learning models does not adversely affect the accuracy of the
models.

23
The categories of this paper are: solv-int
8
The categories of this paper are: hep-ex
33
The categories of this paper are: plasm-ph
32
The categories of this paper are: atom-ph
6
The categories of this paper are: astro-ph
20
The categories of this paper are: acc-phys
29
The categories of this paper are: funct-an
34
The categories of this paper are: chem-ph
14
The categories of this paper are: math-ph
13
The categories of this paper are: quant-ph
25
The categories of this paper are: patt-sol
9
The categories of this paper are: nlin
27
The categories of this paper are: comp-gas
24
The categories of this paper are: dg-ga
30
The categories of this paper are: q-alg
18
The categories of this paper are: econ