Multi-Model based Federated Learning Against Model Poisoning Attack: A Deep Learning Based Model Selection for MEC Systems

TL;DR

A multi-model based FL is proposed as a proactive mechanism to enhance the opportunity of model poisoning attack mitigation and results illustrate a competitive accuracy gain under poisoning attack with the scenario that the system is without attack.

Abstract

Federated Learning (FL) enables training of a global model from distributed data, while preserving data privacy. However, the singular-model based operation of FL is open with uploading poisoned models compatible with the global model structure and can be exploited as a vulnerability to conduct model poisoning attacks. This paper proposes a multi-model based FL as a proactive mechanism to enhance the opportunity of model poisoning attack mitigation. A master model is trained by a set of slave models. To enhance the opportunity of attack mitigation, the structure of client models dynamically change within learning epochs, and the supporter FL protocol is provided. For a MEC system, the model selection problem is modeled as an optimization to minimize loss and recognition time, while meeting a robustness confidence. In adaption with dynamic network condition, a deep reinforcement learning based model selection is proposed. For a DDoS attack detection scenario, results illustrate a competitive accuracy gain under poisoning attack with the scenario that the system is without attack, and also a potential of recognition time improvement.

Figures (5)

• Figure 1: Multi-model FL architecture.
• Figure 2: (a) Policy networks. (b) Simulation grid.
• Figure 3: Cumulative reward.
• Figure 4: DDoS attack detection accuracy.
• Figure 5: Recognition time.