Multi-objective federated learning: Balancing global performance and individual fairness

被引:0
|
作者
Shen, Yuhao [1 ]
Xi, Wei [1 ]
Cai, Yunyun [1 ]
Fan, Yuwei [1 ]
Yang, He [1 ]
Zhao, Jizhong [1 ]
机构
[1] Xi An Jiao Tong Univ, Fac Elect & Informat Engn, Xian 710000, Peoples R China
基金
中国国家自然科学基金;
关键词
Federated learning; Multi-objective optimization; Non-iid data; Gradient conflict; Fairness;
D O I
10.1016/j.future.2024.07.046
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
In federated learning, non-iid data not only diminishes the performance of the global model but also gives rise to the fairness problem which manifests as an increase in the variance of the global model's accuracy across clients. Fairness issues can result in the global model performing poorly or even failing on certain clients. Existing methods addressing the fairness problem in federated learning tend to neglect the comprehensive improvement of both the average performance and fairness of the global model. In addressing it, the multi- objective optimization method for fine-tuning global gradients, FedMC algorithm is introduced in this paper. The primary objective is the average loss function of all clients, and the sub-objective involves fine-tuning the global gradient by reducing the gradient conflict between the global gradient and the local gradients. Specifically, we refine the global gradient by incorporating a sub-optimization objective aimed at alleviating conflicts between the global gradient and the local gradient with the largest deviation, denoted as FedMC. FedMC can enhance the performance and convergence rate of clients with initially poor performance, albeit at the cost of the earlier convergence rate of clients with initially good performance. Nevertheless, it enables the latter to reach the accuracy level achieved before fine-tuning. In addition, we also propose FedMC+ algorithm, owning three additional optimization mechanisms built upon the FedMC optimization objective which includes the decay of hyperparameter, the sliding window mechanism, and data-balanced client selection. Besides, we present a theoretical analysis of the convergence rate of FedMC, demonstrating its convergence to a Pareto stationary solution. Our combined experimental results confirm that FedMC+ achieves an average 4.5% improvement in accuracy and a 22% reduction in the degree of dispersion compared to state-of-the-art federated learning (FL) methods.
引用
下载
收藏
页数:12
相关论文
共 50 条
  • [21] Balancing Learning and Engagement in Game-Based Learning Environments with Multi-objective Reinforcement Learning
    Sawyer, Robert
    Rowe, Jonathan
    Lester, James
    ARTIFICIAL INTELLIGENCE IN EDUCATION, AIED 2017, 2017, 10331 : 323 - 334
  • [22] Learning-based Multi-Objective Resource Allocation for Over-the-Air Federated Learning
    Tu, Xuezhen
    Zhu, Kun
    2022 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM 2022), 2022, : 3065 - 3070
  • [23] ICB FL: Implicit Class Balancing Towards Fairness in Federated Learning
    Li, Yanli
    Zhong, Laicheng
    Yuan, Dong
    Chen, Huaming
    Bao, Wei
    PROCEEDINGS OF 2023 AUSTRALIAN COMPUTER SCIENCE WEEK, ACSW 2023, 2023, : 135 - 142
  • [24] Studying the Influence of the Objective Balancing Parameter in the Performance of a Multi-Objective Ant Colony Optimization Algorithm
    Mora, A. M.
    Merelo, J. J.
    Castillo, P. A.
    Laredo, J. L. J.
    Garcia-Sanchez, P.
    Arenas, M. G.
    NICSO 2010: NATURE INSPIRED COOPERATIVE STRATEGIES FOR OPTIMIZATION, 2010, 284 : 163 - 176
  • [25] Multi-objective optimization in the individual diets
    Nedeva, Cvetana
    Baeva, Silvia
    41ST INTERNATIONAL CONFERENCE APPLICATIONS OF MATHEMATICS IN ENGINEERING AND ECONOMICS (AMEE'15), 2015, 1690
  • [26] A Consensus Algorithm for Multi-Objective Battery Balancing
    Barreras, Jorge Varela
    de Castro, Ricardo
    Wan, Yihao
    Dragicevic, Tomislav
    ENERGIES, 2021, 14 (14)
  • [27] Accuracy and fairness trade-offs in machine learning: a stochastic multi-objective approach
    Suyun Liu
    Luis Nunes Vicente
    Computational Management Science, 2022, 19 : 513 - 537
  • [28] Multi-Objective Optimization for Bandwidth-Limited Federated Learning in Wireless Edge Systems
    Zhou, Yu
    Liu, Xuemei
    Lei, Lei
    IEEE OPEN JOURNAL OF THE COMMUNICATIONS SOCIETY, 2023, 4 : 954 - 966
  • [29] Predictive GAN-Powered Multi-Objective Optimization for Hybrid Federated Split Learning
    Yin, Benshun
    Chen, Zhiyong
    Tao, Meixia
    IEEE TRANSACTIONS ON COMMUNICATIONS, 2023, 71 (08) : 4544 - 4560
  • [30] Towards fairness-aware multi-objective optimization
    Guo Yu
    Lianbo Ma
    Xilu Wang
    Wei Du
    Wenli Du
    Yaochu Jin
    Complex & Intelligent Systems, 2025, 11 (1)