Second-Order Information Matters: Revisiting Machine Unlearning for Large Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Gu, Kang, Rashid, Md Rafi Ur, Sultana, Najrin, Mehnaz, Shagufta
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929278891851776
author Gu, Kang
Rashid, Md Rafi Ur
Sultana, Najrin
Mehnaz, Shagufta
author_facet Gu, Kang
Rashid, Md Rafi Ur
Sultana, Najrin
Mehnaz, Shagufta
contents With the rapid development of Large Language Models (LLMs), we have witnessed intense competition among the major LLM products like ChatGPT, LLaMa, and Gemini. However, various issues (e.g. privacy leakage and copyright violation) of the training corpus still remain underexplored. For example, the Times sued OpenAI and Microsoft for infringing on its copyrights by using millions of its articles for training. From the perspective of LLM practitioners, handling such unintended privacy violations can be challenging. Previous work addressed the ``unlearning" problem of LLMs using gradient information, while they mostly introduced significant overheads like data preprocessing or lacked robustness. In this paper, contrasting with the methods based on first-order information, we revisit the unlearning problem via the perspective of second-order information (Hessian). Our unlearning algorithms, which are inspired by classic Newton update, are not only data-agnostic/model-agnostic but also proven to be robust in terms of utility preservation or privacy guarantee. Through a comprehensive evaluation with four NLP datasets as well as a case study on real-world datasets, our methods consistently show superiority over the first-order methods.
format Preprint
id arxiv_https___arxiv_org_abs_2403_10557
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Second-Order Information Matters: Revisiting Machine Unlearning for Large Language Models
Gu, Kang
Rashid, Md Rafi Ur
Sultana, Najrin
Mehnaz, Shagufta
Machine Learning
Artificial Intelligence
Computation and Language
With the rapid development of Large Language Models (LLMs), we have witnessed intense competition among the major LLM products like ChatGPT, LLaMa, and Gemini. However, various issues (e.g. privacy leakage and copyright violation) of the training corpus still remain underexplored. For example, the Times sued OpenAI and Microsoft for infringing on its copyrights by using millions of its articles for training. From the perspective of LLM practitioners, handling such unintended privacy violations can be challenging. Previous work addressed the ``unlearning" problem of LLMs using gradient information, while they mostly introduced significant overheads like data preprocessing or lacked robustness. In this paper, contrasting with the methods based on first-order information, we revisit the unlearning problem via the perspective of second-order information (Hessian). Our unlearning algorithms, which are inspired by classic Newton update, are not only data-agnostic/model-agnostic but also proven to be robust in terms of utility preservation or privacy guarantee. Through a comprehensive evaluation with four NLP datasets as well as a case study on real-world datasets, our methods consistently show superiority over the first-order methods.
title Second-Order Information Matters: Revisiting Machine Unlearning for Large Language Models
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2403.10557