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 条
  • [1] Federated Multi-Objective Learning
    Yang, Haibo
    Liu, Zhuqing
    Liu, Jia
    Dong, Chaosheng
    Momma, Michinari
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [2] A Multi-Objective Framework for Balancing Fairness and Accuracy in Debiasing Machine Learning Models
    Nagpal, Rashmi
    Khan, Ariba
    Borkar, Mihir
    Gupta, Amar
    MACHINE LEARNING AND KNOWLEDGE EXTRACTION, 2024, 6 (03): : 2130 - 2148
  • [3] Multi-Objective Evolutionary Federated Learning
    Zhu, Hangyu
    Jin, Yaochu
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2020, 31 (04) : 1310 - 1322
  • [4] Federated multi-objective reinforcement learning
    Zhao, Fangyuan
    Ren, Xuebin
    Yang, Shusen
    Zhao, Peng
    Zhang, Rui
    Xu, Xinxin
    INFORMATION SCIENCES, 2023, 624 : 811 - 832
  • [5] Federated Learning Meets Multi-Objective Optimization
    Hu, Zeou
    Shaloudegi, Kiarash
    Zhang, Guojun
    Yu, Yaoliang
    IEEE TRANSACTIONS ON NETWORK SCIENCE AND ENGINEERING, 2022, 9 (04): : 2039 - 2051
  • [6] Multi-Objective Counterfactual Fairness
    Dandl, Susanne
    Pfisterer, Florian
    Bischl, Bernd
    PROCEEDINGS OF THE 2022 GENETIC AND EVOLUTIONARY COMPUTATION CONFERENCE COMPANION, GECCO 2022, 2022, : 328 - 331
  • [7] Multi-objective Optimization of Federated Learning Systems in the Computing Continuum
    Mihai, Bogdan
    Saurabh, Nishant
    Kimovski, Dragi
    FRONTIERS OF ARTIFICIAL INTELLIGENCE, ETHICS, AND MULTIDISCIPLINARY APPLICATIONS, FAIEMA 2023, 2024, : 141 - 154
  • [8] MOREGIN: Multi-Objective Recommendation at the Global and Individual Levels
    Gomez, Elizabeth
    Contreras, David
    Boratto, Ludovico
    Salamo, Maria
    ADVANCES IN INFORMATION RETRIEVAL, ECIR 2024, PT I, 2024, 14608 : 21 - 38
  • [10] Achieving Universal Fairness in Machine Learning: A Multi-objective Optimization Perspective
    Hu, Zirui
    Zhang, Zheng
    Feng, Wenjun
    Liu, Qi
    KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, PT II, KSEM 2024, 2024, 14885 : 164 - 179