文章使用了DD更新的方式,就是先使用蒸馏数据集训练一个模型,然后计算真实数据在这个模型上的损失,更新蒸馏数据集。
文章的做法是:在训练蒸馏数据集网络时,加入了attention损失
这时候生成数据集不仅仅包含原始数据x
和y
,还包含了a
,这是attention模块的输出,作者只取了[CSL]
模块的输出。
之后使用蒸馏数据集训练模型时,不仅需要x,y的预测损失,还需要加入[cls]
的损失。
文章使用了DD更新的方式,就是先使用蒸馏数据集训练一个模型,然后计算真实数据在这个模型上的损失,更新蒸馏数据集。
文章的做法是:在训练蒸馏数据集网络时,加入了attention损失
这时候生成数据集不仅仅包含原始数据x
和y
,还包含了a
,这是attention模块的输出,作者只取了[CSL]
模块的输出。
之后使用蒸馏数据集训练模型时,不仅需要x,y的预测损失,还需要加入[cls]
的损失。
版权声明:
本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。
我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com