Learning to Unlearn: Machine Unlearning via Learning the Unlearning Behaviors
Organizations: State Key Laboratory for Novel Software Technology Nanjing University Nanjing, China · Centre for Cyber Resilience and Trust Deakin University Burwood, VIC, Australia
Abstract
Various machine unlearning techniques have been developed in response to privacy legislation requirements, enabling individuals to exercise their legal right to have their data removed from a machine learning model. This process is typically accomplished via the use of an unlearning function denoted as . Existing methods focus on designing an intricate to unlearn from a previous model , so that the unlearned model performs as closely as possible to the retrained model . However, these methods often suffer from high computational costs when dealing with massive training data, as the complex structures of become a bottleneck even for models with fewer parameters. Inspired by Learning to Optimize, we introduce the first learning-based model-agnostic approach, Learning-to-UnLearn (L2UL). Our core insight is to shift from manually designing to learning the unlearning behaviors from a distribution perspective, thereby acquiring a simple and efficient via learning. Our experimental results demonstrate that the accuracy achieved by L2UL is comparable to that of retraining while exhibiting impressive efficiency, particularly in data-intensive scenarios. Furthermore, we validate the performance and scalability of our method on larger models ResNet.
Figures & tables
| Symbols | Definition |
|---|---|
| Sample space | |
| Label space | |
| Training set | |
| Forget set | |
| Learning algorithm | |
| Unlearning algorithm |
| Magic | Adult | Sepsus | Skin | Covetype | SUSY | HIGGS | |
|---|---|---|---|---|---|---|---|
| n | 19020 | 32562 | 40328 | 245057 | 581012 | 5000000 | 11000000 |
| d | 10 | 14 | 89 | 3 | 54 | 18 | 28 |
| MIA | Accuracy on | ||||||
|---|---|---|---|---|---|---|---|
| Datasets | Retrain | FYEMU | L2UL | Retrain | SISA | FYEMU | L2UL |
| Magic | 0.79 | 0.65 | 0.75 | 0.85 | 0.79 | 0.71 | 0.78 |
| Adult | 0.80 | 0.68 | 0.74 | 0.81 | 0.78 | 0.75 | 0.76 |
| Skin | 1.00 | 1.00 | 1.00 | 1.00 | 1.00 | 1.00 | 1.00 |
| Covetype | 0.73 | 0.68 | 0.67 | 0.82 | 0.76 | 0.76 | 0.71 |
| SUSY | 0.72 | 0.69 | 0.69 | 0.73 | 0.71 | 0.70 | 0.71 |
| -or | -or | -or | L2UL | |
|---|---|---|---|---|
| Factor |
| 10 | 100 | 1000 | 3000 | ||
| ToW( ) | Retrain | 1.00 | 1.00 | 1.00 | 1.00 |
| ADV+IMP | 0.32 | 0.13 | 0.01 | 0.01 | |
| RUM | 0.60 | 0.80 | 0.83 | 0.83 | |
| L2UL(Ours) | 0.80 | 1.00 | 0.88 | 0.99 | |
| MIA gap( ) | Retrain | 0.00 | 0.00 | 0.00 | 0.00 |
| ADV+IMP | 0.30 | 0.13 | 0.17 | 0.14 |
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 349 papers