知识蒸馏论文的一些自己总结

本文概述了知识蒸馏在CNN注意力转移和目标检测任务中的应用。通过教师网络引导,学生网络学习注意力图和特征图,以提高性能。文章讨论了基于响应图和梯度图的注意力转移方法,并介绍了特征蒸馏在目标检测任务中的应用,包括分类的交叉熵损失。
摘要由CSDN通过智能技术生成

本文只是对论文的一些总结,不关注于具体细节公式,主要是了解思想,如果有空我会专门的开博客对一些论文进行细节讲述,如果本文说明有问题的,欢迎联系我,欢迎指正。

一. Attention Transfer

Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfer:https://arxiv.org/abs/1612.03928

该文章启发于CNN中的注意力机制,利用teacher模型的中间层生成空间注意力图(spatial-attention),是一种热力图,图像中越感兴趣的地区颜色越红。

 

和大部分的知识蒸馏方式相同,教师网络通过生成的注意力图(attentior map)来指导学生网络的注意力图学习,通过attention transfer 缩小学生attentior map 和教师网络attentior map 的距离使得学生网络学习到教师网络的注意力图。这样学生网络学习到了这些知识,便能够生成尽可能与教师网络相似的特征图。

这边提一点,在学习中间层特征图,不管是热力图,注意力图还是普通的特征图,两个网络在这个地方的大小是一致的,这样才能通过距

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值