arXiv:2411.15787cs.CV2024-11被引 2

用单模型生成多个辅助标记,低成本提升视觉表征学习效果

Multi-Token Enhancing for Vision Representation Learning

  • 从单个模型中同时提取多个辅助标记,避免多模型训练开销
  • 在多个下游任务上一致提升性能,且推理无额外成本
  • 适用于多种自监督方法,适合追求高效高精度的视觉研究者

视觉表征学习,尤其是自监督学习,在各类视觉应用中至关重要。集成学习虽能提升模型性能与鲁棒性,但传统集成策略对表征学习不切实际,尤其在需要大规模数据集和长时间训练的自监督场景下,因需k倍训练与推理计算成本而难以应用。为此,我们提出多标记增强(MTE),通过单个模型同时提取多个辅助标记(包括辅助CLS标记和自适应池化标记)来增强表征学习,仅带来极小额外训练开销,且无推理开销。这些辅助标记因差异捕捉互补信息。为缓解推理成本增加问题,我们在预训练阶段将辅助标记的知识蒸馏到全局标记中,从而可在推理时移除辅助标记而不增加成本。MTE兼容多种自监督损失函数与架构,在不同下游任务中均持续提升性能。源代码将公开。

原文摘要 · Abstract (English)

Vision representation learning, especially self-supervised learning, is pivotal for various vision applications. Ensemble learning has also succeeded in enhancing the performance and robustness of the vision models. However, traditional ensemble strategies are impractical for representation learning, especially self-supervised representation learning that requires large-scale datasets and long schedules. This is because they require k times more training and inference computation costs for an ensemble of k models. Differently, we introduce Multi-Token Enhancing (MTE) that extracts multiple auxiliary tokens simultaneously from a single model to enhance representation learning, while incurring minimal additional training costs and no additional inference costs. These auxiliary tokens, including auxiliary CLS tokens and adaptively pooled tokens, capture complementary information due to their differences. Meanwhile, to address the increase in inference costs, we distill the knowledge acquired by the auxiliary tokens into a global token during pre-training. Consequently, we can discard the auxiliary tokens during inference without incurring additional costs. Our MTE is compatible with various self-supervised loss functions and architectures, consistently improving performances across different downstream tasks. Our source code will be made publicly available.

视觉表征自监督学习多标记增强模型效率

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