arXiv:2409.11323cs.CVcs.LG2024-09

用少量可训练参数提升视觉模型在长尾数据上的分类能力

LPT++: Efficient Training on Mixture of Long-tailed Experts

  • 通过可学习提示与专家混合机制,适配预训练视觉模型
  • 仅需约1%额外参数,性能媲美现有方法
  • 适合资源有限但需处理长尾分布的任务场景

我们提出LPT++,一个面向长尾分类的综合框架,结合参数高效微调(PEFT)与可学习模型集成。LPT++通过三个核心组件增强冻结的视觉变压器(ViT):一是通用长尾适配模块,聚合长尾提示与视觉适配器,提升模型判别能力;二是长尾专家混合框架,结合视觉专家与视觉-语言(VL)专家的置信度得分,自适应计算重加权系数以生成更准确预测;三是三阶段训练策略,分步学习各模块,确保训练稳定有效。此外,我们还提出简化版LPT,仅融合视觉预训练ViT与长尾提示,实现单模型方法,在不依赖VL预训练模型的情况下仍达到相当性能。实验表明,仅增加约1%可训练参数,LPT++即达到与现有方法相当的准确率。

原文摘要 · Abstract (English)

We introduce LPT++, a comprehensive framework for long-tailed classification that combines parameter-efficient fine-tuning (PEFT) with a learnable model ensemble. LPT++ enhances frozen Vision Transformers (ViTs) through the integration of three core components. The first is a universal long-tailed adaptation module, which aggregates long-tailed prompts and visual adapters to adapt the pretrained model to the target domain, meanwhile improving its discriminative ability. The second is the mixture of long-tailed experts framework with a mixture-of-experts (MoE) scorer, which adaptively calculates reweighting coefficients for confidence scores from both visual-only and visual-language (VL) model experts to generate more accurate predictions. Finally, LPT++ employs a three-phase training framework, wherein each critical module is learned separately, resulting in a stable and effective long-tailed classification training paradigm. Besides, we also propose the simple version of LPT++ namely LPT, which only integrates visual-only pretrained ViT and long-tailed prompts to formulate a single model method. LPT can clearly illustrate how long-tailed prompts works meanwhile achieving comparable performance without VL pretrained models. Experiments show that, with only ~1% extra trainable parameters, LPT++ achieves comparable accuracy against all the counterparts.

长尾分类视觉模型参数高效

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。