arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.06155cs.LGcs.AIcs.DCcs.SYeess.SY
收藏

面向异构联邦学习的数字孪生辅助知识蒸馏框架

Digital Twin-Assisted Knowledge Distillation Framework for Heterogeneous Federated Learning

Xidian University(西安电子科技大学);Chongqing University of Posts and Telecommunications(重庆邮电大学);Queen’s University(女王大学)

Xiucheng Wang, Nan Cheng, Longfei Ma, Ruijin Sun, Rong Chai, Ning Lu

摘要:

针对联邦学习(FL)系统的异构性,提出了一种知识蒸馏(KD)驱动的FL训练框架,每个用户可以根据需要选择自己的神经网络模型,并使用自己的私有数据集从大型教师模型中提取知识。为了克服在资源有限的用户设备上训练大型教师模型的挑战,开发了数字孪生(DT),使得教师模型可以在位于具有足够计算资源的服务器中的DT上训练。然后,在模型蒸馏期间,每个用户可以在物理实体或数字代理处更新其模型的参数。将模型选择、训练卸载和用户资源分配的联合问题转化为一个混合整数规划问题。为了解决这个问题,Q-学习和优化联合使用,其中Q-学习为用户选择模型并确定是在本地还是在服务器上进行训练,并且优化用于基于Q-学习的输出为用户分配资源。仿真结果表明,提出的DT辅助KD框架和联合优化方法可以显着提高用户的平均准确率,同时降低总延迟。

英文摘要:

In this paper, to deal with the heterogeneity in federated learning (FL) systems, a knowledge distillation (KD) driven training framework for FL is proposed, where each user can select its neural network model on demand and distill knowledge from a big teacher model using its own private dataset. To overcome the challenge of train the big teacher model in resource limited user devices, the digital twin (DT) is exploit in the way that the teacher model can be trained at DT located in the server with enough computing resources. Then, during model distillation, each user can update the parameters of its model at either the physical entity or the digital agent. The joint problem of model selection and training offloading and resource allocation for users is formulated as a mixed integer programming (MIP) problem. To solve the problem, Q-learning and optimization are jointly used, where Q-learning selects models for users and determines whether to train locally or on the server, and optimization is used to allocate resources for users based on the output of Q-learning. Simulation results show the proposed DT-assisted KD framework and joint optimization method can significantly improve the average accuracy of users while reducing the total delay.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑