GIFT:在接近零成本的缩减数据集中挖掘标签的全部潜力

💡 原文中文,约500字,阅读约需1分钟。
📝

内容提要

最近的研究表明,使用预训练教师模型生成的软标签在数据集蒸馏中具有优势。本文提出了一种新的方法GIFT,通过优化软标签和使用余弦相似性的损失函数,充分利用标签信息。实验证明,GIFT方法在各种规模的数据集蒸馏中提升了性能。

🎯

关键要点

  • 最近的研究表明,使用预训练教师模型生成的软标签在数据集蒸馏中具有显著优势。

  • 本文强调充分利用标签信息的重要性。

  • 对用于数据集蒸馏的软标签损失函数进行了全面比较,发现模型对损失函数选择敏感。

  • 提出了一种新的方法GIFT,通过优化软标签和使用余弦相似性的损失函数来有效利用标签信息。

  • 实验证明,GIFT方法在各种规模的数据集蒸馏中提升了性能,且不增加额外计算成本。

  • 在ImageNet-1K上,GIFT方法在ConvNet和ResNet-18上分别提升了3.9%和1.8%的性能。

➡️

继续阅读