中文

多注意力任务中基于注意力的知识蒸馏:DCT驱动损失的影响

计算机视觉与模式识别 2022-06-07 v2

摘要

知识蒸馏(Knowledge Distillation, KD)是一组用于定义可迁移通道以提升卷积神经网络效率的策略。基于特征的知识蒸馏是KD的一个子领域,它依赖中间网络表征(未经改动或通过最大激活图降维)作为源知识。在本文中,我们提出并分析了在迁移激活图之前对其施加二维频率变换的方法。我们提出——通过使用全局图像线索而非像素估计,该策略增强了在场景识别等任务中的知识可迁移性,此类任务由多个不同概念间强烈的空间与上下文关系所定义。为验证所提方法,我们给出了场景识别领域最先进方法的广泛评估。实验结果提供有力证据:所提策略使学生网络能更好地聚焦于教师网络学到的相关图像区域,从而产生更具描述性的特征,并实现优于其他所有最先进替代方案的迁移性能。我们在 http://www-vpu.eps.uam.es/publications/DCTBasedKDForSceneRecognition 公开了本文使用的训练与评估框架。

关键词

引用

@article{arxiv.2205.01997,
  title  = {Attention-based Knowledge Distillation in Multi-attention Tasks: The Impact of a DCT-driven Loss},
  author = {Alejandro López-Cifuentes and Marcos Escudero-Viñolo and Jesús Bescós and Juan C. SanMiguel},
  journal= {arXiv preprint arXiv:2205.01997},
  year   = {2022}
}

备注

Preprint under review in TCSVT Journal