轻量微调可显著提升长尾分布下的模型性能
LIFT+: Lightweight Fine-Tuning for Long-Tail Learning
- 仅微调极少量参数,避免重微调导致的尾部类别性能下降
- 训练周期从约100轮缩减至15轮以内,参数量少于1%
- 适合需要快速高效适配长尾数据的部署场景
微调已成为基础模型应对长尾学习任务的主要方法,但现有策略对长尾性能的影响尚未被充分研究。本文发现,传统重微调(大量更新参数)会引发尾部类别的显著性能下降,而轻量微调表现更优。理论与实证分析表明,问题根源在于重微调造成类别条件分布不一致。为此,我们提出LIFT+:一种基于轻量微调的框架,结合语义感知初始化、极简数据增强和测试时集成,实现稳定的类别条件一致性。实验显示,该方法将训练轮数从约100轮降至≤15轮,所学参数少于1%,且在多个基准上显著超越现有最优方法。代码已开源。
原文摘要 · Abstract (English)
The fine-tuning paradigm has emerged as a prominent approach for addressing long-tail learning tasks in the era of foundation models. However, the impact of fine-tuning strategies on long-tail learning performance remains unexplored. In this work, we disclose that existing paradigms exhibit a profound misuse of fine-tuning methods, leaving significant room for improvement in both efficiency and accuracy. Specifically, we reveal that heavy fine-tuning (fine-tuning a large proportion of model parameters) can lead to non-negligible performance deterioration on tail classes, whereas lightweight fine-tuning demonstrates superior effectiveness. Through comprehensive theoretical and empirical validation, we identify this phenomenon as stemming from inconsistent class conditional distributions induced by heavy fine-tuning. Building on this insight, we propose LIFT+, an innovative lightweight fine-tuning framework to optimize consistent class conditions. Furthermore, LIFT+ incorporates semantic-aware initialization, minimalist data augmentation, and test-time ensembling to enhance adaptation and generalization of foundation models. Our framework provides an efficient and accurate pipeline that facilitates fast convergence and model compactness. Extensive experiments demonstrate that LIFT+ significantly reduces both training epochs (from $\sim$100 to $\leq$15) and learned parameters (less than 1%), while surpassing state-of-the-art approaches by a considerable margin. The source code is available at https://github.com/shijxcs/LIFT-plus.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。