国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:何新宇, 张晓龙
单位:1. 武汉科技大学 计算机科学与技术学院, 武汉 430065;2. 武汉科技大学 大数据科学与工程研究院, 武汉 430065;3. 智能信息处理与实时工业系统湖北省重点实验室, 武汉 430065
关键词:肺炎图像分类,迁移学习,深度卷积神经网络,随机森林,敏感度,特异度
基金:国家自然科学基金资助项目(61273225,61702381)。
当前的肺炎图像识别算法面临两个问题:一是肺炎特征提取器使用的迁移学习模型在源数据集与肺炎数据集上图像差异较大,所提取的特征不能很好地契合肺炎图像;二是算法使用的softmax分类器对高维特征处理能力不够强,在识别准确率上仍有提升的空间。针对这两个问题,提出了一种基于深度卷积神经网络的肺炎图像识别模型。首先使用ImageNet数据集训练好的GoogLeNet Inception V3网络模型进行特征提取;其次,增加了特征融合层,使用随机森林分类器进行分类预测。实验在Chest X-Ray Images肺炎标准数据集上进行。实验结果表明,该模型的识别准确率、敏感度、特异度的值分别达到96.77%、97.56%、94.26%。在识别准确率以及敏感度指标上,与经典的GoogLeNet Inception V3+Data Augmentation (GIV+DA)算法相比,所提模型分别提高了1.26、1.46个百分点,在特异度指标上已接近GIV+DA算法的最优结果。
来源:2019年第6期
《计算机应用》期刊编辑部