用二次规划优化模型突变方向,提升异构数据下的联邦学习性能
FedQP: Towards Accurate Federated Learning using Quadratic Programming Guided Mutation
- 通过二次规划约束突变方向,使其更贴近梯度更新路径
- 在多个异构数据集上显著提升全局模型推理准确率
- 适合需要高泛化能力的隐私保护分布式学习场景
由于具备隐私保护优势,联邦学习(FL)被广泛应用于分布式机器学习系统。然而,现有方法因数据异构性导致推理性能低下:不同客户端的局部模型优化方向差异大,难以获得在所有客户端表现良好的通用全局模型。现有基于突变的FL方法虽尝试通过随机突变引导模型进入泛化性好的平坦区域,但随机突变易导致各模型最优方向分散,限制性能提升。本文提出新型突变式联邦学习方法FedQP,利用二次规划策略智能调控突变方向,使模型更倾向于沿梯度更新方向突变,有效引导模型向泛化性好的区域优化。在多个知名数据集上的实验表明,该方法在多种异构数据场景下显著提升了全局模型的推理准确率。
原文摘要 · Abstract (English)
Due to the advantages of privacy-preserving, Federated Learning (FL) is widely used in distributed machine learning systems. However, existing FL methods suffer from low-inference performance caused by data heterogeneity. Specifically, due to heterogeneous data, the optimization directions of different local models vary greatly, making it difficult for the traditional FL method to get a generalized global model that performs well on all clients. As one of the state-of-the-art FL methods, the mutation-based FL method attempts to adopt a stochastic mutation strategy to guide the model training towards a well-generalized area (i.e., flat area in the loss landscape). Specifically, mutation allows the model to shift within the solution space, providing an opportunity to escape areas with poor generalization (i.e., sharp area). However, the stochastic mutation strategy easily results in diverse optimal directions of mutated models, which limits the performance of the existing mutation-based FL method. To achieve higher performance, this paper proposes a novel mutation-based FL approach named FedQP, utilizing a quadratic programming strategy to regulate the mutation directions wisely. By biasing the model mutation towards the direction of gradient update rather than traditional random mutation, FedQP can effectively guide the model to optimize towards a well-generalized area (i.e., flat area). Experiments on multiple well-known datasets show that our quadratic programming-guided mutation strategy effectively improves the inference accuracy of the global model in various heterogeneous data scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。