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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Anjarlekar, Ameya, Pombra, Sandeep
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_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