提出特征空间适配方法VeFA,缓解微调时灾难性遗忘问题。
VeFA: Vector-Based Feature Space Adaptation for Robust Model Fine-Tuning
- 在特征空间而非参数空间进行微调,避免引入干扰维度。
- 在图像分类、NLU、NLG任务上性能相当,鲁棒性显著提升。
- 适合下游数据少或分布差异大的场景,尤其关注模型稳定性。
灾难性遗忘是模型微调中的经典难题,尤其在下游数据标注少或与预训练分布差异大时更为严重。现有参数高效微调方法主要在权重空间操作,易导致模型过度适应下游数据。研究表明,微调中引入的干扰维度是遗忘的重要原因。本文提出向量化特征适配(VeFA),直接在特征空间进行微调,天然避免生成干扰维度。VeFA对每个特征进行逐元素调整,确保微调后的有效权重始终位于预训练权重矩阵的列空间内。该方法受下游隐变量效应等价建模(EEM)启发,通过轻量级特征变换补偿隐变量影响,保留预训练表征,增强分布偏移下的泛化能力。在图像分类、自然语言理解(NLU)和自然语言生成(NLG)基准上,与LoRA相比,VeFA在标准微调性能相当的同时,表现出更强的鲁棒性。
原文摘要 · Abstract (English)
Catastrophic forgetting is a well-documented challenge in model fine-tuning, particularly when the downstream domain has limited labeled data or differs substantially from the pre-training distribution. Existing parameter-efficient fine-tuning methods largely operate in the weight space by modifying or augmenting the parameters of the pre-trained model, which can lead to models that are overly specialized to the observed downstream data. Recent studies suggest that one mechanism underlying such forgetting is the introduction of intruder dimensions into the representation space during fine-tuning. To mitigate the risk of overwriting pre-trained knowledge and to enhance robustness, we propose Vector-based Feature Adaptation (VeFA), a new fine-tuning method that operates directly in the feature space, which naturally avoids generating intruder dimensions. VeFA performs element-wise adaptation on individual features, thereby ensuring that the effective fine-tuned weights always remain within the column space of the pre-trained weight matrix. This feature-space adaptation perspective is inspired by the idea of effect equivalence modeling (EEM) of downstream lurking variables that induce distribution shifts, which posits that the influence of unobserved factors can be represented as an equivalent aggregate effect on observed features. By compensating for the effects of downstream lurking variables via a lightweight feature-level transformation, VeFA preserves the pre-trained representations and improves model generalization under distribution shift. We evaluate VeFA against LoRA on image classification, NLU, and NLG benchmarks, considering both standard fine-tuning performance and robustness; across these tasks, VeFA achieves comparable fine-tuning performance while consistently exhibiting stronger robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。