Search papers, labs, and topics across Lattice.
This paper introduces Gradient Inversion over Feature Domains (GIFD), a novel gradient inversion attack in federated learning that optimizes over hierarchical feature representations of a GAN to improve reconstruction quality. GIFD progressively optimizes across different layers of the GAN, moving from latent space to intermediate feature spaces, and incorporates a regularizer to ensure realistic image generation. The method is further extended to out-of-distribution settings, including label inconsistency, demonstrating superior performance in reconstructing private data compared to existing attacks.
By optimizing gradient inversions across hierarchical GAN feature spaces, GIFD achieves pixel-perfect reconstructions of private data in federated learning, even in challenging out-of-distribution scenarios.
Federated Learning (FL) has emerged as a compelling paradigm for privacy-preserving distributed machine learning, allowing multiple clients to collaboratively train a global model by transmitting locally computed gradients to a central server without exposing their private data. Nonetheless, recent studies find that the gradients exchanged in the FL system are also vulnerable to privacy leakage, e.g., an attacker can invert shared gradients to reconstruct sensitive data by leveraging pre-trained generative adversarial networks (GAN) as prior knowledge. However, existing attacks simply perform gradient inversion in the latent space of the GAN model, which limits their expression ability and generalizability. To tackle these challenges, we propose \textbf{G}radient \textbf{I}nversion over \textbf{F}eature \textbf{D}omains (GIFD), which disassembles the GAN model and searches the hierarchical features of the intermediate layers. Instead of optimizing only over the initial latent code, we progressively change the optimized layer, from the initial latent space to intermediate layers closer to the output images. In addition, we design a regularizer to avoid unreal image generation by adding a small ${l_1}$ ball constraint to the searching range. We also extend GIFD to the out-of-distribution (OOD) setting, which weakens the assumption that the training sets of GANs and FL tasks obey the same data distribution. Furthermore, we consider the challenging OOD scenario of label inconsistency and propose a label mapping technique as an effective solution. Extensive experiments demonstrate that our method can achieve pixel-level reconstruction and outperform competitive baselines across a variety of FL scenarios.
