Mitigating Data Reconstruction Attacks in Federated Learning: Conditions, Bounds, and Theoretical Guarantees
Organizations: Huazhong University of Science and Technology
Abstract
Privacy attacks, particularly data reconstruction attacks, pose significant challenges to federated learning despite its promise in privacy-preserving collaborative training. This paper presents a comprehensive theoretical analysis of privacy leakage in federated learning through dual perspectives: linear algebra and optimization theory. By analyzing the client-update Jacobian, we identify conditions under which distinct data batches produce the same model update and derive a sufficient batch-size criterion for non-unique reconstruction. Through optimization theory, we establish a novel upper bound for privacy leakage that incorporates batch size, distortion extent, and other critical parameters. These theoretical insights elucidate the fundamental relationships between system parameters and privacy vulnerability, laying the groundwork for developing more robust privacy-preserving federated learning frameworks.
Figures & tables
| Notation | Description |
| Number of clients | |
| Local dataset of client | |
| Number of data samples in | |
| The -th data sample and label of client | |
| Global model at round | |
| Local model of client at round |
| Batchsize | Val Acc (U ) | Test Acc (U ) | Stage | Test MSE (P ) | PSNR (P ) | SSIM (P ) | LPIPS (P ) | KL (P ) |
| 8 | E1 | |||||||
| 16 | E1 | |||||||
| 32 | E1 | |||||||
| 64 | E1 | |||||||
| 128 | E1 |
| Batchsize | Val Acc (U ) | Test Acc (U ) | Stage | Test MSE (P ) | PSNR (P ) | SSIM (P ) | LPIPS (P ) | KL (P ) |
| 8 | E1 | |||||||
| 16 | E1 | |||||||
| 32 | E1 | |||||||
| 64 | E1 | |||||||
| 128 | E1 |
| Batchsize | Val Acc (U ) | Test Acc (U ) | Stage | Test MSE (P ) | PSNR (P ) | SSIM (P ) | LPIPS (P ) | KL (P ) |
| 8 | E1 | |||||||
| 16 | E1 | |||||||
| 32 | E1 | |||||||
| 64 | E1 | |||||||
| 128 | E1 |
| Batchsize | Val Acc (U ) | Test Acc (U ) | Stage | Test MSE (P ) | PSNR (P ) | SSIM (P ) | LPIPS (P ) | KL (P ) |
| 8 | E1 | |||||||
| 16 | E1 | |||||||
| 32 | E1 | |||||||
| 64 | E1 | |||||||
| 128 | E1 |
| Batchsize | Val Acc (U ) | Test Acc (U ) | Stage | Test MSE (P ) | PSNR (P ) | SSIM (P ) | LPIPS (P ) | KL (P ) |
| 8 | E1 | |||||||
| 16 | E1 | |||||||
| 32 | E1 | |||||||
| 64 | E1 | |||||||
| 128 | E1 |
| Batch Size ( ) | Accuracy (%) | (%) | Empty Sets (/60) |
| 1 | 92.28 | 0 | |
| 2 | 94.12 | 58.45 | 4 |
| 4 | 94.08 | 18.12 | 27 |
| 8 | 93.98 |
| Model | Dataset | Method | R-1 | R-2 | Utility | R-1 | R-2 | Utility | R-1 | R-2 | Utility | R-1 | R-2 | Utility |
| GPT-2 | CoLA | TAG | 7.0 2.5 | 0.54 0.54 | 32.5 1.8 | 8.0 2.0 | 1.4 1.3 | 35.1 1.5 | 7.8 1.2 | 0.8 0.5 | 34.2 1.6 | 5.3 0.7 | 0.4 0.2 | 36.8 1.4 |
| LAMP | 73.3 4.5 | 43.3 7.0 | 32.5 1.8 | 26.8 2.8 | 11.0 3.0 | 35.1 1.5 | 13.4 1.4 | 3.9 1.2 | 34.2 1.6 | 8.9 1.2 | 1.9 0.6 | 36.8 1.4 | ||
| DAGER | 100.0 0.0 | 100.0 0.0 | 32.5 1.8 | 100.0 0.0 | 100.0 0.0 | 35.1 1.5 | 100.0 0.0 | 100.0 0.0 | 34.2 1.6 | 100.0 0.0 | 100.0 0.0 | 36.8 1.4 | ||
| SST-2 | TAG | 5.3 0.5 | 0.0 0.0 | 82.4 1.1 | 6.0 1.7 | 0.5 0.4 | 84.1 0.9 | 6.1 1.2 | 0.6 0.6 | 83.6 1.3 | 4.4 0.6 | 0.2 0.1 | 86.8 0.8 | |
| LAMP | 62.2 6.9 | 31.8 8.4 | 82.4 1.1 | 21.4 3.1 | 9.2 3.1 | 84.1 0.9 | 9.8 2.0 | 2.7 1.3 | 83.6 1.3 | 8.1 1.1 | 0.7 0.4 | 86.8 0.8 | ||
| Dataset | Model | R-1 | R-2 | Utility | R-1 | R-2 | Utility | R-1 | R-2 | Utility | R-1 | R-2 | Utility |
| CoLA | GPT-2 | 100.0 0.0 | 100.0 0.0 | 36.8 1.1 | 100.0 0.0 | 100.0 0.0 | 36.2 1.3 | 100.0 0.0 | 100.0 0.0 | 37.9 0.9 | 30.3 1.0 | 14.6 0.9 | 37.4 1.0 |
| LLaMA-2 (7B) | 100.0 0.0 | 100.0 0.0 | 39.2 0.8 | 100.0 0.0 | 100.0 0.0 | 40.5 0.7 | 99.9 0.1 | 99.9 0.1 | 39.8 0.9 | 99.5 0.2 | 99.3 0.3 | 41.2 0.6 | |
| SST-2 | GPT-2 | 100.0 0.0 | 94.6 1.1 | 90.5 0.8 | 100.0 0.1 | 93.4 1.0 | 89.8 0.9 | 92.9 3.0 | 85.0 3.5 | 91.6 0.6 | 13.7 1.4 | 4.3 0.5 | 91.1 0.7 |
| LLaMA-2 (7B) | 100.0 0.0 | 100.0 0.0 | 95.1 0.5 | 99.9 0.1 | 99.9 0.1 | 94.6 0.6 | 99.9 0.1 | 99.9 0.1 | 95.8 0.4 | 98.2 0.4 | 97.8 0.4 | 95.3 0.5 | |
| RT | GPT-2 | 100.0 0.0 | 99.9 0.1 | 84.5 0.9 | 98.0 1.7 | 97.8 1.8 | 83.9 1.0 | 2.8 1.1 | 1.1 0.4 | 85.4 0.7 | 0.0 0.0 | 0.0 0.0 | 84.8 0.8 |
| Model | R-1 | R-2 | Utility | R-1 | R-2 | Utility | R-1 | R-2 | Utility | R-1 | R-2 | Utility |
| GPT-2 BASE | 100.0 0.0 | 99.9 0.1 | 84.2 0.5 | 98.0 1.7 | 97.8 1.8 | 85.1 0.4 | 2.8 1.1 | 1.1 0.4 | 85.8 0.4 | 0.0 0.0 | 0.0 0.0 | 86.2 0.3 |
| GPT-2 FineTuned | 100.0 0.0 | 99.8 0.1 | 86.5 0.4 | 96.4 2.3 | 96.0 2.5 | 87.2 0.4 | 0.84 0.6 | 0.2 0.1 | 87.8 0.3 | 0.0 0.0 | 0.0 0.0 | 88.0 0.3 |
| GPT-2 NextToken | 99.9 0.0 | 99.7 0.1 | 82.0 0.6 | 99.6 0.2 | 99.4 0.2 | 83.1 0.5 | 2.3 0.8 | 0.5 0.2 | 84.0 0.5 | 0.0 0.0 | 0.0 0.0 | 84.5 0.4 |
| GPT-2 LARGE | 100.0 0.0 | 99.8 0.1 | 87.1 0.3 | 100.0 0.0 | 99.9 0.1 | 88.0 0.3 | 44.1 4.2 | 38.1 4.7 | 88.7 0.3 | 0.0 0.0 | 0.0 0.0 | 89.1 0.2 |
| Batch Size ( ) | Token Recovery | R1+R2 | Accuracy | Macro-F1 |
| 1 | ||||
| 2 | ||||
| 4 | ||||
| 8 |
| Batch Size | 8 | 16 | 32 | 48 | 64 |
| Patches Recovered (%) | 90.0 | 84.7 | 73.2 | 66.5 | 62.6 |
| Utility (Acc %) | 78.4 0.8 | 79.6 0.6 | 80.3 0.5 | 79.9 0.7 | 80.8 0.4 |
| Enron Emails | PersonaChat | SQuAD v2 | |||||||||||||||||
| PEFT | Attack | Utility | Utility | Utility | Utility | Utility | Utility | Utility | Utility | Utility | |||||||||
| Full-FT | Phish | 10.12 | 71.5 | 10.66 | 76.8 | 10.89 | 76.1 | 9.13 | 63.8 | 9.36 | 68.5 | 9.38 | 68.1 | 4.26 | 74.3 | 4.18 | 79.6 | 4.24 | 79.0 |
| ReCIT | 54.53 | 70.8 | 41.21 | 76.1 | 30.07 | 75.4 | 49.58 | 63.1 | 35.79 | 67.8 | 27.43 | 67.4 | 47.23 | 73.6 | 33.84 | 78.9 | 23.54 | 78.3 | |
| LoRA | Phish | 7.91 | 70.8 | 7.36 | 76.0 | 7.45 | 75.3 | 7.31 | 63.0 | 6.96 | 67.6 | 7.65 | 67.2 | 2.91 | 73.5 | 2.76 | 78.8 | 2.85 | 78.1 |
| ReCIT | 48.27 | 70.1 | 35.87 | 75.4 | 29.42 | 74.7 | 44.21 | 62.4 | 33.85 | 67.1 | 26.62 | 66.7 | 45.71 | 72.8 | 33.16 | 78.1 | 25.21 | 77.5 | |
| Attack | Utility | Utility | Utility | Utility | Utility | Utility | ||||||
| Phish | 2.91 | 73.5 | 2.53 | 68.2 | 1.92 | 64.1 | 2.76 | 78.8 | 2.37 | 74.5 | 1.84 | 70.3 |
| ReCIT | 45.71 | 72.8 | 28.53 | 67.6 | 24.86 | 63.5 | 29.41 | 78.1 | 17.43 | 73.9 | 11.23 | 69.8 |
| CoLA | SST-2 | RT | |||||||
| Gradient Exposure | |||||||||
| LAMP ( ) | 80.0 | 58.9 | 43.8 | 85.6 | 69.8 | 59.5 | 34.6 | 22.7 | 19.0 |
| 80.4 | 66.3 | 46.5 | 87.1 | 71.8 | 57.7 | 35.9 | 25.4 | 19.6 | |
| 58.7 | 48.6 | 38.9 | 82.1 | 54.5 | 42.2 | 34.8 | 20.2 | 19.0 | |
| 65.0 | 40.7 | 35.9 | 79.7 | 55.8 | 40.1 | 36.7 | 23.0 | 20.2 | |
| 57.0 | 41.8 | 33.9 | 73.9 | 50.9 | 47.5 | 25.5 | 21.3 | 17.7 | |
Federated Learning
Federated learning trains shared models across data holders without requiring them to pool their raw data.
See all 921 papers