arXiv:2410.17509cs.LG2024-10NeurIPS被引 27

提出WAGLE方法,精准定位影响模型遗忘的关键权重。

WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Models

  • 基于权重影响力分析,指导不同遗忘任务的权重调整。
  • 在多个模型和任务中实现高效遗忘,同时保持原始性能。
  • 首次系统性地为大模型遗忘提供可解释的权重归因机制。

大语言模型(LLM)亟需有效的遗忘机制以满足数据合规与伦理要求。现有研究多集中于设计不同遗忘方法以提升效率与效果,但对模型权重与遗忘过程之间关系的探讨仍不充分。本文系统研究了权重在遗忘过程中的作用,提出基于权重归因的遗忘方法WAGLE,揭示了权重影响力与需遗忘/保留数据影响力之间的关联。WAGLE通过策略性引导,适用于梯度差、(负)偏好优化等多种遗忘方法,以及虚构内容删除、恶意使用防范、版权信息清除等应用,在Zephyr-7b-beta与Llama2-7b等模型上均表现优异。据我们所知,这是首个系统性的权重归因方法,显著优于以往缺乏归因或仅用简单归因的方法。

原文摘要 · Abstract (English)

The need for effective unlearning mechanisms in large language models (LLMs) is increasingly urgent, driven by the necessity to adhere to data regulations and foster ethical generative AI practices. Despite growing interest of LLM unlearning, much of the existing research has focused on varied unlearning method designs to boost effectiveness and efficiency. However, the inherent relationship between model weights and LLM unlearning has not been extensively examined. In this paper, we systematically explore how model weights interact with unlearning processes in LLMs and we design the weight attribution-guided LLM unlearning method, WAGLE, which unveils the interconnections between 'influence' of weights and 'influence' of data to forget and retain in LLM generation. By strategically guiding the LLM unlearning across different types of unlearning methods and tasks, WAGLE can erase the undesired content, while maintaining the performance of the original tasks. We refer to the weight attribution-guided LLM unlearning method as WAGLE, which unveils the interconnections between 'influence' of weights and 'influence' of data to forget and retain in LLM generation. Our extensive experiments show that WAGLE boosts unlearning performance across a range of LLM unlearning methods such as gradient difference and (negative) preference optimization, applications such as fictitious unlearning, malicious use prevention, and copyrighted information removal, and models including Zephyr-7b-beta and Llama2-7b. To the best of our knowledge, our work offers the first principled method for attributing and pinpointing the influential weights in enhancing LLM unlearning. It stands in contrast to previous methods that lack weight attribution and simpler weight attribution techniques.

大模型遗忘权重归因模型可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。