English

Propulsion: Steering LLM with Tiny Fine-Tuning

Computation and Language 2024-12-17 v3

Abstract

The rapid advancements in Large Language Models (LLMs) have revolutionized natural language processing (NLP) and related fields. However, fine-tuning these models for specific tasks remains computationally expensive and risks degrading pre-learned features. To address these challenges, we propose Propulsion, a novel parameter efficient fine-tuning (PEFT) method designed to optimize task-specific performance while drastically reducing computational overhead. Inspired by the concept of controlled adjustments in physical motion, Propulsion selectively re-scales specific dimensions of a pre-trained model, guiding output predictions toward task objectives without modifying the model's parameters. By introducing lightweight, trainable Propulsion parameters at the pre-trained layer, we minimize the number of parameters updated during fine-tuning, preventing overfitting or overwriting of existing knowledge. Our theoretical analysis, supported by Neural Tangent Kernel (NTK) theory, shows that Propulsion approximates the performance of full fine-tuning with far fewer trainable parameters. Empirically, Propulsion reduces the parameter count from 355.3 million to just 0.086 million, achieving over a 10x reduction compared to standard approaches like LoRA while maintaining competitive performance across benchmarks.

Keywords

Cite

@article{arxiv.2409.10927,
  title  = {Propulsion: Steering LLM with Tiny Fine-Tuning},
  author = {Md Kowsher and Nusrat Jahan Prottasha and Prakash Bhat},
  journal= {arXiv preprint arXiv:2409.10927},
  year   = {2024}
}

Comments

26 pages, 11 figures accepted paper

R2 v1 2026-06-28T18:47:16.833Z