南京邮电大学学报(自然科学版)

作者：黄宇虹,陈思光

单位：南京邮电大学 物联网学院，江苏 南京 210003南京邮电大学 物联网学院，江苏 南京 210003

关键词：个性化联邦学习；聚类；选择性聚合；非独立同分布

基金：国家自然科学基金（61971235）、江苏省“333高层次人才培养工程”和江苏省研究生科研与实践创新计划项目（KYCX24_1205）资助项目

在客户端数据非独立同分布（Non-Independent and Identically Distributed， Non-IID）的场景中，为了向客户端提供个性化且通信高效的解决方案，提出了一种面向 Non-IID 场景的通信高效个性化联邦学习算法。具体地，为充分利用相似客户端之间的知识提升模型性能，同时保留本地客户端的个性化信息，提出一种融合模型分层思想与聚类思想的个性化联邦学习算法。为进一步解决通信开销高的问题，设计选择性模型聚合策略，在中心服务器通过最大均值差异评估客户端数据分布与全局数据分布的相似性，并基于相似性计算各客户端优先级分数，选择优先级较高的客户端进行通信。该策略可有效减少客户端与中心服务器的累计通信次数，提高通信效率并加速模型收敛。最后，仿真实验结果表明，相较于其他联邦学习算法，所提算法能够在保证高准确率的前提下，将累计通信次数减少至少50%。

来源：2025年第05期

《南京邮电大学学报(自然科学版)》期刊编辑部

查看南京邮电大学学报(自然科学版)杂志2025年第05期