针对异构物联网设备,提出自适应双智能体强化学习框架,提升联邦学习效率与精度。
Heterogeneity-aware Personalized Federated Learning via Adaptive Dual-Agent Reinforcement Learning
- 通过双强化学习智能体动态分配模型大小和训练强度,适配不同设备能力。
- 在多个数据集上将训练时间减少20.9%-40.4%,降低延迟19.0%-48.0%。
- 轻量级统一模型促进知识共享,适合边缘计算、资源受限场景应用。
联邦学习(FL)使多个客户端在不共享本地数据的前提下协同训练模型,适用于异构物联网环境。然而,客户端在模型架构和计算能力上的内在差异常导致精度下降和难以解决的慢节点问题,严重影响训练效果。为此,本文提出一种新型异构感知个性化联邦学习方法HAPFL,基于多层级强化学习机制。HAPFL通过三个策略组件优化训练过程:1)基于强化学习的异构模型分配机制,参数服务器使用近端策略优化(PPO)智能体,根据客户端性能自适应分配合适大小的差异化模型,有效缓解性能差异;2)基于强化学习的训练强度调节方案,参数服务器利用另一个PPO智能体动态调整各客户端的训练强度,进一步提升训练效率并减少慢节点延迟;3)基于知识蒸馏的互学习机制,每个客户端同时部署异构本地模型与同质轻量级模型LiteModel,二者通过知识蒸馏相互学习,统一的LiteModel在聚合与共享全局知识中起关键作用,显著提升个性化本地训练效果。实验结果表明,HAPFL在多个基准数据集上不仅实现高精度,且相比现有方案整体训练时间减少20.9%-40.4%,慢节点延迟降低19.0%-48.0%。
原文摘要 · Abstract (English)
Federated Learning (FL) empowers multiple clients to collaboratively train machine learning models without sharing local data, making it highly applicable in heterogeneous Internet of Things (IoT) environments. However, intrinsic heterogeneity in clients' model architectures and computing capabilities often results in model accuracy loss and the intractable straggler problem, which significantly impairs training effectiveness. To tackle these challenges, this paper proposes a novel Heterogeneity-aware Personalized Federated Learning method, named HAPFL, via multi-level Reinforcement Learning (RL) mechanisms. HAPFL optimizes the training process by incorporating three strategic components: 1) An RL-based heterogeneous model allocation mechanism. The parameter server employs a Proximal Policy Optimization (PPO)-based RL agent to adaptively allocate appropriately sized, differentiated models to clients based on their performance, effectively mitigating performance disparities. 2) An RL-based training intensity adjustment scheme. The parameter server leverages another PPO-based RL agent to dynamically fine-tune the training intensity for each client to further enhance training efficiency and reduce straggling latency. 3) A knowledge distillation-based mutual learning mechanism. Each client deploys both a heterogeneous local model and a homogeneous lightweight model named LiteModel, where these models undergo mutual learning through knowledge distillation. This uniform LiteModel plays a pivotal role in aggregating and sharing global knowledge, significantly enhancing the effectiveness of personalized local training. Experimental results across multiple benchmark datasets demonstrate that HAPFL not only achieves high accuracy but also substantially reduces the overall training time by 20.9%-40.4% and decreases straggling latency by 19.0%-48.0% compared to existing solutions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。