论文标题
使用胸部X射线图像探索Deep Covid-19分类的可解释性技术
Exploration of Interpretability Techniques for Deep COVID-19 Classification using Chest X-ray Images
论文作者
论文摘要
Covid-19的爆发以相当快的蔓延震惊了整个世界,并挑战了不同的部门。限制其扩散的最有效方法之一是早期,准确的诊断感染患者。诸如X射线和计算机断层扫描(CT)之类的医学成像结合了人工智能(AI)的潜力,在支持医务人员在诊断过程中起着至关重要的作用。因此,在本文中,五种不同的深度学习模型(resnet18,resnet34,Inceptionv3,InceptionResnetv2和densenet161)及其合奏,使用多数投票已用于使用胸部X射线图像进行分类。进行多标记分类,以预测每个患者的多种病理(如果存在)。首先,使用局部解释性方法对每个网络的可解释性进行了彻底研究 - 遮挡,显着性,输入X梯度,引导后反向传播,集成梯度和深度移植,并使用全球技术 - 神经元激活谱。 COVID-19分类模型的平均Micro-F1得分范围为0.66至0.875,网络模型集合的平均值为0.89。定性结果表明,重新结构是最容易解释的模型。这项研究表明,在决定最佳性能模型之前,使用可解释性方法比较不同的模型的重要性。
The outbreak of COVID-19 has shocked the entire world with its fairly rapid spread and has challenged different sectors. One of the most effective ways to limit its spread is the early and accurate diagnosing infected patients. Medical imaging, such as X-ray and Computed Tomography (CT), combined with the potential of Artificial Intelligence (AI), plays an essential role in supporting medical personnel in the diagnosis process. Thus, in this article five different deep learning models (ResNet18, ResNet34, InceptionV3, InceptionResNetV2 and DenseNet161) and their ensemble, using majority voting have been used to classify COVID-19, pneumoniæ and healthy subjects using chest X-ray images. Multilabel classification was performed to predict multiple pathologies for each patient, if present. Firstly, the interpretability of each of the networks was thoroughly studied using local interpretability methods - occlusion, saliency, input X gradient, guided backpropagation, integrated gradients, and DeepLIFT, and using a global technique - neuron activation profiles. The mean Micro-F1 score of the models for COVID-19 classifications ranges from 0.66 to 0.875, and is 0.89 for the ensemble of the network models. The qualitative results showed that the ResNets were the most interpretable models. This research demonstrates the importance of using interpretability methods to compare different models before making a decision regarding the best performing model.