English

Breaking Secure Aggregation: Label Leakage from Aggregated Gradients in Federated Learning

Cryptography and Security 2024-06-25 v1 Artificial Intelligence

Abstract

Federated Learning (FL) exhibits privacy vulnerabilities under gradient inversion attacks (GIAs), which can extract private information from individual gradients. To enhance privacy, FL incorporates Secure Aggregation (SA) to prevent the server from obtaining individual gradients, thus effectively resisting GIAs. In this paper, we propose a stealthy label inference attack to bypass SA and recover individual clients' private labels. Specifically, we conduct a theoretical analysis of label inference from the aggregated gradients that are exclusively obtained after implementing SA. The analysis results reveal that the inputs (embeddings) and outputs (logits) of the final fully connected layer (FCL) contribute to gradient disaggregation and label restoration. To preset the embeddings and logits of FCL, we craft a fishing model by solely modifying the parameters of a single batch normalization (BN) layer in the original model. Distributing client-specific fishing models, the server can derive the individual gradients regarding the bias of FCL by resolving a linear system with expected embeddings and the aggregated gradients as coefficients. Then the labels of each client can be precisely computed based on preset logits and gradients of FCL's bias. Extensive experiments show that our attack achieves large-scale label recovery with 100\% accuracy on various datasets and model architectures.

Keywords

Cite

@article{arxiv.2406.15731,
  title  = {Breaking Secure Aggregation: Label Leakage from Aggregated Gradients in Federated Learning},
  author = {Zhibo Wang and Zhiwei Chang and Jiahui Hu and Xiaoyi Pang and Jiacheng Du and Yongle Chen and Kui Ren},
  journal= {arXiv preprint arXiv:2406.15731},
  year   = {2024}
}

Comments

10 pages, conference to IEEE INFOCOM 2024

R2 v1 2026-06-28T17:15:43.557Z