中文

基于知识蒸馏的任务特定场景可扩展卷积神经网络

计算机视觉与模式识别 2017-01-09 v2

摘要

在本文中,我们探讨了卷积神经网络中随视觉任务复杂度而变化的冗余。考虑到许多前端视觉系统仅对有限范围的视觉目标感兴趣,去除任务特定的网络冗余可以促进广泛的潜在应用。我们提出了一种任务特定的知识蒸馏算法,以导出一个具有预设计算成本且精度损失最小化的简化模型,该模型非常适合资源受限的前端系统。在 MNIST 和 CIFAR10 数据集上的实验证明了所提方法的可行性以及任务特定冗余的存在。

关键词

引用

@article{arxiv.1609.05695,
  title  = {A scalable convolutional neural network for task-specified scenarios via knowledge distillation},
  author = {Mengnan Shi and Fei Qin and Qixiang Ye and Zhenjun Han and Jianbin Jiao},
  journal= {arXiv preprint arXiv:1609.05695},
  year   = {2017}
}