English

GAIN: Multiplicative Modulation for Domain Adaptation

Machine Learning 2026-04-22 v2 Artificial Intelligence

Abstract

Adapting LLMs to new domains causes forgetting because standard methods (e.g., full fine-tuning, LoRA) inject new directions into the weight space. We show that forgetting is governed by one algebraic property: whether the update preserves the column span of the pretrained weight matrix (Proposition 1). We propose GAIN, the simplest multiplicative alternative (W_new = S * W), which satisfies this by construction and can be absorbed into existing weights for zero inference cost. Across five models (774M to 70B) adapted sequentially over eight domains, GAIN improves earlier-domain perplexity by 7-13%, while LoRA degrades it by 18-36%. GAIN matches replay-augmented LoRA without storing prior data and dominates EWC on the forgetting-adaptation Pareto front. While LoRA can only reduce forgetting by sacrificing in-domain adaptation, GAIN achieves both with no domain boundaries and no regularization. The principle generalises: (IA)^3, an independent multiplicative method, also improves earlier domains.

Keywords

Cite

@article{arxiv.2604.04516,
  title  = {GAIN: Multiplicative Modulation for Domain Adaptation},
  author = {Hengshuai Yao and Xing Chen and Ahmed Murtadha and Guan Wang},
  journal= {arXiv preprint arXiv:2604.04516},
  year   = {2026}
}
R2 v1 2026-07-01T11:55:04.753Z