A Novel Incentive Mechanism for Federated Learning Over Wireless Communications

被引:3
作者
Wang Y. [1 ]
Zhou Y. [1 ]
Huang P. [1 ]
机构
[1] Central South University, School of Automation, Changsha
来源
IEEE Transactions on Artificial Intelligence | 2024年 / 5卷 / 11期
基金
中国国家自然科学基金;
关键词
Bilevel optimization; federated learning; incentive mechanism; meta-learning; multiagent reinforcement learning;
D O I
10.1109/TAI.2024.3419757
中图分类号
学科分类号
摘要
This article studies a federated learning system over wireless communications, where a parameter server shares a global model trained by distributed devices. Due to limited communication resources, not all devices can participate in the training process. To encourage suitable devices to participate, this article proposes a novel incentive mechanism, where the parameter server assigns rewards to the devices, and the devices make participation decisions to maximize their overall profit based on the obtained rewards and their energy costs. Based on the interaction between the parameter server and the devices, the proposed incentive mechanism is formulated as a bilevel optimization problem (BOP), in which the upper level optimizes reward factors for the parameter server and the lower level makes participation decisions for the devices. Note that each device needs to make an independent participation decision due to limited communication resources and privacy concerns. To solve this BOP, a bilevel optimization approach called BIMFL is proposed. BIMFL adopts multiagent reinforcement learning (MARL) to make independent participation decisions with local information at the lower level, and introduces multiagent meta-reinforcement learning to accelerate the training by incorporating meta-learning into MARL. Moreover, BIMFL utilizes covariance matrix adaptation evolutionary strategy to optimize reward factors at the upper level. The effectiveness of BIMFL is demonstrated on different datasets using multilayer perceptron and convolutional neural networks. © 2024 IEEE.
引用
收藏
页码:5561 / 5574
页数:13
相关论文
共 64 条
[1]  
Curzon J., Kosa T.A., Akalu R., El-Khatib K., Privacy and artificial intelligence, IEEE Trans. Artif. Intell., 2, 2, pp. 96-108, (2021)
[2]  
Arel I., Rose D.C., Karnowski T.P., Deep machine learning-A new frontier in artificial intelligence research [research frontier], IEEE Comput. Intell. Mag., 5, 4, pp. 13-18, (2010)
[3]  
Chang J., Lan Z., Cheng C., Wei Y., Data uncertainty learning in face recognition, Proc. IEEE/CVF Conf. Comput. Vis. Pattern Recognit. (CVPR), pp. 5710-5719, (2020)
[4]  
Yao X., Zhu Z., Kang C., Wang S.-H., Gorriz J.M., Zhang Y.-D., AdaD-FNN for chest CT-based COVID-19 diagnosis, IEEE Trans. Emerg. Topics Comput. Intell., 7, 1, pp. 5-14, (2023)
[5]  
Zhong G., Ding W., Chen L., Wang Y., Yu Y.-F., Multi-scale attention generative adversarial network for medical image enhancement, IEEE Trans. Emerg. Topics Comput. Intell., 7, 4, pp. 1113-1125, (2023)
[6]  
Chen M., Shlezinger N., Poor H.V., Eldar Y.C., Cui S., Communication-efficient federated learning, Proc. Nat. Acad. Sci., 118, 17, (2021)
[7]  
Wang M., Guo J., Jia W., Federated multi-phase curriculum learning to synchronously correlate user heterogeneity, IEEE Trans. Artif. Intell., 5, 5, pp. 2026-2039, (2024)
[8]  
Huang K., Liu X., Li F., Yang C., Kaynak O., Huang T., A federated dictionary learning method for process monitoring with industrial applications, IEEE Trans. Artif. Intell., 4, 5, pp. 1017-1028, (2023)
[9]  
Wang S., Et al., Adaptive federated learning in resource constrained edge computing systems, IEEE J. Sel. Areas in Commun., 37, 6, pp. 1205-1221, (2019)
[10]  
Zhu H., Wang R., Jin Y., Liang K., PIVODL: Privacy-preserving vertical federated learning over distributed labels, IEEE Trans. Artif. Intell., 4, 5, pp. 988-1001, (2023)