Proceedings

Competitive-Cooperative Multi-Agent Reinforcement Learning for Auction-based Federated Learning

Download BibTeX Cite

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. Main Track. Pages 4262-4270.

https://doi.org/10.24963/ijcai.2023/474

Abstract

Auction-based Federated Learning (AFL) enables open collaboration among self-interested data consumers and data owners. Existing AFL approaches cannot manage the mutual influence among multiple data consumers competing to enlist data owners. Moreover, they cannot support a single data owner to join multiple data consumers simultaneously. To bridge these gaps, we propose the Multi-Agent Reinforcement Learning for AFL (MARL-AFL) approach to steer data consumers to bid strategically

towards an equilibrium with desirable overall system characteristics. We design a temperature-based reward reassignment scheme to make tradeoffs between cooperation and competition among AFL data consumers. In this way, it can reach an equilibrium state that ensures individual data consumers can achieve good utility, while preserving system-level social welfare. To circumvent potential collusion behaviors among data consumers, we introduce a bar agent to set a personalized bidding

lower bound for each data consumer. Extensive experiments on six commonly adopted benchmark datasets show that MARL-AFL is significantly more advantageous compared to six state-of-the-art approaches, outperforming the best by 12.2%, 1.9% and 3.4% in terms of social welfare, revenue and accuracy, respectively.

  1. [Bahreini et al., 2018] Tayebeh Bahreini, Hossein Badri, and Daniel Grosu. An envy-free auction mechanism for re- source allocation in edge computing systems. In SEC, pages 313-322, 2018.
  2. [Cai et al., 2017] Han Cai, Kan Ren, Weinan Zhang, Klean- this Malialis, Jun Wang, Yong Yu, and Defeng Guo. Real- time bidding by reinforcement learning in display adver- tising. In WSDM, pages 661-670, 2017.
  3. [Chen et al., 2020] Yiqiang Chen, Xiaodong Yang, Xin Qin, Han Yu, Piu Chan, and Zhiqi Shen. Dealing with la- bel quality disparity in federated learning. In Federated Learning: Privacy and Incentive, pages 108-121. 2020.
  4. [Clanuwat et al., 2018] Tarin Clanuwat, Mikel Bober-Irizar, Asanobu Kitamoto, Alex Lamb, Kazuaki Yamamoto, and David Ha. Deep learning for classical japanese literature. arXiv preprint, page 1812.01718, 2018.
  5. [Cohen et al., 2017] Gregory Cohen, Saeed Afshar, Jonathan Tapson, and Andre Van Schaik. EMNIST: Extending M- NIST to handwritten letters. In IJCNN, pages 2921-2926, 2017.
  6. [Deng et al., 2021] Yongheng Deng, Feng Lyu, Ju Ren, Yi- Chao Chen, Peng Yang, Yuezhi Zhou, and Yaoxue Zhang. Fair: Quality-aware federated learning with precise user incentive and model aggregation. In INFOCOM, 2021.
  7. [Gao et al., 2019] Guoju Gao, Mingjun Xiao, Jie Wu, He Huang, Shengqi Wang, and Guoliang Chen. Auction- based vm allocation for deadline-sensitive tasks in dis- tributed edge cloud. IEEE Transactions on Services Com- puting, 2019.
  8. [Hong et al., 2020] Hsiang-Jen Hong, Wenjun Fan, C Ed- ward Chow, Xiaobo Zhou, and Sang-Yoon Chang. Op- timizing social welfare for task offloading in mobile edge computing. In Networking, pages 524-528, 2020.
  9. [Jiao et al., 2018] Yutao Jiao, Ping Wang, Dusit Niyato, and Zehui Xiong. Social welfare maximization auction in edge computing resource allocation for mobile blockchain. In ICC, pages 1-6. IEEE, 2018.
  10. [Jiao et al., 2019] Yutao Jiao, Ping Wang, Dusit Niyato, and Kongrath Suankaewmanee. Auction mechanism- s in cloud/fog computing resource allocation for public blockchain networks. IEEE Transactions on Parallel and Distributed Systems, 30(9):1975-1989, 2019.
  11. [Jiao et al., 2020] Yutao Jiao, Ping Wang, Dusit Niyato, Bin Lin, and Dong In Kim. Toward an automated auction framework for wireless federated learning services market. IEEE Transactions on Mobile Computing, 20(10):3034- 3048, 2020.
  12. [Khan et al., 2020] Latif U. Khan, Shashi Raj Pandey, N- guyen H. Tran, Walid Saad, Zhu Han, Minh N. H. Nguyen, and Choong Seon Hong. Federated learning for edge net- works: Resource optimization and incentive mechanism. IEEE Communications Magazine, 58(10):88-93, 2020.
  13. [Krishnaraj et al., 2022] Nagappan Krishnaraj, Kiranmai Bellam, B Sivakumar, and Arockiam Daniel. The fu- ture of cloud computing: Blockchain-based decentralized cloud/fog solutions-challenges, opportunities, and stan- dards. Blockchain Security in Cloud Computing, pages 207-226, 2022.
  14. [Le et al., 2020] Tra Huong Thi Le, Nguyen H. Tran, Yan Kyaw Tun, Zhu Han, and Choong Seon Hong. Auc- tion based incentive design for efficient federated learning in cellular wireless networks. In WCNC, pages 1-6, 2020.
  15. [Lee et al., 2012] Kuang-chih Lee, Burkay Orten, Ali Das- dan, and Wentong Li. Estimating conversion rate in dis- play advertising from past erformance data. In KDD, pages 768-776, 2012.
  16. [Littman, 1994] Michael L Littman. Markov games as a framework for multi-agent reinforcement learning. In ICML, pages 157-163, 1994.
  17. [Liu et al., 2020] Yang Liu, Anbu Huang, Yun Luo, He Huang, Youzhi Liu, Yuanyuan Chen, Lican Feng, Tian- jian Chen, Han Yu, and Qiang Yang. FedVision: An on- line visual object detection platform powered by federated learning. In IAAI, pages 13172-13179, 2020.
  18. [Liu et al., 2022] Zelei Liu, Yuanyuan Chen, Yansong Zhao, Han Yu, Yang Liu, Renyi Bao, Jinpeng Jiang, Zaiqing Nie, Qian Xu, and Qiang Yang. Contribution-aware federat- ed learning for smart healthcare. In IAAI, pages 12396- 12404, 2022.
  19. [Lyu et al., 2022] Lingjuan Lyu, Han Yu, Xingjun Ma, Chen Chen, Lichao Sun, Jun Zhao, Qiang Yang, and Philip S. Yu. Privacy and robustness in federated learning: Attacks and defense. IEEE Transactions on Neural Networks and Learning Systems, 2022.
  20. [Perlich et al., 2012] Claudia Perlich, Brian Dalessandro, Rod Hook, Ori Stitelman, Troy Raeder, and Foster Provost. Bid optimizing and inventory scoring in target- ed online advertising. In KDD, pages 804-812, 2012.
  21. [Ren et al., 2017] Kan Ren, Weinan Zhang, Ke Chang, Yifei Rong, Yong Yu, and Jun Wang. Bidding machine: Learn- ing to bid for directly optimizing profits in display adver- tising. IEEE Transactions on Knowledge and Data Engi- neering, 30(4):645-659, 2017.
  22. [Roy et al., 2021] Palash Roy, Sujan Sarker, Md Abdur Raz- zaque, Md Mamun-or Rashid, Mohmmad Mehedi Has- san, and Giancarlo Fortino. Distributed task alloca- tion in mobile device cloud exploiting federated learn- ing and subjective logic. Journal of Systems Architecture, 113(2):doi:10.1016/j.sysarc.2020.101972, 2021. DOI
  23. [Shi et al., 2023] Yuxin Shi, Han Yu, and Cyril Leung. To- wards fairness-aware federated learning. IEEE Transac- tions on Neural Networks and Learning Systems, 2023.
  24. [Simonyan and Zisserman, 2015] Karen Simonyan and An- drew Zisserman. Very deep convolutional networks for large-scale image recognition. In ICLR, 2015.
  25. [Tan et al., 2022] Alysa Ziying Tan, Han Yu, Lizhen Cui, and Qiang Yang. Towards personalized federated learn- ing. IEEE Transactions on Neural Networks and Learning Systems, 2022.
  26. [Thi Le et al., 2021] Tra Huong Thi Le, Nguyen H. Tran, Yan Kyaw Tun, Minh N. H. Nguyen, Shashi Raj Pandey, Zhu Han, and Choong Seon Hong. An incentive mechanis- m for federated learning in wireless cellular networks: An auction approach. IEEE Transactions on Wireless Com- munications, 20(8):4874-4887, 2021.
  27. [Xiao et al., 2017] Han Xiao, Kashif Rasul, and Roland Vollgraf. Fashion-MNIST: A novel image dataset for benchmarking machine learning algorithms. arXiv preprint, page 1708.07747, 2017.
  28. [Yang et al., 2019] Qiang Yang, Yang Liu, Tianjian Chen, and Yongxin Tong. Federated machine learning: Concept and applications. ACM Transactions on Intelligent Systems and Technology, 10(2):12:1-12:19, 2019.
  29. [Yang, 2020] Shi Yang. A task offloading solution for inter- net of vehicles using combination auction matching model based on mobile edge computing. IEEE Access, 8:53261- 53273, 2020.
  30. [Ying et al., 2020] Chenhao Ying, Haiming Jin, Xudong Wang, and Yuan Luo. Double insurance: Incentivized fed- erated learning with differential privacy in mobile crowd- sensing. In SRDS, pages 81-90, 2020.
  31. [Yu et al., 2017] Han Yu, Chunyan Miao, Yiqiang Chen, Si- mon Fauvel, Xiaoming Li, and Victor R. Lesser. Algorith- mic management for improving collective productivity in crowdsourcing. Scientific Reports, 7(12541), 2017.
  32. [Zavodovski et al., 2019] Aleksandr Zavodovski, Suzan Bayhan, Nitinder Mohan, Pengyuan Zhou, Walter Wong, and Jussi Kangasharju. Decloud: Truthful decentral- ized double auction for edge clouds. In ICDCS, pages 2157-2167. IEEE, 2019.
  33. [Zeng et al., 2020] Rongfei Zeng, Shixun Zhang, Jiaqi Wang, and Xiaowen Chu. Fmore: An incentive scheme of multi-dimensional auction for federated learning in MEC. In ICDCS, pages 278-288, 2020.
  34. [Zhan et al., 2020] Yufeng Zhan, Peng Li, Zhihao Qu, Deze Zeng, and Song Guo. A learning-based incentive mecha- nism for federated learning. IEEE Internet of Things Jour- nal, 7(7):6360-6368, 2020.
  35. [Zhan et al., 2021] Yufeng Zhan, Jie Zhang, Zicong Hong, Leijie Wu, Peng Li, and Song Guo. A survey of incentive mechanism design for federated learning. IEEE Trans- actions on Emerging Topics in Computing, 10(2):1035- 1044, 2021.
  36. [Zhang et al., 2014] Weinan Zhang, Shuai Yuan, and Jun Wang. Optimal real-time bidding for display advertising. In KDD, pages 1077-1086, 2014.
  37. [Zhang et al., 2021] Jingwen Zhang, Yuezhou Wu, and Rong Pan. Incentive mechanism for horizontal federated learn- ing based on reputation and reverse auction. In WWW, page 947-956, 2021.
  38. [Zhang et al., 2022a] Jingwen Zhang, Yuezhou Wu, and Rong Pan. Auction-based ex-post-payment incentive mechanism design for horizontal federated learning with reputation and contribution measurement. arXiv preprint, page 2201.02410, 2022.
  39. [Zhang et al., 2022b] Jingwen Zhang, Yuezhou Wu, and Rong Pan. Online auction-based incentive mechanism de- sign for horizontal federated learning with budget con- straint. arXiv preprint, page 2201.09047, 2022.
  40. [Zou et al., 2020] Liang Zou, Xinhui Yu, Ming Li, Meng Lei, and Han Yu. Nondestructive identification of coal and gangue via near-infrared spectroscopy based on improved broad learning. IEEE Transactions on Instrumentation and Measurement, 69(10):8043-8052, 2020.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards