Anti-Jamming Strategy for Federated Learning in Internet of Medical Things: A Game Approach.

Abstract

Federated learning (FL) is a new dawn of artificial intelligence (AI), in which machine learning models are constructed in a distributed manner while communicating only model parameters between a centralized aggregator and client internet-of-medical-things (IoMT) nodes. The performance of such a learning technique can be seriously hampered by the activities of a malicious jammer robot. In this paper, we study client selection and channel allocation along with the power control problem of the uplink FL process in IoMT domain under the presence of a jammer from the perspective of long-term learning duration. We map the interaction between the FL network and the jammer in each learning iteration as a Stackelberg game, in which the jammer acts as the leader and the FL network serves as the follower. We consider the client and channel selection as well as the power control jointly as the strategy of this game. Upon formulating the game, we find the joint best response strategy for both types of players by leveraging the difference of convex (DC) programming approach and the dual decomposition technique. Beside the availability of the complete information to both the players, we also study the problem from the perspective that the FL network knows the partial information of the other player. Extensive simulations have been conducted to verify the effectiveness of the proposed algorithms in the jamming game.