English

OBLIVIATE: Robust and Practical Machine Unlearning for Large Language Models

Computation and Language 2025-09-10 v2 Artificial Intelligence Cryptography and Security Machine Learning

Abstract

Large language models (LLMs) trained over extensive corpora risk memorizing sensitive, copyrighted, or toxic content. To address this, we propose \textbf{OBLIVIATE}, a robust unlearning framework that removes targeted data while preserving model utility. The framework follows a structured process: extracting target tokens, building retain sets, and fine-tuning with a tailored loss function comprising three components -- masking, distillation, and world fact. Using low-rank adapters (LoRA) ensures efficiency without compromising unlearning quality. We conduct experiments on multiple datasets, including Harry Potter series, WMDP, and TOFU, using a comprehensive suite of metrics: \emph{forget quality} (via a new document-level memorization score), \emph{model utility}, and \emph{fluency}. Results demonstrate its effectiveness in resisting membership inference attacks, minimizing the impact on retained data, and maintaining robustness across diverse scenarios.

Keywords

Cite

@article{arxiv.2505.04416,
  title  = {OBLIVIATE: Robust and Practical Machine Unlearning for Large Language Models},
  author = {Xiaoyu Xu and Minxin Du and Qingqing Ye and Haibo Hu},
  journal= {arXiv preprint arXiv:2505.04416},
  year   = {2025}
}

Comments

To appear at EMNLP 25 main conference

R2 v1 2026-06-28T23:24:29.156Z