Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Fora: From Weight-Space to Function-Space Protection in Capability-Preserving Fine-Tuning

About

Full fine-tuning adapts large language models to new tasks but can erode capabilities they already possess. Existing remedies protect through proxies such as parameter distances, importance penalties, output matching, or dominant singular directions of the weights, but none directly asks which activation directions the preserved capability relies on. We argue that a capability is characterized more faithfully by the activation subspace it induces than by the singular geometry of the weight matrix, and develop function-space protection, instantiated as FORA (Function-space Orthogonal Residual Adaptation). From label-free calibration inputs, FORA estimates, per layer, the principal directions $Q$ of the input-activation covariance and forms a right projector $P_Q = I - QQ^T$. Paired with a left projector $P_U$ from the weight SVD, the update is $\Delta W = P_U M P_Q + U_2 D_{\delta} V_2^T$: a high-capacity branch structurally barred from reading capability-relevant function directions, plus a narrow spectral channel for controlled plasticity. The construction extends to parameter-efficient adaptation via $M \to (\alpha/r) BA$. Across three settings on Qwen3-1.7B, including COGS and GSM8K learned while preserving translation and translation learned while preserving math, FORA consistently improves preservation over weight-space projection and standard regularization, with only a small new-task trade-off in the math-preservation setting. A controlled ablation isolating the projection source shows that the advantage comes not from projection itself, but from projecting onto capability-derived rather than weight-derived directions. Code is available at https://github.com/zrui239/FORA.

Rui Zhou, Tianci Xie• 2026

Related benchmarks

TaskDatasetResultRank
Semantic ParsingCOGS (test)
Exact Match Accuracy98.68
23
Semantic ParsingCOGS
Accuracy94.6
22
TranslationTranslation
chrF41.3
20
Mathematical ReasoningGSM8K
Accuracy46.87
19
Mathematical ReasoningGSM8K (test)
GSM8K EM48.84
12
TranslationOPUS English-Chinese (test)
chrF41.4
8
TranslationTranslation
chrF0.411
8
Semantic ParsingCOGS
COGS EM94.6
7
Language Modeling and TranslationGeneral Language Preservation Task
chrF41.3
7
Showing 9 of 9 rows

Other info

Follow for update