Multi-Model based Federated Learning Against Model Poisoning Attack: A Deep Learning Based Model Selection for MEC Systems
<|reference_start|>Multi-Model based Federated Learning Against Model Poisoning Attack: A Deep Learning Based Model Selection for MEC Systems: 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.<|reference_end|>
@article{kianpisheh2024multi-model,
    title={Multi-Model based Federated Learning Against Model Poisoning Attack: A
  Deep Learning Based Model Selection for MEC Systems},
    author={Somayeh Kianpisheh, Chafika Benzaid, Tarik Taleb},
    journal={arXiv preprint arXiv:2409.08237},
    year={2024},
    archivePrefix={arXiv},
    eprint={2409.08237},
    primaryClass={cs.LG cs.NI}
}