Break the Output Geometry for Large Language Model Unlearning
Yejin Kim ⋅ Shen ⋅ Seokwon Jung ⋅ Seong Joon Oh
Abstract
Current machine unlearning methods for large language models (LLMs) struggle with a persistent trade-off between forgetting effectiveness and overall model utility. We attribute this trade-off to two empirical observations: (i) layer-wise logit accumulation toward a target token is driven more by the output token itself than by the input query, and (ii) hidden states that produce the same token vary only along directions orthogonal to the unembedding row $u_k$, creating what we term the same-output plane. Because a forget input shares its logit pathway with all retained contexts generating the same token, simply suppressing the forget logit inevitably compromises performance on those contexts. To overcome this, we propose **Break the Output Geometry (BOG)**. This approach preserves the same-output plane and specifically displaces the forget input away from it along the single direction $u_k$, using a margin derived from the model’s cross-target statistics. Empirically, BOG demonstrates a superior forget–retain trade-off on the TOFU benchmark.
Chat is not available.
Successful Page Load