Re:LoRA Learns Less and Forgets Less
by sp332 on Friday May 17, 2024 @12:10PM
(#40391394)
Do you mean leaving most of the model in its initial, randomised state and only training a LoRA?
[ Parent
| on HN ]
I’ve tested specifically this (on my personal time) :) It will train but I found the loss is proportional to the number of trainable parameters. So roughly to hit the performance of a standard 70m param model, you need to train ~70m lora params anyway.
It's worse than that, because lora requires two matrices per layer. At full rank, you have an additional NxN parameters to learn versus full finetuning, where N is min(input_features, output_features).
For example, tuning a layer of 128 in x 256 out is 32k params. Learning a full-rank lora for that layer would be two matrices of 128x128 and 128x256 = 48k params.