Not All Adapters Matter: Selective Adapter Freezing for Memory-Efficient Fine-Tuning of Language Models

Hyegang Son, Yonglak Son, Changhoon Kim, Young Geun Kim
11/26/2024
cs.CLcs.AIcs.LG

No official code found. Best implementation: CASL-KU/SAFE (1 star, MIT, last commit 2025-05-23).

Abstract

Transformer-based large-scale pre-trained models achieve great success. Fine-tuning is the standard practice for leveraging these models in downstream tasks. Among the fine-tuning methods, adapter-tuning provides a parameter-efficient fine-tuning by introducing lightweight trainable modules while keeping most pre-trained parameters frozen. However, existing adapter-tuning methods still impose substantial resource usage. Through our investigation, we show that each adapter unequally contributes to both task performance and resource usage. Motivated by this insight, we propose Selective Adapter FrEezing (SAFE), which gradually freezes less important adapters early to reduce unnecessary resource usage while maintaining performance. In our experiments, SAFE reduces memory usage, computation amount, and training time by 42.85\%, 34.59\%, and 11.82\%, respectively, while achieving comparable or better task performance compared to the baseline. We also demonstrate that SAFE induces regularization effect, thereby smoothing the loss landscape, which enables the model to generalize better by avoiding sharp minima.

View on arXivView PDF

Code Implementations(1)

CASL-KU/SAFECommunity64%
MIT

[NAACL 2025] Not All Adapters Matter: Selective Adapter Freezing for Memory-Efficient Fine-Tuning of Language Models

10Mar 25, 20251 years agoMIT

Similar Papers

Cite this paper

@article{son2024adapters,
  title  = {Not All Adapters Matter: Selective Adapter Freezing for Memory-Efficient Fine-Tuning of Language Models},
  author = {Hyegang Son and Yonglak Son and Changhoon Kim and Young Geun Kim},
  year   = {2024},
  eprint = {2412.03587},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.03587v2}
}

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

Discussion