arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.08562cs.LGcs.CVeess.IV

辅以全连接子网络的CNN中的集成学习

Ensemble learning in CNN augmented with fully connected subnetworks

Industrial Technology Center of Okayama Prefecture(冈山县工业技术中心);Okayama University(冈山大学)

Daiki Hirata, Norikazu Takahashi

AI总结:

本研究提出由基础CNN和多个独立训练的全连接子网络构成的EnsNet模型,通过划分卷积层特征图子集分配给子网络、多数投票整合输出,在多个数据集上提升了CNN性能,在MNIST上达0.16%的最优错误率。

摘要:

卷积神经网络(CNN)在通用目标识别任务中已展现出卓越性能。本文提出一种名为EnsNet的新型模型,它由一个基础CNN和多个全连接子网络(FCSN)构成。在该模型中,基础CNN最后一个卷积层生成的特征图集合沿通道被划分为互不相交的子集,这些子集被分配给各个FCSN。每个FCSN都独立于其他FCSN进行训练,从而能够根据分配给它的特征图子集预测类别标签。整个模型的输出由基础CNN与各FCSN的多数投票决定。在MNIST、Fashion-MNIST和CIFAR-10数据集上的实验结果表明,所提方法进一步提升了CNN的性能。具体而言,一个EnsNet在MNIST上实现了0.16%的当前最优错误率。

英文摘要:

Convolutional Neural Networks (CNNs) have shown remarkable performance in general object recognition tasks. In this paper, we propose a new model called EnsNet which is composed of one base CNN and multiple Fully Connected SubNetworks (FCSNs). In this model, the set of feature-maps generated by the last convolutional layer in the base CNN is divided along channels into disjoint subsets, and these subsets are assigned to the FCSNs. Each of the FCSNs is trained independent of others so that it can predict the class label from the subset of the feature-maps assigned to it. The output of the overall model is determined by majority vote of the base CNN and the FCSNs. Experimental results using the MNIST, Fashion-MNIST and CIFAR-10 datasets show that the proposed approach further improves the performance of CNNs. In particular, an EnsNet achieves a state-of-the-art error rate of 0.16% on MNIST.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑