系统仿真学报 ›› 2020, Vol. 32 ›› Issue (6): 1188-1194.doi: 10.16182/j.issn1004731x.joss.18-0712

• 国民经济仿真 • 上一篇    下一篇

基于卷积神经网络的X线胸片疾病分类研究

黄欣, 方钰, 顾梦丹   

  1. 同济大学电子信息工程学院计算机科学与技术系,上海 201804
  • 收稿日期:2018-10-25 修回日期:2019-03-11 出版日期:2020-06-25 发布日期:2020-06-25
  • 作者简介:黄欣(1984-),男,江西,博士生,研究方向为图像处理;方钰(1977-),女,上海,博士,教授,研究方向为机器学习;顾梦丹(1994-),女,广西,硕士生,研究方向为图像处理。
  • 基金资助:
    上海市科委项目(16511102800)

Classification of Chest X-ray Disease Based on Convolutional Neural Network

Huang Xin, Fang Yu, Gu Mengdan   

  1. Department of Computer Science and Technology, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, China
  • Received:2018-10-25 Revised:2019-03-11 Online:2020-06-25 Published:2020-06-25

摘要: 利用人工智能技术可以有效的辅助胸片的诊断。通过对中文X 线胸片检查报告文本挖掘,提出一种以基于异常部位的疾病分类标注方法,并整理出一个基于中文X 线胸片检查报告的疾病分类标注数据集。使用AlexNet ,VGGNet ,ResNet 及DenseNet 4 种不同的卷积神经网络,以及直接训练,ImageNet 预训练和ChestX-14 预训练3 种不同预训练方式对胸部疾病分类评估。结果显示更为复杂以及参数更多的卷积神经网络模型,在X 线胸片图像关键信息的获取方面有着更大的优势,同时采用大型X 线胸片数据集ChestX-14 预训练的模型效果要明显优于其他预训练。

关键词: X线胸片, 卷积神经网络, 胸部疾病, 检查报告

Abstract: The artificial intelligence technology can effectively assist the chest X-ray diagnosis. On the basis of the analysis of Chinese reports of chest X-rays, a labeling method of the thoracic disease classification for the chest abnormal parts is proposed and a dataset of the thoracic disease classification labels is complied. The thoracic disease classification is evaluated through four kinds of convolutional neural networks, AlexNet, VGGNet, ResNet and DenseNet and through three kinds of training methods, direct training, ImageNet pre-training and Chest X-14 pre-training. The result shows that the more complicated convolutional neural network with the more parameters, the better performance in obtaining the key information from the chest X-ray images can be. The model pre-trained by Chest X-14, a large dataset of chest X-ray, has the better result than the other pre-training methods.

Key words: Chest X-Rays, Convolutional Neural Network, Thoracic Disease, Reports

中图分类号: