Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Scissorhands: Scrub Data Influence via Connection Sensitivity in Networks

About

Machine unlearning has become a pivotal task to erase the influence of data from a trained model. It adheres to recent data regulation standards and enhances the privacy and security of machine learning applications. In this work, we present a new machine unlearning approach Scissorhands. Initially, Scissorhands identifies the most pertinent parameters in the given model relative to the forgetting data via connection sensitivity. By reinitializing the most influential top-k percent of these parameters, a trimmed model for erasing the influence of the forgetting data is obtained. Subsequently, Scissorhands fine-tunes the trimmed model with a gradient projection-based approach, seeking parameters that preserve information on the remaining data while discarding information related to the forgetting data. Our experimental results, conducted across image classification and image generation tasks, demonstrate that Scissorhands, showcases competitive performance when compared to existing methods. Source code is available at https://github.com/JingWu321/Scissorhands.

Jing Wu, Mehrtash Harandi• 2024

Related benchmarks

TaskDatasetResultRank
Concept UnlearningUnlearnDiffAtk
UnlearnDiffAtk0.2324
36
Style UnlearningUnlearnCanvas
UA0.9584
25
Art Style UnlearningUnlearnCanvas Van Gogh style
FID44.16
18
Machine UnlearningSequential Unlearning Concepts (T1-T6) Stable Diffusion XL, SD v1.5, SANA
UA (T1)97
15
Adversarial Robustness in Concept ErasingRing-A-Bell K-16, K-38, K-77
K-16 Score0.1158
14
Utility PreservationCOCO
CLIP Score0.309
14
Inappropriate Content ErasingI2P
I2P (%)0.79
14
Adversarial Robustness in Concept ErasingMMA-Diffusion
MMA-Diffusion Score1.7
14
Style ErasingUnlearnCanvas
UA95.84
13
Object UnlearningObject Unlearning
Unlearning Accuracy (UA)80.73
13
Showing 10 of 19 rows

Other info

Follow for update