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

Rethinking LLM Unlearning Objectives: A Gradient Perspective and Go Beyond

About

Large language models (LLMs) should undergo rigorous audits to identify potential risks, such as copyright and privacy infringements. Once these risks emerge, timely updates are crucial to remove undesirable responses, ensuring legal and safe model usage. It has spurred recent research into LLM unlearning, focusing on erasing targeted undesirable knowledge without compromising the integrity of other, non-targeted responses. Existing studies have introduced various unlearning objectives to pursue LLM unlearning without necessitating complete retraining. However, each of these objectives has unique properties, and no unified framework is currently available to comprehend them thoroughly. To fill the gap, we propose a toolkit of the gradient effect (G-effect), quantifying the impacts of unlearning objectives on model performance from a gradient perspective. A notable advantage is its broad ability to detail the unlearning impacts from various aspects across instances, updating steps, and LLM layers. Accordingly, the G-effect offers new insights into identifying drawbacks of existing unlearning objectives, further motivating us to explore a series of new solutions for their mitigation and improvements. Finally, we outline promising directions that merit further studies, aiming at contributing to the community to advance this important field.

Qizhou Wang, Jin Peng Zhou, Zhanke Zhou, Saebyeol Shin, Bo Han, Kilian Q. Weinberger• 2025

Related benchmarks

TaskDatasetResultRank
Multi-turn Dialogue EvaluationMT-Bench
Overall Score6.98
532
Machine UnlearningTOFU Forget 10%
Aggregation Score53
81
Model UnlearningTOFU Forget 5% 1.0
Model Utility6.613
60
Knowledge RetentionMMLU (full)
MMLU Accuracy48
60
Language Model UnlearningTOFU Forget10
Forget Quality (FQ)0.1328
54
Machine UnlearningTOFU Forget 1%
Aggregation Score50
54
Machine UnlearningTOFU forget05 1.0
Model Utility (MU)61
53
Machine UnlearningTOFU 1.0 (forget01)
Average Score51
53
Knowledge UnlearningWMDP bio
Accuracy40.06
51
Machine UnlearningTOFU 1.0 (Retain Set)
ROUGE-L82.1
48
Showing 10 of 34 rows

Other info

Follow for update