A Stackelberg Game-based Wireless Powered Federated Learning

By sharing model parameters instead of raw data to train machine models, Federated Learning (FL) can protect End equipment Workers (EWs)’ data privacy. However, due to energy constraints and selfishness, EWs may not be willing to participate or train slowly, which affects the performance of global FL model. To address these issues, we propose a three-stage Stackelberg game-based wireless powered FL framework to incentivize all players to participate in the system while ensuring the successful completion of FL tasks. Specifically, Base Station (BS) publishes the FL task and wants to obtain a better FL model at a lower cost. EWs train local FL models, and want to get more payment with less energy consumption. When EWs train and upload their local models, Charging Service Provider (CSP) transmits energy to them via Wireless Power Transfer (WPT) while charging fees. In order to obtain the optimal strategy for all participants, we analyze the proposed game problem using the backward induction method. Meanwhile, we prove that the unique Stackelberg equilibrium and Nash equilibrium can be obtained, and we obtain the approximate optimal solution of BS using the subgradient method. Finally, extensive simulations are conducted to evaluate the performance of the proposed method in different scenarios. The results show that the proposed method improves the utility of three parties by an average of 19.09% - 51.86% compared with the benchmark methods.