国内刊号:31-1289/TP
国际刊号:1000-3428
发布日期:
作者:张驰名, 王庆凤, 刘志勤, 黄俊, 周莹, 刘启榆, 徐卫云
单位:1. 西南科技大学 计算机科学与技术学院, 四川 绵阳 621010;2. 绵阳市中心医院, 四川 绵阳 621010
关键词:迁移学习,卷积神经网络,医学图像分类,计算机辅助诊断,肺结节诊断
基金:四川省军民融合研究院开放基金(2017SCII0219,2017SCII0220);四川省科技创新苗子工程重大项目(19MZGC0123)。
在肺癌早期筛查过程中,人工诊断胸部CT扫描图像费时费力,而深度学习网络缺乏足够的医学数据进行训练。为此,提出一种渐进式微调(PFT)策略,将其应用于深度迁移学习网络以辅助诊断肺结节良恶性。利用神经网络在粗粒度的自然图像大数据集中学习特征知识,经重构网络分类层将所学到的特征信息迁移至肺结节的细粒度小数据集。采用PFT策略从全连接分类层开始,逐层释放、微调训练卷积层直至所有网络层,并通过定量分析各层微调后肺结节良恶性分类的AUC值,确定最佳微调深度。此外,采用梯度加权类激活映射图和t-SNE算法为网络预测结果提供相应的视觉支持与解释。在LIDC数据集中的实验结果表明,该方法对肺结节良恶性诊断的准确率可达91.44%,其AUC值为0.962 1。
来源:2020年第1期
《计算机工程》期刊编辑部