CLAIDec 18, 2024

Refining Salience-Aware Sparse Fine-Tuning Strategies for Language Models

arXiv:2412.13488v23 citationsh-index: 9Has CodeACL
Originality Incremental advance
AI Analysis

This work provides a simple and effective baseline for SPEFT, challenging the need for complexity in parameter-efficient fine-tuning for large language models.

The paper tackled the problem of selecting effective salience metrics for sparsity-based parameter-efficient fine-tuning (SPEFT) in language models, finding that a simple gradient-based metric with static masking outperforms other methods across NLP tasks while being computationally efficient.

Parameter-Efficient Fine-Tuning (PEFT) has gained prominence through low-rank adaptation methods like LoRA. In this paper, we focus on sparsity-based PEFT (SPEFT), which introduces trainable sparse adaptations to the weight matrices in the model, offering greater flexibility in selecting fine-tuned parameters compared to low-rank methods. We conduct the first systematic evaluation of salience metrics for SPEFT, inspired by zero-cost NAS proxies, and identify simple gradient-based metrics is reliable, and results are on par with the best alternatives, offering both computational efficiency and robust performance. Additionally, we compare static and dynamic masking strategies, finding that static masking, which predetermines non-zero entries before training, delivers efficiency without sacrificing performance, while dynamic masking offers no substantial benefits. Across NLP tasks, a simple gradient-based, static SPEFT consistently outperforms other fine-tuning methods for LLMs, providing a simple yet effective baseline for SPEFT. Our work challenges the notion that complexity is necessary for effective PEFT, while our open-source framework establishes a reproducible benchmark for future research, which is available at [https://github.com/0-ml/speft].

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes