arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2109.04094cs.LG
收藏

联邦学习中班级不平衡的实验研究

An Experimental Study of Class Imbalance in Federated Learning

University of Birmingham(伯明翰大学)

C. Xiao, S. Wang

摘要:

联邦学习是一种分布式机器学习范例,它基于客户端的多个本地模型训练全局模型进行预测,同时保留本地数据隐私。类不平衡被认为是降低全局模型性能的因素之一。然而,关于班级不平衡是否以及如何影响整体绩效的研究却很少。联邦学习中的类不平衡比传统的非分布式机器学习中的类不平衡复杂得多,这是由于本地客户端处的类不平衡情况不同。在分布式学习环境中,班级不平衡需要重新定义。在本文中,首先,我们提出了两个新的度量来定义类不平衡-全局类不平衡度(MID)和客户端之间的类不平衡的局部差异(WCS)。然后,我们进行了广泛的实验,分析类不平衡的影响,在各种情况下的全局性能的基础上,我们的定义。我们的研究结果表明,一个更高的MID和一个更大的WCS降低更多的全球模型的性能。此外,WCS被证明是减缓收敛的全局模型的误导优化。

英文摘要:

Federated learning is a distributed machine learning paradigm that trains a global model for prediction based on a number of local models at clients while local data privacy is preserved. Class imbalance is believed to be one of the factors that degrades the global model performance. However, there has been very little research on if and how class imbalance can affect the global performance. class imbalance in federated learning is much more complex than that in traditional non-distributed machine learning, due to different class imbalance situations at local clients. Class imbalance needs to be re-defined in distributed learning environments. In this paper, first, we propose two new metrics to define class imbalance -- the global class imbalance degree (MID) and the local difference of class imbalance among clients (WCS). Then, we conduct extensive experiments to analyze the impact of class imbalance on the global performance in various scenarios based on our definition. Our results show that a higher MID and a larger WCS degrade more the performance of the global model. Besides, WCS is shown to slow down the convergence of the global model by misdirecting the optimization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑