Learning Rate Scaling across LoRA Ranks and Transfer to Full Finetuning
Abstract
Lay Summary
Today's AI models are first trained on huge amounts of data, then "finetuned" — adjusted to do a specific job. Updating every part of such a large model is expensive, so practitioners often use LoRA, a lightweight method that trains only a small add-on rather than the whole model. But LoRA has several settings that interact in complicated ways. The most important of these is the learning rate, which sets how aggressively the model adjusts as it learns: its best value shifts whenever you resize the add-on, forcing slow trial-and-error retuning. We developed a mathematical framework that predicts how the learning rate should scale with the add-on's size, keeping learning in a stable sweet spot — neither overshooting nor stalling. We also discovered that, in one common setup, the best learning rate for LoRA coincides with the one needed to fully update the model. Researchers can therefore tune this setting cheaply with LoRA and reuse it for expensive full finetuning, skipping costly searches. This saves time, money, and energy, and lowers the barrier for those with limited computing power. We confirmed these rules across language, vision, image-generation, and reinforcement-learning tasks.