Federated learning on non-IID and globally long-tailed data via meta re-weighting networks

Yang Lu,Pinxin Qian,Shanshan Yan,Gang Huang,Yuan Yan Tang
DOI: https://doi.org/10.1142/s0219691323500637
2024-01-06
International Journal of Wavelets Multiresolution and Information Processing
Abstract:International Journal of Wavelets, Multiresolution and Information Processing, Ahead of Print. Under a federated learning environment, the training samples are generally collected and stored locally on each client's device, which makes the machine learning procedure not meet the requirement of independent and identical distribution (IID). Existing federated learning methods to deal with non-IID data generally assume that the data is globally balanced. However, real-world multi-class data tend to exhibit long-tail distribution, where the majority of samples are in a few head classes and a large number of tail classes only have a small amount of data. This paper, therefore, focuses on addressing the problem of handling non-IID and globally long-tailed data in a federated learning scenario. Accordingly, we propose a new federated learning method called Federated meta re-weighting networks (FedReN), which assigns weights during the local training process from the class-level and instance-level perspectives, respectively. To deal with data non-IIDness and global long-tail, both of the two re-weighting functions are globally trained by the meta-learning approach to acquire the knowledge of global long-tail distribution. Experiments on several long-tailed image classification benchmarks show that FedReN outperforms the state-of-the-art federated learning methods. The code is available at https://github.com/pxqian/FedReN.
computer science, software engineering,mathematics, interdisciplinary applications
What problem does this paper attempt to address?