Nasser Adnane Ait, Akhloufi Moulay A
Perception, Robotics, and Intelligent Machines (PRIME), Université de Moncton, Moncton, NB E1C 3E9 Canada.
SN Comput Sci. 2023;4(4):388. doi: 10.1007/s42979-023-01818-w. Epub 2023 May 11.
X-ray images are the most widely used medical imaging modality. They are affordable, non-dangerous, accessible, and can be used to identify different diseases. Multiple computer-aided detection (CAD) systems using deep learning (DL) algorithms were recently proposed to support radiologists in identifying different diseases on medical images. In this paper, we propose a novel two-step approach for chest disease classification. The first is a multi-class classification step based on classifying X-ray images by infected organs into three classes (normal, lung disease, and heart disease). The second step of our approach is a binary classification of seven specific lungs and heart diseases. We use a consolidated dataset of 26,316 chest X-ray (CXR) images. Two deep learning methods are proposed in this paper. The first is called DC-ChestNet. It is based on ensembling deep convolutional neural network (DCNN) models. The second is named VT-ChestNet. It is based on a modified transformer model. VT-ChestNet achieved the best performance overcoming DC-ChestNet and state-of-the-art models (DenseNet121, DenseNet201, EfficientNetB5, and Xception). VT-ChestNet obtained an area under curve (AUC) of 95.13% for the first step. For the second step, it obtained an average AUC of 99.26% for heart diseases and an average AUC of 99.57% for lung diseases.
X射线图像是使用最广泛的医学成像方式。它们价格低廉、无危险、易于获取,可用于识别不同疾病。最近提出了多个使用深度学习(DL)算法的计算机辅助检测(CAD)系统,以支持放射科医生在医学图像上识别不同疾病。在本文中,我们提出了一种新颖的两步法用于胸部疾病分类。第一步是基于将X射线图像按受感染器官分类为三类(正常、肺部疾病和心脏病)的多类分类步骤。我们方法的第二步是对七种特定的肺部和心脏病进行二元分类。我们使用了一个包含26316张胸部X射线(CXR)图像的综合数据集。本文提出了两种深度学习方法。第一种称为DC-ChestNet,它基于集成深度卷积神经网络(DCNN)模型。第二种名为VT-ChestNet,它基于改进的变压器模型。VT-ChestNet克服了DC-ChestNet和现有最先进模型(DenseNet121、DenseNet201、EfficientNetB5和Xception),取得了最佳性能。VT-ChestNet在第一步中获得了95.13%的曲线下面积(AUC)。在第二步中,它对心脏病获得了平均99.26%的AUC,对肺部疾病获得了平均99.57%的AUC。