用视觉语言模型指导小样本主动学习,提升标注效率。
PCoreSet: Effective Active Learning through Knowledge Distillation from Vision-Language Models
- 利用大模型零样本能力构建教师模型,解决小数据下知识蒸馏难题。
- 提出概率核心集策略,在概率空间最大化多样性,提升知识迁移效率。
- 在11个数据集上表现优异,64/73场景最优,显著降低标注成本。
知识蒸馏(KD)通过从教师模型迁移知识来训练轻量级任务专用模型,但其在主动学习(AL)中的应用尚未充分探索。原因在于,传统KD依赖充足标注数据,而主动学习处于数据稀缺场景,难以获取特定任务的教师模型。本文提出ActiveKD框架,结合主动学习与知识蒸馏,利用大型视觉-语言模型(VLMs)的零样本和少样本能力。关键发现是,VLMs的预测在概率空间中呈现结构化聚集,这种模式可作为教师模型的归纳偏置,捕获泛化性输出特征。为此,我们提出概率核心集(PCoreSet),通过最大化概率空间覆盖度而非特征空间,选择具有概率多样性的未标注样本,从而在有限标注预算下更高效地传递教师知识。在11个数据集上的广泛实验表明,ActiveKD在多种选择方法中持续提升性能(如ImageNet上平均提升29.07%)。在5种学生网络和3种教师网络组合中,PCoreSet在73个设置中取得64次最佳结果(约87.7%),仅前两轮主动学习周期除外。代码已公开于https://github.com/erjui/PCoreSet。
原文摘要 · Abstract (English)
Knowledge distillation (KD) is a widely used framework for training compact, task-specific models by transferring the knowledge from teacher models. However, its application to active learning (AL), which aims to minimize annotation costs through iterative sample selection, remains underexplored. This gap stems from the fact that KD typically assumes access to sufficient labeled data, whereas AL operates in data-scarce scenarios where task-specific teacher models are often unavailable. In this paper, we first introduce ActiveKD, a framework that integrates AL with KD by leveraging the zero- and few-shot capabilities of large vision-language models (VLMs). A key aspect of ActiveKD is the structured prediction bias of VLMs-i.e., their predictions form clusters in the probability space. We regard this structure as an inductive bias of the teacher model, capturing generalizable output patterns beneficial to student learning. To exploit this bias, we propose Probabilistic CoreSet (PCoreSet), a selection strategy that maximizes coverage in the probability space rather than the feature space. PCoreSet strategically selects probabilistically diverse unlabeled samples, facilitating more efficient transfer of teacher knowledge under limited annotation budgets. Extensive evaluations on 11 datasets show that ActiveKD consistently improves performance across selection methods (e.g., +29.07% on ImageNet, averaged over methods). Under ActiveKD, PCoreSet ranks first in 64/73 settings (approximately 87.7%) across 5 student and 3 teacher networks, always achieving the best performance except for first 2 AL rounds. Our code is available at https://github.com/erjui/PCoreSet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。