GIFT:在接近零成本的缩减数据集中挖掘标签的全部潜力
💡
原文中文,约500字,阅读约需1分钟。
📝
内容提要
最近的研究表明,使用预训练教师模型生成的软标签在数据集蒸馏中具有优势。本文提出了一种新的方法GIFT,通过优化软标签和使用余弦相似性的损失函数,充分利用标签信息。实验证明,GIFT方法在各种规模的数据集蒸馏中提升了性能。
🎯
关键要点
-
最近的研究表明,使用预训练教师模型生成的软标签在数据集蒸馏中具有显著优势。
-
本文强调充分利用标签信息的重要性。
-
对用于数据集蒸馏的软标签损失函数进行了全面比较,发现模型对损失函数选择敏感。
-
提出了一种新的方法GIFT,通过优化软标签和使用余弦相似性的损失函数来有效利用标签信息。
-
实验证明,GIFT方法在各种规模的数据集蒸馏中提升了性能,且不增加额外计算成本。
-
在ImageNet-1K上,GIFT方法在ConvNet和ResNet-18上分别提升了3.9%和1.8%的性能。
➡️