计算机工程

作者：顾永跟, 李国笑, 吴小红, 陶杰, 张艳琼

单位：1. 湖州师范学院信息工程学院, 浙江 湖州 313000;2. 湖州学院电子信息学院, 浙江 湖州 313000

关键词：联邦学习,多任务,逆向拍卖,激励机制,数据质量

基金：浙江省现代农业资源智慧管理与应用研究重点实验室项目(2020E10017);湖州师范学院研究生科研创新项目(2023KYCX39)。

联邦学习是一种实现数据隐私保护的分布式机器学习范式,性能取决于数据源的质量和数据规模。客户端是理性个体,参与联邦学习将耗费计算、通信和隐私等成本,需要通过激励提高客户端的参与意愿。因此联邦学习能成功应用的关键之一是尽可能多地激励高质量数据客户端参与训练。多任务联邦学习环境下客户端拥有面向不同任务且质量不同的数据,并具有执行能力的约束。为提高多个学习任务的整体性能,在预算受限的条件下设计一种面向任务的客户选择和报酬机制。通过分析影响模型精度的重要因素,提出一种基于客户端数据样本分布特征的质量评估标准,并结合客户端成本信息,设计一种逆向拍卖的激励机制(EMD-MQMFL),实现客户端的任务指派和支付策略。从理论上分析和证明了该机制具有诚实性、个人理性以及预算可行性,并通过大量实验验证了该方法在联邦学习性能上的有效性。在MNIST、Fashion-MNIST、Cifar-10数据集上的实验结果表明, EMD-MQMFL在数据不平衡的情况下,平均模型精度比已有的机制至少提高5.6个百分点。

来源：2024年第5期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第5期