Preserving Privacy and Security in Federated Learning

Truc Nguyen,My T. Thai
DOI: https://doi.org/10.1109/TNET.2023.3302016
2023-08-29
Abstract:Federated learning is known to be vulnerable to both security and privacy issues. Existing research has focused either on preventing poisoning attacks from users or on concealing the local model updates from the server, but not both. However, integrating these two lines of research remains a crucial challenge since they often conflict with one another with respect to the threat model. In this work, we develop a principle framework that offers both privacy guarantees for users and detection against poisoning attacks from them. With a new threat model that includes both an honest-but-curious server and malicious users, we first propose a secure aggregation protocol using homomorphic encryption for the server to combine local model updates in a private manner. Then, a zero-knowledge proof protocol is leveraged to shift the task of detecting attacks in the local models from the server to the users. The key observation here is that the server no longer needs access to the local models for attack detection. Therefore, our framework enables the central server to identify poisoned model updates without violating the privacy guarantees of secure aggregation.
Machine Learning,Cryptography and Security
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to protect user privacy and prevent model - poisoning attacks simultaneously in federated learning. Specifically: 1. **Privacy Protection**: Existing research mainly focuses on protecting the privacy of user data by using secure aggregation protocols to prevent the server from obtaining users' local model updates. However, these methods usually do not allow the server to check each user's model update to prevent malicious users from attacking. 2. **Preventing Model - Poisoning Attacks**: Other research focuses on how to detect and defend against model - poisoning attacks from malicious users. These methods usually require the server to be able to access users' local model updates in order to run defense mechanisms. However, these two methods conflict in the threat model, because the secure aggregation protocol requires that the server cannot obtain any information about individual users' model updates, while detecting model - poisoning attacks requires the server to be able to access these updates. Therefore, combining these two methods to achieve privacy protection and attack defense simultaneously is an important challenge. The paper proposes a new framework, which solves the above problems through the following two key technologies: - **Homomorphic Encryption**: It is used to achieve secure aggregation, enabling the server to aggregate all users' model updates without knowing the specific model updates of each user. - **Zero - Knowledge Proof**: It allows users to run defense mechanisms by themselves and prove to the server that their model updates have not been tampered with without revealing any specific information about the model updates. In this way, this framework can effectively detect and prevent model - poisoning attacks while protecting user privacy.