Separate the Wheat from the Chaff: Model Deficiency Unlearning via Parameter-Efficient Module Operation

Xinshuo Hu, Dongfang Li, Baotian Hu, Zihao Zheng, Zhenyu Liu +1 more
8/16/2023
cs.CL

No official code found. Best implementation: HITsz-TMG/Ext-Sub (11 stars, MIT, last commit 2024-09-20). 1 more below.

Abstract

Large language models (LLMs) have been widely used in various applications but are known to suffer from issues related to untruthfulness and toxicity. While parameter-efficient modules (PEMs) have demonstrated their effectiveness in equipping models with new skills, leveraging PEMs for deficiency unlearning remains underexplored. In this work, we propose a PEMs operation approach, namely Extraction-before-Subtraction (Ext-Sub), to enhance the truthfulness and detoxification of LLMs through the integration of ``expert'' PEM and ``anti-expert'' PEM. Remarkably, even anti-expert PEM possess valuable capabilities due to their proficiency in generating fabricated content, which necessitates language modeling and logical narrative competence. Rather than merely negating the parameters, our approach involves extracting and eliminating solely the deficiency capability within anti-expert PEM while preserving the general capabilities. To evaluate the effectiveness of our approach in terms of truthfulness and detoxification, we conduct extensive experiments on LLMs, encompassing additional abilities such as language modeling and mathematical reasoning. Our empirical results demonstrate that our approach effectively improves truthfulness and detoxification, while largely preserving the fundamental abilities of LLMs.

View on arXivView PDF

Code Implementations(2)

HITsz-TMG/Ext-SubCommunity100%
MIT

Official implementation of our paper "Separate the Wheat from the Chaff: Model Deficiency Unlearning via Parameter-Efficient Module Operation". A model merge method for deficiency unlearning, compitable with huggingface peft (LoRA).

111Shell, PythonDec 11, 20232 years agoMIT
YanshekWoo/Ext-SubCommunity59%

[DEPRECATED] Official implementation of our paper "Separate the Wheat from the Chaff: Model Deficiency Unlearning via Parameter-Efficient Module Operation".

41Sep 18, 20232 years ago

Similar Papers

Cite this paper

@article{hu2023separate,
  title  = {Separate the Wheat from the Chaff: Model Deficiency Unlearning via Parameter-Efficient Module Operation},
  author = {Xinshuo Hu and Dongfang Li and Baotian Hu and Zihao Zheng and Zhenyu Liu and Min Zhang},
  year   = {2023},
  eprint = {2308.08090},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2308.08090v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2308.08090.svg)](https://codeofpaper.com/paper/2308.08090)

Discussion