本文通过运用知识蒸馏方面的知识,通过对源网络提取预训练的综合知识提取到一个更简单的克隆网络中训练,然后利用梯度得到精确的异常定位热图。
首先对于源网络,使用的是VGG16,将网络分为四层,分割点为最后四个最大池化层。
对于克隆网络,使用的是比源网络简单的网络,并且是无偏置网络。
损失函数有两部分组成。
源网络的激活值 克隆网络的激活值
(1)Lval:计算源网络和克隆网络激活值之间的欧几米德距离。
(2)Ldir:计算源网络和克隆网络激活值之间的方向相似性。
源网络vec()后任意维度变为一维的激活向量
优化器用的是Adam优化器,学习率为0.001,batch_size=64
论文复现代码