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

Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning

About

This work studies the problem of large language model (LLM) unlearning, aiming to remove unwanted data influences (e.g., copyrighted or harmful content) while preserving model utility. Despite the increasing demand for unlearning, a technically-grounded optimization framework is lacking. Gradient ascent (GA)-type methods, though widely used, are suboptimal as they reverse the learning process without controlling optimization divergence (i.e., deviation from the pre-trained state), leading to risks of over-forgetting and potential model collapse. Negative preference optimization (NPO) has been proposed to address this issue and is considered one of the state-of-the-art LLM unlearning approaches. In this work, we revisit NPO and identify another critical issue: reference model bias. This bias arises from using the reference model (i.e., the model prior to unlearning) to evaluate the unlearning success, which can compromise NPO's effectiveness. Specifically, it leads to (a) uneven allocation of optimization power across forget data with varying difficulty levels and (b) ineffective gradient weight smoothing during the early stages of unlearning optimization. To overcome these challenges, we propose a simple yet effective unlearning optimization framework, called SimNPO, showing that `simplicity' in removing the reliance on a reference model (through the lens of simple preference optimization) benefits unlearning. We provide deeper insights into SimNPO's advantages through an analysis based on mixtures of Markov chains. Extensive experiments further validate SimNPO's efficacy on benchmarks like TOFU and MUSE, as well as its robustness against relearning attacks. Codes are available at https://github.com/OPTML-Group/Unlearn-Simple.

Chongyu Fan, Jiancheng Liu, Licong Lin, Jinghan Jia, Ruiqi Zhang, Song Mei, Sijia Liu• 2024

Related benchmarks

TaskDatasetResultRank
Multi-task Language UnderstandingMMLU
MMLU Accuracy49.5
456
General Knowledge EvaluationMMLU
MMLU Accuracy58.2
167
Knowledge UnlearningWMDP bio
Accuracy22.2
93
Machine UnlearningMUSE Books
Privacy Leakage-54.4
90
Machine UnlearningTOFU (5%)
Forget Quality0.6284
82
Machine UnlearningTOFU Forget 10%
Aggregation Score47
81
Machine UnlearningTOFU 10% forget
Privacy Leakage1.99
72
Language UnderstandingMMLU
MMLU Score59.6
70
Model UnlearningTOFU Forget 5% 1.0
Model Utility6.809
60
Machine UnlearningTOFU (10%)
Forget Quality (FQ)0.45
60
Showing 10 of 114 rows
...

Other info

Follow for update