Distance Is Not Enough: Forget-Retain Alignment Gap Predicts LLM Relearning Robustness

Yi Chen, Hanna Hsieh, Shuhong Liu, Chuanbo Hua, Zihan Ma +2 more
8/26/2026
cs.AIcs.LG

Official code: Yi1-Chen/FRAG (0 stars, last commit 2026-08-27).

Abstract

Machine unlearning aims to make a model forget specific data, yet unlearned LLMs often fail to stay unlearned: brief fine-tuning can revive removed knowledge. Existing robustness predictors rely on global weight-space displacement, but distance alone can be misleading when random or destructive updates collapse performance. We argue that relearning robustness depends on update structure: robust unlearning should affect forget-critical weights while sparing retain-critical ones. We introduce the Forget-Retain Alignment Gap (FRAG), a training-free predictor that scores an update's forget-retain alignment without running a relearning attack, and separates selective from dense updates more reliably than global distance. Building on the forget-critical, retain-sparing principle, Forget-Retain Pruning (FRP) improves relearning robustness. Our results suggest that weight selectivity better explains robustness than distance alone. Code is available at https://github.com/Yi1-Chen/FRAG.

View on arXivView PDF

Code Implementations(1)

Yi1-Chen/FRAGOfficial100%
00Aug 24, 20261 months ago

Similar Papers

Cite this paper

@article{chen2026distance,
  title  = {Distance Is Not Enough: Forget-Retain Alignment Gap Predicts LLM Relearning Robustness},
  author = {Yi Chen and Hanna Hsieh and Shuhong Liu and Chuanbo Hua and Zihan Ma and Kun Wang and Joo-Young Kim},
  year   = {2026},
  eprint = {2608.25429},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2608.25429v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2608.25429.svg)](https://codeofpaper.com/paper/2608.25429)

Discussion