Parameter-Efficient Fine-Tuning with Layer Pruning on Free-Text Sequence-to-Sequence Modeling

Yunqi Zhu, Xuebing Yang, Yuanyuan Wu, Wensheng Zhang
5/15/2023
cs.CLcs.AI

No official code found. Best implementation: zhuyunqi96/LoraLPrun (13 stars, last commit 2023-05-21).

Abstract

The increasing size of language models raises great research interests in parameter-efficient fine-tuning such as LoRA that freezes the pre-trained model, and injects small-scale trainable parameters for multiple downstream tasks (e.g., summarization, question answering and translation). To further enhance the efficiency of fine-tuning, we propose a framework that integrates LoRA and structured layer pruning. The integrated framework is validated on two created deidentified medical report summarization datasets based on MIMIC-IV-Note and two public medical dialogue datasets. By tuning 0.6% parameters of the original model and pruning over 30% Transformer-layers, our framework can reduce 50% of GPU memory usage and speed up 100% of the training phase, while preserving over 92% generation qualities on free-text sequence-to-sequence tasks.

View on arXivView PDF

Code Implementations(1)

131PythonMay 13, 20233 years ago

Similar Papers

Cite this paper

@article{zhu2023parameterefficient,
  title  = {Parameter-Efficient Fine-Tuning with Layer Pruning on Free-Text Sequence-to-Sequence Modeling},
  author = {Yunqi Zhu and Xuebing Yang and Yuanyuan Wu and Wensheng Zhang},
  year   = {2023},
  eprint = {2305.08285},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2305.08285v3}
}

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

Discussion