LLM Unlearning using Gradient Ratio-Based Influence Estimation and Noise Injection

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Anjarlekar, Ameya, Pombra, Sandeep
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912528105209856
author Anjarlekar, Ameya
Pombra, Sandeep
author_facet Anjarlekar, Ameya
Pombra, Sandeep
contents The growing legal and ethical scrutiny of large language models (LLMs) necessitates effective machine unlearning, particularly for sensitive or unauthorized data. Existing empirical methods often yield incomplete forgetting or unintended degradation of unrelated knowledge due to poor localization. In this work, we propose GRIN: a modular and targeted framework for LLM unlearning. GRIN introduces a novel gradient-ratio-based metric to identify parameters most responsible for memorizing forget data. We then perform selective noise injection into these parameters prior to fine-tuning, which improves unlearning performance while maintaining model utility. Finally, we propose new evaluation metrics tailored to the LLM setting and validate our approach on standard benchmarks such as TOFU, WMDP, and SafePKU.
format Preprint
id arxiv_https___arxiv_org_abs_2508_06467
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle LLM Unlearning using Gradient Ratio-Based Influence Estimation and Noise Injection
Anjarlekar, Ameya
Pombra, Sandeep
Machine Learning
The growing legal and ethical scrutiny of large language models (LLMs) necessitates effective machine unlearning, particularly for sensitive or unauthorized data. Existing empirical methods often yield incomplete forgetting or unintended degradation of unrelated knowledge due to poor localization. In this work, we propose GRIN: a modular and targeted framework for LLM unlearning. GRIN introduces a novel gradient-ratio-based metric to identify parameters most responsible for memorizing forget data. We then perform selective noise injection into these parameters prior to fine-tuning, which improves unlearning performance while maintaining model utility. Finally, we propose new evaluation metrics tailored to the LLM setting and validate our approach on standard benchmarks such as TOFU, WMDP, and SafePKU.
title LLM Unlearning using Gradient Ratio-Based Influence Estimation and Noise Injection
topic Machine Learning
url https://arxiv.org/abs/2508.06467