Search papers, labs, and topics across Lattice.
The paper introduces GAIN, a parameter-efficient adaptation method that modulates existing features in LLMs through multiplicative scaling, aiming to mitigate catastrophic forgetting during domain adaptation. GAIN learns a diagonal scaling matrix applied to the attention output projection and/or FFN layers, inspired by gain modulation in neuroscience. Experiments across multiple models and domains demonstrate that GAIN achieves comparable in-domain adaptation performance to LoRA while significantly reducing forgetting, even improving performance on previously trained domains.
Forget catastrophic forgetting: a neuroscience-inspired scaling method actually *improves* performance on prior domains while adapting LLMs to new tasks.
Adapting LLMs to new domains causes forgetting because standard methods (full fine-tuning, LoRA) inject new directions into the weight space. We propose GAIN, which re-emphasizes existing features through multiplicative modulation W_new = S * W. The learned diagonal matrix S is applied to the attention output projection and optionally the FFN. The principle mirrors gain modulation in neuroscience, where neurons adapt to context by scaling response strength while preserving selectivity. We evaluate GAIN on five models from four families (774M to 70B), adapting sequentially across eight domains. GAIN-FFN matches LoRA's in-domain adaptation, but their effects on previously trained domains are opposite: GAIN-FFN improves them by 7-13% (validation PPL), while LoRA degrades them by 18-36%. Downstream accuracy confirms the pattern: for example, after seven sequential adaptations on Qwen2.5, GAIN-FFN degrades BoolQ by only 0.8% while LoRA damages it by 14.9%. GAIN adds 46K-230K parameters per model and can be absorbed into the pretrained weights for zero inference cost.