论文标题
完全连接的层与视网膜图像分析的类数量之间的关系
The relationship between Fully Connected Layers and number of classes for the analysis of retinal images
论文作者
论文摘要
本文实验了深度卷积神经网络中完全连接的层的数量,该层应用于眼底视网膜图像的分类。分析的图像对应于ODIR 2019(北京大学国际眼疾病智能识别竞赛)[9],其中包括各种眼部疾病的图像(白内障,青光眼,近视,糖尿病性视网膜病,与年龄相关的黄斑变性(AMD),高血压)以及正常情况。这项工作的重点是对正常,白内障,AMD和近视的分类。在更改网络的特征映射(线性)部分时,神经网络的特征提取(卷积)部分保持相同。在这些神经网上还探讨了不同的数据集。每个数据集与另一个类别的类别不同。因此,本文旨在找到类数量与完全连接层数之间的关系。人们发现,增加神经网络完全连接层数的效果取决于所使用的数据集类型。对于简单,线性可分离的数据集,添加完全连接的层是应该探索的东西,并且可能会导致更好的训练准确性,但是找不到直接相关性。但是,随着数据集的复杂性上升(更多重叠的类),增加了完全连接的层的数量会导致神经网络停止学习。这种现象发生的情况越快,数据集越复杂。
This paper experiments with the number of fully-connected layers in a deep convolutional neural network as applied to the classification of fundus retinal images. The images analysed corresponded to the ODIR 2019 (Peking University International Competition on Ocular Disease Intelligent Recognition) [9], which included images of various eye diseases (cataract, glaucoma, myopia, diabetic retinopathy, age-related macular degeneration (AMD), hypertension) as well as normal cases. This work focused on the classification of Normal, Cataract, AMD and Myopia. The feature extraction (convolutional) part of the neural network is kept the same while the feature mapping (linear) part of the network is changed. Different data sets are also explored on these neural nets. Each data set differs from another by the number of classes it has. This paper hence aims to find the relationship between number of classes and number of fully-connected layers. It was found out that the effect of increasing the number of fully-connected layers of a neural networks depends on the type of data set being used. For simple, linearly separable data sets, addition of fully-connected layer is something that should be explored and that could result in better training accuracy, but a direct correlation was not found. However as complexity of the data set goes up(more overlapping classes), increasing the number of fully-connected layers causes the neural network to stop learning. This phenomenon happens quicker the more complex the data set is.