LAVA: A Unified Framework for Finetuning Language and Vision Models
Abstract
Lay Summary
Large AI models are powerful, but adapting them to new tasks often requires substantial memory and computation. Existing efficient fine-tuning methods, such as Low-Rank Adaptation (LoRA), reduce this cost by training only a small number of additional low-rank parameters. However, they may not fully use this limited training space, and they are less suitable for vision models that rely on convolution layers, where preserving the spatial structure of images is important. We introduce LAVA, a unified method for efficiently fine-tuning both language and vision models. Instead of treating all model components as the same flattened matrix, LAVA keeps the structure of convolution kernels and represents updates as a small set of learnable components. It also encourages these components to capture different information rather than becoming redundant, making training more stable. Across language understanding, commonsense reasoning, depth estimation, and image generation tasks, LAVA achieves stronger performance than LoRA while tuning only a small fraction of the model parameters. This makes adapting large AI models more practical and resource-efficient, especially when computing resources are limited.