Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

BitDelta: Your Fine-Tune May Only Be Worth One Bit

About

Large Language Models (LLMs) are typically trained in two phases: pre-training on large internet-scale datasets, and fine-tuning for downstream tasks. Given the higher computational demand of pre-training, it's intuitive to assume that fine-tuning adds less new information to the model, and is thus more compressible. We explore this assumption by decomposing the weights of fine-tuned models into their pre-trained components and an additional delta. We introduce a simple method, BitDelta, which successfully quantizes this delta down to 1 bit without compromising performance. This interesting finding not only highlights the potential redundancy of information added during fine-tuning, but also has significant implications for the multi-tenant serving and multi-tenant storage of fine-tuned models. By enabling the use of a single high-precision base model accompanied by multiple 1-bit deltas, BitDelta dramatically reduces GPU memory requirements by more than 10x, which can also be translated to enhanced generation latency in multi-tenant settings. We validate BitDelta through experiments across Llama-2 and Mistral model families, and on models up to 70B parameters, showcasing minimal performance degradation over all tested settings.

James Liu, Guangxuan Xiao, Kai Li, Jason D. Lee, Song Han, Tri Dao, Tianle Cai• 2024

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
Accuracy59.88
1460
Code GenerationHumanEval
Pass@183.5
850
Mathematical ReasoningGSM8K (test)
Accuracy61.11
797
Code GenerationHumanEval (test)
Pass@151.83
444
Mathematical ReasoningMATH (test)
Overall Accuracy12.12
433
Visual Question AnsweringGQA
Accuracy61.2
374
Multi-turn Dialogue EvaluationMT-Bench
Overall Score7.2
331
Physical Commonsense ReasoningPIQA
Accuracy81.22
329
Code GenerationMBPP (test)
Pass@158.5
276
Mathematical ReasoningAIME 2024
Accuracy23.3
251
Showing 10 of 27 rows

Other info

Code

Follow for update