Foundation-Preserving Adaptation via Generalized Rayleigh-Quotient Optimization
About
While finetuning effectively adapts foundation models to specialized downstream tasks, it can degrade nontarget capabilities acquired during pretraining. Existing forgetting aware methods typically seek safer updates through specialized initialization or fixed constraints, but do not regulate the adaptation preservation trade-off during training. We propose Foundation Preserving LoRA (FoLoRA), a forgetting aware optimization framework. Guided by a first order preservation condition, FoLoRA defines a forgetting penalty over pretraining-proxy activations and a task utility over downstream task activations. It then scores update directions by task utility per unit forgetting penalty via a generalized Rayleigh quotient. The resulting spectral coordinate system enables direction wise gated Adam updates, attenuating low utility to penalty directions during training. To estimate the forgetting penalty, FoLoRA constructs pretraining proxy calibration data by sampling from the pretrained model rather than relying on a single proxy dataset. Experiments on math, code, and instruction following adaptation show that FoLoRA achieves the strongest preservation adaptation balance over baselines, improving target task performance with best aggregate preservation of non target capabilities.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Instruction Following | IFEval | IFEval Accuracy23.84 | 836 | |
| Multitask Knowledge | MMLU | Accuracy60.11 | 92 | |
| Factual Question Answering | TriviaQA | Accuracy32.46 | 46 | |
| Mathematical Reasoning | GSM8K and MATH | GSM8K Score46.09 | 38 | |
| Factual QA | NQ-Open | Accuracy7.73 | 36 | |
| Reading Comprehension | SQuAD v2 | Accuracy27.98 | 20 | |
| Long-context language modeling | LongBench v2 | Accuracy29.62 | 12 | |
| Long-context Reasoning | BABILong | Accuracy66.5 | 12 | |
| Hallucination Robustness | TruthfulQA | TruthfulQA Accuracy48.46 | 12 | |
| Open-domain Question Answering | WebQuestions (WebQs) | WebQS Score7.97 | 12 |