用双头优化提升视觉语言模型知识蒸馏效果,无需额外训练
Simple yet Effective Semi-supervised Knowledge Distillation from Vision-Language Models via Dual-Head Optimization
- 为不同信号设计双预测头,解决监督与蒸馏损失的梯度冲突
- 在15个数据集上超越基线,小模型性能超过教师模型
- 计算开销低,部署简单,适合实际应用和快速迭代
半监督学习(SSL)通过利用未标注数据缓解数据稀缺问题。近期,基于海量图文对预训练的视觉语言模型(VLMs)展现出卓越的零样本/少样本性能,常优于传统SSL方法,归因于其强大泛化能力。这引发关键问题:如何有效将VLM的泛化能力迁移到特定任务模型?知识蒸馏(KD)为此提供自然框架,但存在监督损失与蒸馏损失间的梯度冲突。为此,本文提出双头优化(DHO),为每类信号引入独立预测头。实验表明,DHO可缓解梯度冲突,显著提升特征学习能力,且计算开销极低,无需测试时调参或重新训练。在15个数据集上的广泛实验显示,DHO持续优于单头KD基线,常以更小的学生模型超越教师模型。同时,在ImageNet半监督学习(分布内)及跨ImageNet变体的分布外泛化任务中均取得新最优性能。代码与模型权重已公开:https://github.com/erjui/DHO。
原文摘要 · Abstract (English)
Semi-supervised learning (SSL) has emerged as a practical solution for addressing data scarcity challenges by leveraging unlabeled data. Recently, vision-language models (VLMs), pre-trained on massive image-text pairs, have demonstrated remarkable zero-/few-shot performance that often surpasses SSL approaches due to their exceptional generalization capabilities. This gap motivates us to question: how can we effectively harness the powerful generalization capabilities of VLMs into task-specific models? Knowledge distillation (KD) offers a natural framework for transferring VLM capabilities, but we identify that it suffers from gradient conflicts between supervised and distillation losses. To address this challenge, we propose Dual-Head Optimization (DHO), which introduces dual prediction heads for each distinct signal. We observe that DHO resolves gradient conflicts, enabling improved feature learning compared to single-head KD baselines, with practical benefits of minimal computational overhead and test-time hyperparameter tuning without retraining. Extensive experiments across 15 datasets show that DHO consistently outperforms KD baselines, often outperforming teacher models with smaller student models. DHO also achieves new state-of-the-art performance on both in-distribution ImageNet semi-supervised learning and out-of-distribution generalization across ImageNet variants. We publicly release our code and model checkpoints to facilitate future research at https://github.com/erjui/DHO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。