计算机应用 ›› 2020, Vol. 40 ›› Issue (10): 2904-2909.DOI: 10.11772/j.issn.1001-9081.2020020192

• 人工智能 • 上一篇    下一篇

基于深度混合卷积模型的肺结节检测方法

戚永军1,2,3, 顾军华1,2,4,5, 张亚娟4,5, 王锋4, 田泽培4   

  1. 1. 省部共建电工装备可靠性与智能化国家重点实验室(河北工业大学), 天津 300132;
    2. 河北省电磁场与电器可靠性重点实验室(河北工业大学), 天津 300132;
    3. 北华航天工业学院 信息技术中心, 河北 廊坊 065000;
    4. 河北工业大学 人工智能与数据科学学院, 天津 300401;
    5. 河北省大数据计算重点实验室(河北工业大学), 天津 300401
  • 收稿日期:2020-02-26 修回日期:2020-05-22 出版日期:2020-10-10 发布日期:2020-05-27
  • 通讯作者: 顾军华
  • 作者简介:戚永军(1976-),男,河北唐山人,高级工程师,博士研究生,主要研究方向:计算机视觉、深度学习;顾军华(1966-),男,河北赵县人,教授,博士,CCF会员,主要研究方向:智能信息处理、数据挖掘;张亚娟(1984-),女,河北廊坊人,高级实验师,硕士,主要研究方向:数据挖掘、图像处理;王锋(1995-),男,河北保定人,硕士研究生,主要研究方向:医学图像处理、图像检索;田泽培(1994-),女,河北石家庄人,硕士研究生,主要研究方向:医学图像处理、目标检测。
  • 基金资助:
    国家自然科学基金资助项目(61702157);河北省创新能力提升计划项目(199676146H);北华航天工业学院青年基金资助项目(KY202022)。

Deep mixed convolution model for pulmonary nodule detection

QI Yongjun1,2,3, GU Junhua1,2,4,5, ZHANG Yajuan4,5, WANG Feng4, TIAN Zepei4   

  1. 1. State Key Laboratory of Reliability and Intelligence of Electrical Equipment;(Hebei University of Technology), Tianjin 300132, China;
    2. Hebei Key Laboratory of Electromagnetic Field and Electrical Apparatus Reliability;(Hebei University of Technology), Tianjin 300132, China;
    3. Information Technology Center, North China Institute of Aerospace Engineering, Langfang Hebei 065000, China;
    4. School of Artificial Intelligence, Hebei University of Technology, Tianjin 300401, China;
    5. Hebei Key Laboratory of Big Data Calculation(Hebei University of Technology), Tianjin 300401, China
  • Received:2020-02-26 Revised:2020-05-22 Online:2020-10-10 Published:2020-05-27
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (61702157), the Innovation Capacity Improvement Program of Hebei Province (199676146H), the Youth Foundation of North China Institute of Aerospace Engineering (KY202022).

摘要: 基于高维肺部计算机断层扫描(CT)图像的肺结节检测是一项极具挑战性的任务。在诸多肺结节检测算法中,深度卷积神经网络(CNN)最引人注目,其中二维(2D) CNN具有预训练模型多、检测效率高等优点,应用非常广泛,但肺结节本质是三维(3D)病灶,2D CNN会不可避免地造成信息损失,从而影响检测精度。3D CNN能充分利用CT图像空间信息,有效提升检测精度,但是3D CNN存在参数多、计算消耗大、过拟合风险高等不足。为了兼顾两者的优势,提出基于深度混合CNN的肺结节检测模型,通过在神经网络模型的浅层部署3D CNN,在模型的深层部署2D CNN,并增加反卷积模块,融合了多层级的图像特征,达到了在不损失检测精度的情况下减少模型参数、增强模型泛化能力,提高检测效率的目的。在LUNA16数据集上的实验结果表明,所提出的模型在平均每次扫描8个假阳性的情况下的敏感度为0.924,优于现有的先进模型。

关键词: 深度学习, 特征融合, 肺结节检测, 计算机辅助诊断, CT图像

Abstract: Pulmonary nodule detection is a very challenging task based on high-dimensional lung Computed Tomography (CT) images. Among many pulmonary nodule detection algorithms, the deep Convolutional Neural Network (CNN) is the most attractive one. In this kind of networks, the Two-Dimensional (2D) CNNs with many pre-trained models and high detection efficiency are widely used. However, the nature of pulmonary nodules is the Three-Dimensional (3D) lesion, so that the 2D CNNs will inevitably cause information loss and thereby affect the detection accuracy. The 3D CNNs can make full use of the spatial information of CT images and effectively improve the detection accuracy, but the 3D CNNs have shortcomings such as many parameters, large calculation consumption and high risk of over fitting. In order to take the advantages of the two networks, a pulmonary nodule detection model based on a deep mixed CNN was proposed. By deploying 3D CNN in the shallow layer of the neural network model and 2D CNN in the deep layer of the model, and adding a deconvolution module to fuse multi-layer image features together, the model parameters were reduced and the generalization ability and the detection efficiency of the model were improved without decreasing the detection accuracy. Experimental results on LUNA16 dataset show that the proposed model has the sensitivity reached 0.924 under the condition of average 8 false positives per scan, which outperforms the existing state-of-the-art models.

Key words: deep learning, feature fusion, pulmonary nodule detection, Computer Aided Diagnosis (CAD), Computed Tomography (CT) image

中图分类号: