【AAAI 2021】零样本知识蒸馏:Data-Free Knowledge Distillation with Soft Targeted Transfer Set Synthesis

本文提出了无数据知识蒸馏的新方法,通过建模教师模型的中间特征空间为多元正态分布,生成高质量的伪样本作为知识传输的载体。此方法在多个网络架构和数据集上优于现有工作,改善了模型压缩的效果。
摘要由CSDN通过智能技术生成

论文地址:

https://arxiv.org/abs/2104.04868

主要问题:

实现无数据 KD 的关键思想是生成信息丰富的伪样本,从而可以近似原始训练样本的分布

主要思路:

这篇文章提出了一种新的无数据 KD 方法,即利用多元正态分布对教师的中间特征空间进行建模,并利用该分布生成的软目标标签合成伪样本作为传输集

这样的方法与直接建模软最大空间相比,对较浅层的输出分布进行建模可以获得更广义的软目标

主要贡献:

  1. 作者用多元正态分布对教师中间层的特征空间进行建模,并对从该分布中采样的目标优化伪样本,从而提高了合成样本的质量
  2. 通过对较浅层的输出分布进行建模,而不是直接建模目标采样的软最大空间,从而获得更广义的软目标,这有助于提高性能
  3. 利用各种基准网络架构和数据集进行评估可以发现,该算法比现有工作有了明显的改进

具体实现:

基本符号:

假设知识蒸馏中,教师模型和学生模型的输出分别是:

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

BIT可达鸭

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值