摘要
arXiv:2504.08872v1 交叉
摘要:在层次联邦学习(HFL)中,在物联网设备与云服务器之间放置边缘网络可以提高通信效率,同时不损害数据隐私。然而,连接到同一边缘的设备通常具有地理或上下文相似性,这导致边缘层面的数据异质性多样化,每个边缘具有不同的标签子集,这意味着在设备层面异质性之上存在边缘层面的非独立且非同分布(non-IID)性质。这种层次上的非-IID 性质暗示每个边缘有自己的优化目标,在 HFL 研究中已被忽略。因此,现有的边缘容纳型 HFL 在各种层次非-IID 场景下的边缘性能不一致。为了确保在具有多样化边缘层面非-IID 数据时具有稳健的性能,我们提出了一种个性化层次边缘增强联邦学习(PHE-FL),该方法将每个边缘模型个性化,使其在每个边缘独特的类分布上表现良好。我们在四种不同层级非-IID 性的场景中评估了 PHE-FL,其中包括极端的物联网设备层非-IID 性。为了准确评估个性化方法的有效性,我们在每个边缘服务器上而不是云服务器上部署了测试集,并使用了平衡和不平衡的测试集。大量的实验表明,在相同的训练轮次下,与包含边缘网络的现有联邦学习方法相比,PHE-FL 的准确性最高可提高 83%。此外,PHE-FL 表现出改进的稳定性,这体现在与两层(边缘和云)聚合的最新 FedAvg 相比,其准确性的波动减少。