Learnability and Privacy Vulnerability are Entangled in a Few Critical Weights
Organizations: Department of Computer Science, North Carolina State University, Raleigh, NC 27695, USA
Abstract
Prior approaches for membership privacy preservation usually update or retrain all weights in neural networks, which is costly and can lead to unnecessary utility loss or even more serious misalignment in predictions between training data and non-training data. In this work, we observed three insights: i) privacy vulnerability exists in a very small fraction of weights; ii) however, most of those weights also critically impact utility performance; iii) the importance of weights stems from their locations rather than their values. According to these insights, to preserve privacy, we score critical weights, and instead of discarding those neurons, we rewind only the weights for fine-tuning. We show that, through extensive experiments, this mechanism exhibits outperforming resilience in most cases against Membership Inference Attacks while maintaining utility.
Figures & tables
| Model | Weight Type | PCC | Proportion |
|---|---|---|---|
| ResNet18 | All | 0.8329 | 100.00% |
| Conv | 0.9410 | 99.50% | |
| Linear | 0.8096 | 0.45% | |
| Norm | 0.6776 | 0.05% | |
| ViT | All | 0.7667 | 100.00% |
| Att+MLP | 0.9068 | 99.39% |
| Approach | 0.1% | 1.0% | 3.0% | 5.0% |
|---|---|---|---|---|
| A2 - train | 1.2268 | 0.8570 | 0.4326 | 0.4619 |
| A2 - test | 1.3797 | 1.2728 | 0.9288 | 0.9610 |
| A3 - train | 0.1502 | 0.3376 | 0.4473 | 0.4815 |
| A3 - test | 0.7720 | 0.7433 | 0.8044 | 0.8330 |
| From scratch - train | 0.8087 | |||
| From scratch - test | 1.5398 | |||
| CPU | GPU | RAM | OS | CUDA | Python | PyTorch |
|---|---|---|---|---|---|---|
| AMD Ryzen™ 7 7700X | NVIDIA GeForce RTX 5090 | 64 GB | Ubuntu 24.04 LTS | 12.9 | 3.12.3 | 2.80 |
| Model | Defense | Test Acc. (%, ) | LiRA ( ) | RMIA( ) | ||||
|---|---|---|---|---|---|---|---|---|
| AUC (%) | TPR(%)@FPR | AUC(%) | TPR(%)@FPR | |||||
| 0.1% | 0.1‱ | 0.1% | 0.1‱ | |||||
| ResNet18 | No Defense | 79.44 (0.23) | 85.00 (2.20) | 2.18 (0.59) | 1.78 (0.34) | 74.76 (1.59) | 5.88 (0.70) | 3.90 (1.31) |
| RelaxLoss | 77.10 (1.21) | 70.51 (2.72) | 1.38 (0.42) | 0.52 (0.21) | 66.60 (1.67) | 0.52 (0.34) | 0.12 (0.16) | |
| + CWRF (Ours) | 76.86 (0.29) | 68.31 (0.68) | 0.03 (0.05) | 0.03 (0.05) | 68.18 (1.53) | 1.22 (0.97) | 0.27 (0.19) | |
| HAMP | 77.79 (0.33) | 79.71 (0.20) | 3.33 (0.73) | 1.80 (1.47) | 80.07 (0.58) | 7.28 (1.64) | 1.93 (1.28) | |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Training | Reference |
|---|---|---|
| CIFAR-10 | ||
| CIFAR-100 | ||
| CINIC-10 |
Privacy-Preserving ML
Privacy-preserving machine learning develops methods to learn, share, and run models while limiting exposure of sensitive data.
See all 432 papers