arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2603.01792cs.CLcs.AI
收藏

ALTER: 用于LLM中基于令牌熵引导的不对称LoRA去学习

ALTER: Asymmetric LoRA for Token-Entropy-Guided Unlearning of LLMs

Xunlei Chen, Jinyu Guo, Yuang Li, Zhaokun Wang, Yi Gong, Jie Zou, Jiwei Wei, Wenhong Tian

AI总结:

ALTER通过不对称LoRA架构实现基于令牌熵引导的LLM高效去学习,取得SOTA性能并保留高模型效用。

摘要:

大型语言模型(LLMs)已发展到涵盖广泛领域中的大量知识。然而,控制LLMs不应知道的内容对于确保对齐和安全使用至关重要。然而,由于知识保留与遗忘之间的模糊边界,LLMs中的有效去学习困难重重。这种挑战在连续多领域训练导致的纠缠参数空间中进一步加剧,通常导致次生损害,尤其是在使用激进的去学习策略时。此外,优化具有数十亿参数的最新(SOTA)模型所需的计算开销也是一个额外的障碍。在本工作中,我们提出了ALTER,一种轻量级的去学习框架,用于解决LLMs中知识纠缠和去学习效率的挑战。ALTER通过两个阶段运行:(I)高熵令牌通过LoRA中的共享A矩阵被捕获和学习,随后(II)不对称LoRA架构通过参数隔离和在目标子领域内去学习令牌来实现指定的遗忘目标。作为在不对称框架中通过令牌级隔离实现去学习的新研究方向。ALTER在TOFU、WMDP和MUSE基准测试中实现了SOTA性能,去学习质量超过95%,并通过保留基础令牌表现出最小的副作用。通过将去学习与LLM的十亿级参数解耦,该框架在保持超过90%的模型效用的同时,实现了卓越的效率,超过基线保留率的47.8-83.6%。

英文摘要:

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%.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑