1: \begin{abstract}
2: Federated learning is a collaborative model training method that iterates model updates by multiple clients and aggregation of the updates by a central server.
3: Device and statistical heterogeneity of participating clients cause significant performance degradation
4: so that an appropriate aggregation weight should be assigned to each client in the aggregation phase of the server.
5: To adjust the aggregation weights, this paper employs deep unfolding,
6: which is known as the parameter tuning method that leverages both
7: learning capability using training data like deep learning and domain knowledge.
8: This enables us to directly incorporate the heterogeneity of the environment of interest into the tuning of the aggregation weights.
9: The proposed approach can be combined with various federated learning algorithms.
10: % and provide performance improvement.
11: % to obtain the weights that can adapt to the heterogeneity,
12: % which results in the model with high accuracy on uniform test data.
13: The results of numerical experiments indicate that
14: a higher test accuracy for unknown class-balanced data can be obtained with the proposed method
15: than that with conventional heuristic weighting methods.
16: The proposed method can handle large-scale learning models with the aid of pretrained models
17: such that it can perform practical real-world tasks.
18: % The proposed method also yields the learned weights with interpretable behavior.
19: Convergence rate of federated learning algorithms with the proposed method is also provided in this paper.
20: \end{abstract}
21: