arXiv:2610.41616v1 Announce Type: new
Abstract: Large language models (LLMs) acquire extensive factual knowledge during pre-training, which may include sensitive, private, or copyrighted information, motivating the development of unlearning methods that selectively remove specific knowledge while preserving general capabilities. However, existing approaches typically optimize entire responses without explicitly identifying the knowledge to be forgotten, resulting in coarse and uncontrolled forgetting. To address this, we revisit the forgetting target of LLM unlearning and introduce Knowledge Association (KA), which characterizes the association between a query context and its target knowledge and is instantiated by the corresponding conditional predictive distribution, as the fundamental target of forgetting. Based on KA, we propose Knowledge Association Reinforcement Unlearning (KARU), a unified reinforcement learning framework that suppresses forgotten knowledge associations while preserving retained ones. Specifically, KARU derives complementary mechanisms from the KA-based formulation: token-level reward construction for finer-grained forgetting supervision, distribution-level regularization for precise knowledge suppression, and proximity-aware retain sampling to mitigate collateral degradation. Experiments across diverse LLMs on representative benchmarks demonstrate that KARU consistently improves forgetting effectiveness over existing baselines while maintaining comparable retention performance and general utility.
arxiv
KARU: Reinforcement Unlearning of LLM Knowledge Associations
Researchers propose Knowledge Association (KA) as the target of LLM unlearning, defined as the conditional predictive distribution linking a query context to its target knowledge. They introduce Knowledge Association Reinforcement Unlearning (KARU), a reinforcement learning framework that combines token-level reward construction, distribution-level regularization, and proximity-aware retain sampling. Experiments across multiple LLMs and benchmarks report improved forgetting effectiveness while maintaining comparable retention and general utility.
It offers a concrete new unlearning method for researchers and practitioners concerned with removing sensitive, private, or copyrighted knowledge from LLMs.
Comments & Questions (0)
No comments yet
Be the first to comment!