Instruct: Identify the main and secondary category of Arxiv papers based on the titles and abstracts
Query: Federated Learning over Wireless Networks: Convergence Analysis and
  Resource Allocation   There is an increasing interest in a fast-growing machine learning technique
called Federated Learning, in which the model training is distributed over
mobile user equipments (UEs), exploiting UEs' local computation and training
data. Despite its advantages in data privacy-preserving, Federated Learning
(FL) still has challenges in heterogeneity across UEs' data and physical
resources. We first propose a FL algorithm which can handle the heterogeneous
UEs' data challenge without further assumptions except strongly convex and
smooth loss functions. We provide the convergence rate characterizing the
trade-off between local computation rounds of UE to update its local model and
global communication rounds to update the FL global model. We then employ the
proposed FL algorithm in wireless networks as a resource allocation
optimization problem that captures the trade-off between the FL convergence
wall clock time and energy consumption of UEs with heterogeneous computing and
power resources. Even though the wireless resource allocation problem of FL is
non-convex, we exploit this problem's structure to decompose it into three
sub-problems and analyze their closed-form solutions as well as insights to
problem design. Finally, we illustrate the theoretical analysis for the new
algorithm with Tensorflow experiments and extensive numerical results for the
wireless resource allocation sub-problems. The experiment results not only
verify the theoretical convergence but also show that our proposed algorithm
outperforms the vanilla FedAvg algorithm in terms of convergence rate and
testing accuracy.

21
The categories of this paper are: adap-org
11
The categories of this paper are: hep-lat
18
The categories of this paper are: econ
19
The categories of this paper are: eess
7
The categories of this paper are: hep-th
15
The categories of this paper are: nucl-ex
22
The categories of this paper are: chao-dyn
9
The categories of this paper are: nlin
32
The categories of this paper are: atom-ph
36
The categories of this paper are: cmp-lg
14
The categories of this paper are: math-ph
8
The categories of this paper are: hep-ex
20
The categories of this paper are: acc-phys
37
The categories of this paper are: supr-con
17
The categories of this paper are: q-fin
5
The categories of this paper are: gr-qc