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

Towards LLM Unlearning Resilient to Relearning Attacks: A Sharpness-Aware Minimization Perspective and Beyond

About

The LLM unlearning technique has recently been introduced to comply with data regulations and address the safety and ethical concerns of LLMs by removing the undesired data-model influence. However, state-of-the-art unlearning methods face a critical vulnerability: they are susceptible to ``relearning'' the removed information from a small number of forget data points, known as relearning attacks. In this paper, we systematically investigate how to make unlearned models robust against such attacks. For the first time, we establish a connection between robust unlearning and sharpness-aware minimization (SAM) through a unified robust optimization framework, in an analogy to adversarial training designed to defend against adversarial attacks. Our analysis for SAM reveals that smoothness optimization plays a pivotal role in mitigating relearning attacks. Thus, we further explore diverse smoothing strategies to enhance unlearning robustness. Extensive experiments on benchmark datasets, including WMDP and MUSE, demonstrate that SAM and other smoothness optimization approaches consistently improve the resistance of LLM unlearning to relearning attacks. Notably, smoothness-enhanced unlearning also helps defend against (input-level) jailbreaking attacks, broadening our proposal's impact in robustifying LLM unlearning. Codes are available at https://github.com/OPTML-Group/Unlearn-Smooth.

Chongyu Fan, Jinghan Jia, Yihua Zhang, Anil Ramakrishna, Mingyi Hong, Sijia Liu• 2025

Related benchmarks

TaskDatasetResultRank
Multi-task Language UnderstandingMMLU
Accuracy27.9
353
Machine UnlearningMUSE Books--
83
UnlearningMUSE-News 1.0 (test)
Privacy Leak0.00e+0
55
Harmful score evaluationBeaverTails (test)
Harmful Score0.738
52
Commonsense ReasoningHellaSwag
HS Score42.1
43
Question AnsweringTruthfulQA
TruthfulQA23.6
37
General KnowledgeHellaSwag
Accuracy57
36
Machine UnlearningMUSE-Books (test)
No VerbMem0.00e+0
30
General KnowledgeMMLU
General Score72.1
25
UnlearningMUSE-Books 1.0 (test)
Unlearn Score74.8
24
Showing 10 of 27 rows

Other info

Follow for update