首页|期刊导航|南京邮电大学学报(自然科学版)|基于深度强化学习的联邦学习客户端自适应选择策略

基于深度强化学习的联邦学习客户端自适应选择策略OA

Adaptive selection for clients in federated learning based on deep reinforcement learning

中文摘要英文摘要

联邦学习作为解决数据隔离问题的新兴范式,能够在不需要客户端上传原始数据的情况下训练全局模型,有效保护用户隐私.由于客户端数量众多但通信资源有限,只能选择部分客户端参与模型聚合.然而联邦学习系统存在设备异构和数据异质等挑战,简单的客户端选择策略无法考虑环境的动态特性,会拖慢模型的收敛速度,降低模型性能.考虑到客户端状态的时变,提出了全新的客户端可用性评估指标,建立了多重约束下的联邦学习客户端选择模型,建模为损失最小化问题;将优化问题转化为马尔可夫决策过程,提出了一种基于深度强化学习的联邦学习客户端自适应选择(Adaptive Selection for Clients in Federated Learning based on Deep Reinforcement Learn-ing,ASC-DRL)算法,综合考虑通信延迟、资源消耗及客户端可用性,通过代理服务器与环境之间的持续交互最大化奖励函数,得到最优客户端选择方案.实验结果表明,提出的ASC-DRL算法相比于传统联邦学习算法,在模型精度和训练损失方面有着最高89.2%和99.8%的效果提升,能够有效适应动态环境变化,提升联邦学习整体性能和稳定性.

Federated learning(FL)is a distributed paradigm for addressing data isolation by enabling global model training without clients sharing raw data,thereby protecting user privacy.Due to the large number of clients and limited communication resources,only a subset of clients can participate in model aggregation.Nonetheless,FL systems still face challenges such as device heterogeneity and data hetero-geneity.Naive client selection strategies fail to adapt to environmental dynamics,resulting in slow model convergence and degraded performance.To address these,this paper proposes a novel client availability metric by considering time-varying client states,and formulates a multi-constrained client selection model as a loss minimization problem.Then,this problem is further modeled as a Markov decision pro-cess,and a deep reinforcement learning-based adaptive client selection algorithm(ASC-DRL)is de-signed.ASC-DRL comprehensively optimizes communication latency,resource consumption,and client availability through continuous agent-environment interactions,maximizing a reward function,to obtain the optimal client selection scheme.Experiment results demonstrate that ASC-DRL improves model accu-racy by 89.2%and reduces training loss by 99.8%compared to traditional methods,while adaptively en-hancing the performance and stability of FL in dynamic environments.

孙洪波;王国成;张林;王晔;郭永安

南京邮电大学智能信息处理与通信技术省高校重点实验室,江苏 南京 210003||边缘智能研究院南京有限公司,江苏 南京 210003南京邮电大学智能信息处理与通信技术省高校重点实验室,江苏 南京 210003南京邮电大学智能信息处理与通信技术省高校重点实验室,江苏 南京 210003江苏移动信息系统集成有限公司,江苏 南京 210029南京邮电大学智能信息处理与通信技术省高校重点实验室,江苏 南京 210003

信息技术与安全科学

联邦学习深度强化学习客户端选择自适应选择

federated learning(FL)deep reinforcement learningclient selectionadaptive selection

《南京邮电大学学报(自然科学版)》 2026 (2)

84-93,10

江苏省前沿引领技术基础研究专项(BK20202001)和江苏省创新支撑计划政府间双边创新合作项目(BZ2023018)资助项目

10.14132/j.cnki.1673-5439.2026.02.010

评论