论文标题

弱监督的疾病定位分割胸部X射线图像

Weakly-Supervised Segmentation for Disease Localization in Chest X-Ray Images

论文作者

Viniavskyi, Ostap, Dobko, Mariia, Dobosevych, Oles

论文摘要

深度卷积神经网络已被证明有效地解决了语义分割的任务。但是,它们的效率在很大程度上取决于像素级注释,这些注释昂贵,并且通常需要域专业知识,尤其是在医学成像中。弱监督的语义细分有助于克服这些问题,并提供可解释的深度学习模型。在本文中,我们提出了一种新颖的方法,用于将医学胸部X射线图像的语义分割,仅以图像级别的类标签作为监督。我们通过将三种方法与连续步骤相结合,提高了疾病定位的精度。首先,我们通过使用正则化程序增强的监督分类模型在训练图像中生成异常区域中异常区域的伪分割标签。然后将获得的激活图进行后处理并传播到第二个分类模型 - 像素关系网络,从而改善了不同对象类之间的边界。最后,所得的伪标签用于训练提出的完全监督的分割模型。我们分析了提出方法的鲁棒性,并在两个不同的数据集上测试了其性能:Pascal VOC 2012和Siim-Acr Pneumothorax。我们仅使用图像级注释在两个数据集的分割中取得了重大结果。我们表明,这种方法适用于胸部X射线,用于检测肺和胸壁之间胸膜空间中的空气异常。我们的代码已公开可用。

Deep Convolutional Neural Networks have proven effective in solving the task of semantic segmentation. However, their efficiency heavily relies on the pixel-level annotations that are expensive to get and often require domain expertise, especially in medical imaging. Weakly supervised semantic segmentation helps to overcome these issues and also provides explainable deep learning models. In this paper, we propose a novel approach to the semantic segmentation of medical chest X-ray images with only image-level class labels as supervision. We improve the disease localization accuracy by combining three approaches as consecutive steps. First, we generate pseudo segmentation labels of abnormal regions in the training images through a supervised classification model enhanced with a regularization procedure. The obtained activation maps are then post-processed and propagated into a second classification model-Inter-pixel Relation Network, which improves the boundaries between different object classes. Finally, the resulting pseudo-labels are used to train a proposed fully supervised segmentation model. We analyze the robustness of the presented method and test its performance on two distinct datasets: PASCAL VOC 2012 and SIIM-ACR Pneumothorax. We achieve significant results in the segmentation on both datasets using only image-level annotations. We show that this approach is applicable to chest X-rays for detecting an anomalous volume of air in the pleural space between the lung and the chest wall. Our code has been made publicly available.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源