通过分阶段的LoRA架构,精准删除大模型特定知识而不损伤其他能力。
ALTER: Asymmetric LoRA for Token-Entropy-Guided Unlearning of LLMs
- 用熵值筛选高信息量词元,分两阶段实现知识隔离与定向遗忘。
- 在TOFU等基准上遗忘准确率超95%,保留超过90%模型原有性能。
- 轻量级设计适合大规模模型,避免传统方法的副作用和高算力开销。
大型语言模型(LLMs)已涵盖广泛领域的知识,但控制其不应掌握的内容对对齐与安全使用至关重要。然而,由于知识保留与遗忘之间的边界模糊,有效实现未学习仍具挑战性。连续多领域训练导致参数空间高度纠缠,激进的遗忘策略常引发附带损伤。此外,优化数十亿参数的SOTA模型带来巨大计算开销。本文提出ALTER,一种轻量级的未学习框架,解决知识纠缠与效率问题。ALTER分为两阶段:(I) 通过LoRA共享矩阵A捕获高熵词元并学习;(II) 采用非对称LoRA结构,通过参数隔离,在目标子域内实现指定遗忘目标。该框架为基于词元级隔离的未学习开辟新方向。ALTER在TOFU、WMDP和MUSE基准上达到SOTA表现,遗忘质量超95%,同时通过保留基础词元实现最小副作用。通过解耦未学习与千亿级模型参数,该框架在保持超过90%模型效用的同时,显著提升效率,优于基线47.8%-83.6%的保留率。
原文摘要 · Abstract (English)
Large language models (LLMs) have advanced to encompass extensive knowledge across diverse domains. Yet controlling what a LLMs should not know is important for ensuring alignment and thus safe use. However, effective unlearning in LLMs is difficult due to the fuzzy boundary between knowledge retention and forgetting. This challenge is exacerbated by entangled parameter spaces from continuous multi-domain training, often resulting in collateral damage, especially under aggressive unlearning strategies. Furthermore, the computational overhead required to optimize State-of-the-Art (SOTA) models with billions of parameters poses an additional barrier. In this work, we present ALTER, a lightweight unlearning framework for LLMs to address both the challenges of knowledge entanglement and unlearning efficiency. ALTER operates through two phases: (I) high entropy tokens are captured and learned via the shared A matrix in LoRA, followed by (II) an asymmetric LoRA architecture that achieves a specified forgetting objective by parameter isolation and unlearning tokens within the target subdomains. Serving as a new research direction for achieving unlearning via token-level isolation in the asymmetric framework. ALTER achieves SOTA performance on TOFU, WMDP, and MUSE benchmarks with over 95% forget quality and shows minimal side effects through preserving foundational tokens. By decoupling unlearning from LLMs' billion-scale parameters, this framework delivers excellent efficiency while preserving over 90% of model utility, exceeding baseline preservation rates of 47.8-83.6%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。