一种基于卷积神经网络VGG16的乳腺肿瘤分类算法

摘要

本发明涉及一种基于卷积神经网络VGG16的乳腺肿瘤分类算法，包括：数据预处理：对呈现数据不平衡状态的数据集，进行不平衡处理和数据增强处理。卷积神经网络的搭建：1)网络预训练：利用VGG16，在ImageNet大型自然图像数据集上进行网络训练，并保存训练之后的权重；2)网络关键节点选取：利用VGG16网络的不同层，在乳腺肿瘤DDSM数据集上进行特征的提取，对于提取的特征，应用相同的支持向量机SVM分类器分类，选取分类性能最高的层，作为新网络搭建的节点；3)在选取的网络搭建的节点后连接两层全连接和一层softmax形成新的网络；迁移学习。