Dec, 2023

基于复制树的有限数据联邦学习

TL;DR在本研究中,我们提出了一种名为 RepTreeFL 的新型联邦学习框架,通过复制参与客户端的模型架构并扰动其本地数据分布的方式,在有限的数据和少数客户端的情况下,通过聚合多个具有不同数据分布的模型来实现学习。我们利用客户端网络的分层结构以及副本之间的模型差异性,引入了一种基于多样性的树聚合方法,在树状结构中组合副本,并根据模型差异性动态更新聚合权重。实验结果表明,在数据和客户端都有限的情况下,RepTreeFL 在图生成和图像分类(二分类和多分类)的两个任务和两种类型的数据中表现出良好的效果和优势。