中文

XNOR-Net:使用二值卷积神经网络的ImageNet分类

计算机视觉与模式识别 2016-08-04 v4

摘要

我们提出了标准卷积神经网络的两种高效近似:二值权重网络(Binary-Weight-Networks)和XNOR-Networks。在二值权重网络中,滤波器用二值近似,从而节省32倍内存。在XNOR-Networks中,滤波器和卷积层的输入均为二值。XNOR-Networks主要使用二值运算近似卷积。这带来了58倍的卷积运算加速和32倍内存节省。XNOR-Nets提供了在CPU(而非GPU)上实时运行最先进网络的可能性。我们的二值网络简单、准确、高效,并能在具有挑战性的视觉任务上工作。我们在ImageNet分类任务上评估我们的方法。AlexNet的二值权重网络版本的分类精度仅比全精度AlexNet低2.9%(以top-1衡量)。我们将我们的方法与最近的网络二值化方法BinaryConnect和BinaryNets比较,并在ImageNet上大幅优于这些方法,top-1精度高出16%以上。

关键词

引用

@article{arxiv.1603.05279,
  title  = {XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networks},
  author = {Mohammad Rastegari and Vicente Ordonez and Joseph Redmon and Ali Farhadi},
  journal= {arXiv preprint arXiv:1603.05279},
  year   = {2016}
}