Forget the Data, Keep the Capability: Switchable Unlearning for Retrieval
Organizations: South Dakota State University · Aalto University · ELLIS Institute Finland · Michigan State University
Abstract
Privacy compliance requires re-identification models to selectively forget individuals while preserving their ability to recognize others. For this, we present MoME, a switchable framework that can adapt pre-trained backbones for privacy compliance, allowing the same model to switch between Full mode for standard inference and Core mode for unlearned inference. MoME splits one backbone into core parameters that preserve retrieval and zero-initialized parasitic parameters that carry the full-data knowledge, and training them under gradient isolation so the core never receives a forget-set gradient. Disabling the branch yields Core, which satisfies a request to delete a person's training data, and enabling it yields Full, which retains what the deleted data taught, so honoring the request is a flag flip rather than a retraining run. We prove Core independent of the forget set up to an audited set of coupling channels and introduce the rank-based Forgetting-Rank statistic to audit that independence in a trained model. On Market-1501, VeRi-776 and RegDB, Core reads at a retrained reference under three membership attacks while retrieving above it, and the retrieval advantage carries to ResNet and CLIP backbones. Code and weights are available at https://github.com/Jingong-Chen/MoME.
Figures & tables
| Market-1501 | VeRi-776 | |||||||||||
| Utility | Privacy | Utility | Privacy | |||||||||
| Method | R-1 | mAP | mAP | MIA | RMIA | U-LiRA | R-1 | mAP | mAP | MIA | RMIA | U-LiRA |
| Original | ||||||||||||
| Fine-tune [ 46 ] | ||||||||||||
| GA [ 44 ] | ||||||||||||
| NegGrad+ [ 26 ] | ||||||||||||
| Method | Utility | Privacy | ||||||
|---|---|---|---|---|---|---|---|---|
| V T R-1 | V T mAP | T V R-1 | T V mAP | mAP | MIA | RMIA | U-LiRA | |
| Original | ||||||||
| Fine-tune [ 46 ] | ||||||||
| GA [ 44 ] | ||||||||
| NegGrad+ [ 26 ] | ||||||||
| SalUn [ 11 ] | ||||||||
| CIFAR-10 | CIFAR-100 | TinyImageNet | |||||||
| Method | retain | MIA | retain | MIA | retain | MIA | |||
| Original | 94.2 | 100.0 | 0.44 | 76.4 | 100.0 | 0.68 | 65.0 | 99.9 | 0.70 |
| Fine-tune | 94.3 | 79.3 | 0.07 | 77.4 | 91.3 | 0.32 | 65.4 | 8.0 | 0.20 |
| GA | 11.1 | 0.0 | 0.50 | 1.1 | 0.0 | 0.50 | 0.6 | 0.0 | 0.50 |
| NegGrad+ | 12.7 | 0.0 | 0.50 | 2.5 | 0.0 | 0.53 | 0.6 | 0.0 | 0.47 |
| SalUn | 94.4 | 10.0 | 0.01 | 76.6 | 63.8 | 0.07 | 65.0 | 95.6 | 0.44 |
| Utility | Privacy, | |||||||
|---|---|---|---|---|---|---|---|---|
| Backbone | Norm | Pretrain | Task | Core | Retr. | Core | Retr. | Full |
| ViT-B/16 | LN | ImageNet | ReID, Mkt | 74.2 | 70.5 | |||
| ResNet-50 AGW | BN | ImageNet | ReID, Mkt | 47.5 | 33.3 | |||
| CLIP ViT-B/16 | LN | contrastive | ReID, Mkt | 77.7 | 67.6 | |||
| ResNet-18 | BN | scratch | CIFAR-10 | 47.8 | 47.8 | |||
| Variant | Core mAP | Full mAP | MIA |
|---|---|---|---|
| w/o Gradient Isolation | 85.1 | 88.3 | 0.995 |
| w/o Step 2 | 74.3 | 74.3 | 0.491 |
| Frozen backbone | 2.4 | 73.7 | 0.506 |
| Parasitic only | 75.4 | 78.1 | 0.487 |
| Parasitic only | 75.4 | 77.8 | 0.487 |
| Ours co-trained | 74.4 | 81.5 | 0.493 |
Appendix figures & tables17 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Rate | test mAP | mAP | RMIA |
|---|---|---|---|---|
| SCRUB | 31.3 | 45.6 | 0.560 | |
| 43.9 | 65.2 | 0.709 | ||
| 85.4 | 99.7 | 0.987 | ||
| NegGrad+ | 28.2 | 31.7 | 0.494 | |
| 86.2 | 99.7 | 0.990 | ||
| 86.4 | 99.7 | 0.990 |
| Row | AUC | int. | difference, int. | |
|---|---|---|---|---|
| Market | Retrain | 0.511 | – | |
| Core-only | 0.511 | |||
| Core s42 | 0.511 | |||
| Core s123 | 0.505 | |||
| Core s456 | 0.513 | |||
| Core s789 | 0.507 |
| teacher refs | disjoint refs | ||||
|---|---|---|---|---|---|
| Row | ent | RMIA | U-LiRA | RMIA | U-LiRA |
| Original | 0.999 | 0.991 | 0.979 | 0.991 | 0.979 |
| Fine-tune | 0.982 | 0.984 | 0.975 | 0.984 | 0.974 |
| Retrain | 0.493 | 0.511 | 0.514 | 0.511 | 0.511 |
| Core-only | 0.491 | – | – | 0.511 | 0.515 |
| Core s42 | 0.492 | 0.510 | 0.503 | 0.511 | 0.513 |
| Statistic | Retrain band | MoME-Core | distance | |
|---|---|---|---|---|
| Market | MIA | |||
| RMIA | ||||
| U-LiRA | ||||
| TPR@ | ||||
| VeRi | MIA | |||
| RMIA |
| Retain set skew | Core | Full | Core MIA, AUC | ||||
|---|---|---|---|---|---|---|---|
| mAP | R-1 | mAP | MIA H | RMIA | U-LiRA | ||
| Uniform , identity-level | 1,491 | 74.3 | 88.2 | 82.0 | 0.492 | 0.511 | 0.513 |
| Camera skew, delete cam-0 | 2,954 | 76.2 | 88.3 | 83.6 | 0.872 | 0.559 | 0.652 |
| Identity skew, delete 50 PIDs | 12,025 | 87.5 | 94.5 | 86.0 | 0.541 | 0.517 | 0.518 |
| Imbalanced | 3,222 | 81.3 | 91.8 | 84.1 | 0.917 | 0.548 | 0.711 |
| Identities named | Core | Retrain | Full | half-width |
|---|---|---|---|---|
| 5 | 0.518 | 0.534 | 0.926 | 0.153 |
| 10 | 0.510 | 0.522 | 0.937 | 0.092 |
| 25 | 0.504 | 0.518 | 0.943 | 0.067 |
| 50 | 0.519 | 0.507 | 0.944 | 0.046 |
| 100 | 0.511 | 0.509 | 0.944 | 0.036 |
| 200 | 0.515 | 0.515 | 0.939 | 0.028 |
| Method | Core | Full | Core MIA, AUC | |||
|---|---|---|---|---|---|---|
| mAP | R-1 | mAP | MIA H | RMIA | U-LiRA | |
| Original, trained on | 88.0 | 94.8 | – | 0.999 | 0.991 | 0.979 |
| Retrained reference: retrain on | 70.5 | 86.9 | – | 0.493 | 0.511 | 0.511 |
| MoME-Core, by design | 74.3 | 88.2 | 82.0 | 0.492 | 0.511 | 0.513 |
| Naive post-hoc, no isolation | 88.5 | 95.2 | 88.8 | 0.997 | 0.967 | 0.909 |
| PH1: freeze backbone | 88.0 | 94.8 | 88.4 | 0.999 | 0.991 | 0.979 |
| Family | sd | p99 | max | above |
|---|---|---|---|---|
| Retrain | 2.68 | 6.30 | 9.63 | 137 |
| MoME-Core | 3.15 | 7.19 | 10.23 | 220 |
| Full vs Core | 1.80 | 10.86 | 12.00 | 1520 |
| Family | leave-one-out | sd | ||||
|---|---|---|---|---|---|---|
| Market retrain | 0.603 | 0.525 | 0.400 | 0.427 | 0.544 | .085 |
| Market Core | 0.538 | 0.480 | 0.486 | 0.501 | 0.495 | .023 |
| VeRi retrain | 0.559 | 0.354 | 0.553 | 0.437 | 0.597 | .101 |
| VeRi Core | 0.487 | 0.522 | 0.440 | 0.544 | 0.507 | .040 |
| RegDB retrain | 0.611 | 0.392 | 0.582 | 0.487 | 0.428 | .095 |
| RegDB Core | 0.494 | 0.539 | 0.520 | 0.509 | 0.437 | .039 |
| Dataset | Method | retained, held out | ||
|---|---|---|---|---|
| Market | Retrain | 99.8 | 86.4 | 77.3 |
| MoME-Core | 100.0 | 90.3 | 80.1 | |
| MoME-Full | 99.8 | 99.8 | 99.8 | |
| VeRi | Retrain | 56.0 | 47.7 | 35.5 |
| MoME-Core | 99.9 | 52.1 | 40.8 | |
| MoME-Full | 100.0 | 100.0 | 100.0 |
| Method | retained, held out | |||
|---|---|---|---|---|
| Original | 0.996 | 0.996 | 0.996 | 0.997 |
| Fine-tune | 0.990 | 0.982 | 0.999 | 0.998 |
| GA | 0.994 | 0.994 | 0.991 | 0.996 |
| Retrain | 0.722 | 0.493 | 1.000 | 0.938 |
| Core | 0.725 | 0.493 | 1.000 | 0.945 |
| Full | 0.998 | 0.998 | 0.997 | 0.999 |
| Setting | Core mAP | Full mAP | MIA on |
|---|---|---|---|
| Default | 74.34 | 81.99 | 0.493 |
| BN freeze | 75.12 | 80.07 | 0.487 |
| RMIA | U-LiRA | ||||
| Method | AUC | TPR@1% | [email protected]% | AUC | TPR@1% |
| Original | 0.99 | 0.784 | 0.358 | 0.98 | 0.356 |
| Fine-tune | 0.98 | 0.805 | 0.485 | 0.97 | 0.525 |
| GA | 0.98 | 0.744 | 0.381 | 0.97 | 0.327 |
| NegGrad+ | 0.99 | 0.773 | 0.400 | 0.98 | 0.342 |
| SalUn | 0.88 | 0.302 | 0.122 | 0.87 | 0.200 |
| Method | Market-1501 | VeRi-776 | RegDB |
|---|---|---|---|
| Original | 0.784 | 0.254 | 0.169 |
| Fine-tune | 0.805 | 0.321 | 0.391 |
| GA | 0.744 | 0.088 | 0.347 |
| NegGrad+ | 0.773 | 0.004 | 0.155 |
| SalUn | 0.302 | 0.061 | 0.293 |
| Bad Teacher | 0.311 | 0.023 | 0.348 |
| Forgetting-Rank | RMIA | ||||||
|---|---|---|---|---|---|---|---|
| Violation | Kept | Identity AUC | Clean at floor | AUC | int. | ||
| 14 | 0.5102 | 0.9995 | 2.0% | 0.531 | |||
| 34 | 0.4793 | 0.9973 | 5.5% | 0.542 | |||
| 68 | 0.4820 | 0.9921 | 7.2% | 0.562 | |||
| 169 | 0.4645 | 0.9940 | 7.3% | 0.635 | |||
Machine Unlearning
Machine unlearning removes the influence of selected data or concepts from trained models while preserving useful behavior and avoiding full retraining.
See all 354 papers