Parameter-Efficient Fine-Tuning (PEFT) strategies such as Low-Rank Adaptation (LoRA) are effective solutions for fine-tuning large-scale pre-trained models; however, their memory requirements scale with the size of the model, O(dr), where d is the model's hidden dimension and r is the rank. Our proposal, FrameFT, models the parameter update ΔW with a sparse coefficient matrix in a Fusion Frame basis. Fusion Frames can be generated algorithmically and shared across model layers, enabling very efficient updates. Only the sparse coefficients of the basis expansion are stored/optimized, reducing the memory footprint. The sparse structure of the coefficient matrix in FrameFT and the sparsity in the Fusion Frames give large compute benefits, and our analysis provides formal convergence results. We evaluate the idea across a suite of supervised fine-tuning benchmarks, focusing on language tasks, but also report application to vision models. Our experiments show that FrameFT achieves performance on par with/exceeding state-of-the-art PEFT techniques, but needs far fewer trainable parameters.
Figures & tables
Figure 1: Examples of tight frames with k=5 vectors in R3 .
Figure 2: ( left ) How parameter update δW is modeled by LoRA and by FrameFT. ( right ) The parameter update rule in FrameFT. We freeze the Fusion Frame projection matrices and train only the coefficients Cl .
Figure 3: An example of the coefficient matrix Cl from a Query layer (RoBERTa-L) with different subspace dimensions ρm=ρn=ρ .
Model
Method
Params
SST-2
MRPC
CoLA
QNLI
RTE
STS-B
Avg.
RoBERTa Base
FF*
125 M
94.8
90.2
63.6
92.8
78.7
91.2
85.2
SMT
55 K
94.2
89.5
63.9
91.8
78.1
90.4
84.7
LoRA*
0.3 M
95.1
89.7
63.4
93.3
78.4
91.5
85.2
SVFTd=2R
92 K
94.3
89.4
62.4
92.0
78.3
91.1
84.6
AdaLoRA*
0.3 M
94.5
88.7
62.0
93.1
81.0
90.5
85.0
FourierFT*
24 K
94.2
90.0
63.8
92.2
79.1
90.8
85.0
Table 1: Fine-Tuning RoBERTa Base and Large models on GLUE benchmark. * indicates the results reported in prior work. FrameFT performs better than full fine-tuning and LoRA, using 10× fewer parameters.
Method
#params
#toks/sec
Llama-2-7b
LoRA
262k
23.6k
DoRA
262k
14.1k
SVFit
1k
9.0k
FourierFT
1k
1.8k
FrameFT
1k
39.4k
Gemma-2-9b
LoRA
245k
22.1k
Table 2: Tokens per second for different PEFT methods.
Figure 4: Performance of RoBERTa base model fine-tuned with FrameFT versus the number of non-zero coefficients. The red line indicates the performance of full fine-tuning.
Model
Method
#Params
ARC-c
ARC-e
BoolQ
HellaSwag
OBQA
PIQA
RTE
WinoGrande
Avg.
Llama-2-7b
LoRA
16.7 M
45.82
77.02
78.81
58.08
35.2
78.83
61.01
70.72
63.18
(IA)3
614K
45.38
76.60
77.89
57.74
34.6
78.78
64.26
68.82
63.01
DoRA
16.7 M
45.14
76.39
78.35
57.78
34.0
78.73
67.15
67.17
63.09
S2FT
56.6 M
46.16
77.44
78.38
58.10
32.8
78.84
61.01
68.58
62.66
SVFTd=8B
4.5 M
45.79
77.07
78.83
57.98
34.6
78.56
61.52
70.64
63.12
FourierFT
320 K
44.96
77.14
79.05
58.21
34.6
78.89
62.45
70.48
63.22
Table 3: Performance of LLMs fine-tuned on the Alpaca dataset by various methods and then evaluated on the LM-evaluation-harness. FrameFT performs competitively with all the baseline methods under comparison, using a 30× fewer number of parameters than LoRA. FrameFT needs minimal hyperparameter tuning.
Method
SST-2
MRPC
CoLA
QNLI
RTE
STS-B
Avg.
Shared random mask
94.40±0.17
92.31±0.46
66.58±0.77
92.35±0.12
80.21±0.82
90.96±0.11
86.18±0.18
Per-layer random mask
94.10±0.16
93.53±0.32
66.44±0.61
92.31±0.14
80.34±0.39
90.86±0.09
86.26±0.17
Magnitude-based mask
94.19
92.49
65.82
91.61
78.40
91.01
85.60
Gradient-based mask
89.39
74.59
60.30
84.80
63.49
76.40
74.83
Table 4: Sparsity mask ablation on RoBERTa-Base/GLUE (averaged over five seeds). Shared and per-layer random masks perform within noise of each other while adaptive heuristics perform strictly worse.
Metric
Method
Llama-2-7B
Llama-2-13B
Gemma-2-9B
Gemma-2-2B
Llama-3.1-8B
Training time (min)
LoRA
238
363
323
126
231
FrameFT
225
345
301
115
212
GPU memory (GiB)
LoRA
30.1
34.8
37.8
26.9
38.2
FrameFT
29.5
34.7
37.7
26.8
37.6
Table 5: Training time (minutes) and peak GPU memory (GiB) for LoRA and FrameFT on the Alpaca instruction tuning task.
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 5: Performance of FrameFT on CoLA and MRPC for different values of subspace dimension ρ
Method
Params
Pets
Cars
CIFAR10
DTD
EuroSAT
FGVC
RESISC45
CIFAR100
Avg.
ViT-B
Full finetune*
85.8 M
93.14
79.78
98.92
77.68
99.05
54.84
96.13
92.38
86.49
LinearProbe*
−
90.28
25.76
96.41
69.77
88.72
17.44
74.22
84.28
68.36
LoRA*
581 K
93.19
45.38
98.78
74.95
98.44
25.16
92.70
92.02
77.58
FourierFT
72 K
93.21
46.11
98.58
75.09
98.29
27.51
91.97
91.20
77.75
FourierFT
239 K
93.05
56.36
98.69
77.30
98.78
32.44
94.26
91.45
80.29
FrameFT ( ours )
24 K
93.71
70.84
98.50
77.80
98.52
44.45
92.30
90.80
83.36
Appendix
Table 6: Performance of different methods for Fine-Tuning ViT Base and Large models on different datasets. * indicates the results reported in previous work. We highlight non PEFT methods in gray . FrameFT performs better than all the baseline methods on average, using a 30x lower number of parameters than LoRA.
Figure 6: Performance of RoBERTa base model fine-tuned with FrameFT versus the number of non-zero coefficients on the GLUE benchmark. The red line indicates the performance of full fine-tuning.
Optimizer
AdamW
LR scheduler
Linear schedule with warmup
Batch size
32
Head learning rate
3E-3
Adapter learning rate
0.12
Warmup ratio
0.06
Max. Seq. length
512
Appendix
Table 7: Hyperparameters for the GLUE benchmark
Optimizer
AdamW
LR scheduler
Linear schedule with warmup
Batch size
50
Head learning rate
3E-2
Adapter learning rate
0.33
Warmup ratio
0.06
α
250.0
Appendix
Table 8: Hyperparameters for the Image classification task
Optimizer
AdamW
LR scheduler
Linear schedule with warmup
Batch size
128
Learning rate
0.1
Warmup ratio
0.03
α
200.0
Appendix
Table 9: Hyperparameters for the Instruction tuning task