天天看点

Google开源Inception-ResNet-v2,提升图像分类水准

2016年8月31日,google团队宣布针对tensorflow开源了最新发布的tf-slim资料库,它是一个可以定义、训练和评估模型的轻量级的软件包,也能对图像分类领域中几个主要有竞争力的网络进行检验和定义模型。

为了进一步推进这个领域的进步,今天google团队宣布发布inception-resnet-v2(一种卷积神经网络——cnn),它在ilsvrc图像分类基准测试中实现了当下最好的成绩。inception-resnet-v2是早期inception v3模型变化而来,从微软的残差网络(resnet)论文中得到了一些灵感。相关论文信息可以参看我们的论文inception-v4, inception-resnet and the impact of residual connections on learning(inception-v4, inception-resnet以及残差连接在学习上的影响):

Google开源Inception-ResNet-v2,提升图像分类水准

残差连接(residual connections )允许模型中存在shortcuts,可以让研究学者成功地训练更深的神经网络(能够获得更好的表现),这样也能明显地简化inception块。将两种模型架构对比,见下图:

Google开源Inception-ResNet-v2,提升图像分类水准
Google开源Inception-ResNet-v2,提升图像分类水准

在第二幅inception-resnet-v2图中最上部分,你能看到整个网络扩展了。注意该网络被认为比先前的inception v3还要深一些。在图中主要部分重复的残差区块已经被压缩了,所以整个网络看起来更加直观。另外注意到图中inception区块被简化了,比先前的inception v3种要包含更少的并行塔 (parallel towers)。

如下方图表所示,inception-resnet-v2架构的精确度比之前的最优模型更高,图表中所示为基于单个图像的ilsvrc 2012图像分类标准得出的排行第一与排行第五的有效精确度。此外,该新模型仅仅要求两倍于inception v3的容量与计算能力。

Google开源Inception-ResNet-v2,提升图像分类水准

例如,尽管inception v3与inception-resnet-v2都很擅长识别每一条狗所属的类别,这种新模型的性能更为显著。例如,旧模型可能会错误地将右边的图片识别为阿拉斯加雪橇犬,新模型inception-resnet-v2能够准确地识别两个图片中狗的类别。

Google开源Inception-ResNet-v2,提升图像分类水准

阿拉斯加雪橇犬(左)和西伯利亚雪橇犬(右)。图片来源:维基百科。

为了使人们理解开始试验,我们也将要发布新型inception-resnet-v2预训练实例,作为tf-slim 图像模型库的组成部分。

看到本研究领域关于这一经过改进的模型所取得的进展,而且人们开始采用这种新模型,并将其性能在多种任务作对比,我们为此感到非常兴奋。你也想要开始使用这种新模型吗?让我们一起来看看附带的操作指示,学会如何训练,评估或微调一个网络。

继续阅读