模型汤只需一种成分
Model soups need only one ingredient
AI总结:
提出MonoSoup方法,利用SVD分解单检查点的层更新,通过熵有效秩自动重加权成分,实现强分布内-分布外平衡,无需多检查点。
摘要:
在目标分布上微调大型预训练模型通常会提高分布内(ID)准确性,但代价是分布外(OOD)鲁棒性下降,因为表示会专门适应微调数据。权重空间集成方法,如模型汤(Model Soups),通过平均多个检查点来缓解这一影响,但它们在计算上代价高昂,需要训练和存储数十个微调模型。在本文中,我们介绍了MonoSoup,一种简单、无数据、无超参数的事后方法,仅使用单个检查点即可实现强大的ID-OOD平衡。我们的方法对每一层的更新应用奇异值分解(SVD),将其分解为捕捉任务特定适应的高能量方向和引入噪声但可能仍编码对鲁棒性有用的残余信号的低能量方向。然后,MonoSoup使用基于熵的有效秩自动重新加权这些分量,并考虑模型的谱和几何结构的逐层系数。在ImageNet上微调并在自然分布偏移下评估的CLIP模型,以及在数学推理和多选题基准上测试的Qwen语言模型上的实验表明,这种即插即用方法是多检查点方法的实用且有效的替代方案,保留了其大部分好处而无需计算开销。
英文摘要:
Fine-tuning large pre-trained models on a target distribution often improves in-distribution (ID) accuracy, but at the cost of out-of-distribution (OOD) robustness as representations specialize to the fine-tuning data. Weight-space ensembling methods, such as Model Soups, mitigate this effect by averaging multiple checkpoints, but they are computationally prohibitive, requiring the training and storage of dozens of fine-tuned models. In this paper, we introduce MonoSoup, a simple, data-free, hyperparameter-free, post-hoc method that achieves a strong ID-OOD balance using only a single checkpoint. Our method applies Singular Value Decomposition (SVD) to each layer's update and decomposes it into high-energy directions that capture task-specific adaptation and low-energy directions that introduce noise but may still encode residual signals useful for robustness. MonoSoup then uses entropy-based effective rank to automatically re-weigh these components with layer-wise coefficients that account for the spectral and geometric structure of the model. Experiments on CLIP models fine-tuned on ImageNet and evaluated under natural distribution shifts, as well as on Qwen language models tested on mathematical reasoning and multiple-choice benchmarks, show that this plug-and-play approach is a practical and effective alternative to multi-checkpoint methods, retaining much of their benefits without their computational overhead.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。