Advancing Parameter Efficiency in Fine-tuning via Representation Editing

Muling Wu, Wenhao Liu, Xiaohua Wang, Tianlong Li, Changze Lv +5 more
2/23/2024
cs.LGcs.CL

No official code found. Best implementation: mlwu22/RED (15 stars, last commit 2024-04-20).

Abstract

Parameter Efficient Fine-Tuning (PEFT) techniques have drawn significant attention due to their ability to yield competitive results while updating only a small portion of the adjustable parameters. However, existing PEFT methods pose challenges in hyperparameter selection, such as choosing the rank for LoRA or Adapter, or specifying the length of soft prompts. To address these challenges, we propose a novel fine-tuning approach for neural models, named Representation EDiting (RED), which modifies the representations generated at some layers through the application of scaling and biasing operations. While existing PEFT methods still demonstrate over-parameterization that could potentially undermine the generalization ability acquired from pre-training, RED can substantially reduce the number of trainable parameters by a factor of 25, 700 compared to full parameter fine-tuning and by a factor of 32 relative to LoRA. Remarkably, RED achieves results comparable or superior to both full parameter fine-tuning and other PEFT methods. Extensive experiments across various model architectures and scales, including RoBERTa, GPT-2, T5, and LLaMA-2, have demonstrated the effectiveness and efficiency of RED1, thereby positioning it as a promising PEFT strategy for large-scale neural models.

View on arXivView PDF

Code Implementations(1)

mlwu22/REDCommunity100%

Implementation code for ACL2024:Advancing Parameter Efficiency in Fine-tuning via Representation Editing

152C++, ShellApr 20, 20242 years ago

Similar Papers

Cite this paper

@article{wu2024advancing,
  title  = {Advancing Parameter Efficiency in Fine-tuning via Representation Editing},
  author = {Muling Wu and Wenhao Liu and Xiaohua Wang and Tianlong Li and Changze Lv and Zixuan Ling and Jianhao Zhu and Cenyuan Zhang and Xiaoqing Zheng and Xuanjing Huang},
  year   = {2024},
  eprint = {2402.15179},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.15179v3}
}

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/2402.15179.svg)](https://codeofpaper.com/paper/2402.15179)

Discussion