Back to AI News
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.

AI Brief

It offers a concrete new unlearning method for researchers and practitioners concerned with removing sensitive, private, or copyrighted knowledge from LLMs.

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.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!