Vocabulary-level Memory Efficiency for Language Model Fine-tuning

Code

• github.com/mlsw/dynamic-embedding-pruning
  OfficialIn paperpytorch★ 8
• github.com/mlsw/partial-embedding-matrix-adaptation
  OfficialIn paperpytorch★ 8

Abstract

The extensive memory footprint of language model (LM) fine-tuning poses a challenge for both researchers and practitioners. LMs use an embedding matrix to represent extensive vocabularies, forming a substantial proportion of the model parameters. While previous work towards memory-efficient fine-tuning has focused on minimizing the number of trainable parameters, reducing the memory footprint of the embedding matrix has yet to be explored. We first demonstrate that a significant proportion of the vocabulary remains unused during fine-tuning. We then propose a simple yet effective approach that leverages this finding to minimize memory usage. We show that our approach provides substantial reductions in memory usage across a wide range of models and tasks. Notably, our approach does not impact downstream task performance, while allowing more efficient use of computational resources.

Tasks

Reproductions