arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.03809cs.CV
收藏

PSGR:用于CT图像中COVID-19肺炎分割的逐像素稀疏图推理

PSGR: Pixel-wise Sparse Graph Reasoning for COVID-19 Pneumonia Segmentation in CT Images

Northwestern Polytechnical University(西北工业大学);Intel Labs(英特尔实验室);The University of Sydney(悉尼大学);University of Pittsburgh(匹兹堡大学);JD Finance America Corporation(京东金融美国公司)

Haozhe Jia, Haoteng Tang, Guixiang Ma, Weidong Cai, Heng Huang, Liang Zhan, Yong Xia

摘要:

计算机断层扫描(CT)图像中感染区域的自动准确分割对于预测COVID-19的病理阶段和治疗反应至关重要。几个深度卷积神经网络(DCNN)已被设计用于这一任务,其性能,然而,往往被其有限的局部感受野和不足的全局推理能力抑制。在本文中,我们提出了一个逐像素稀疏图推理(PSGR)模块,并将其插入到分割网络中,以增强CT图像中COVID-19感染区域分割的长程依赖关系建模。在PSGR模块中,首先通过基于分割主干产生的特征将每个像素投影到节点上来构建图,然后通过仅保留与每个不确定像素的K个最强连接来将其转换为稀疏连接图。在稀疏连接图上进行长距离信息推理,生成增强特征。该模块的优点有两个方面:(1)逐像素映射策略不仅避免了不精确的像素到节点投影,而且保留了每个像素的固有信息用于全局推理;以及(2)稀疏连接图构造导致有效的信息检索和噪声传播的减少。在三个公共数据集上对四种广泛使用的分割模型进行了评估。结果表明,配备PSGR模块的分割模型可以有效地分割CT图像中的COVID-19感染区域,优于所有其他竞争模型。

英文摘要:

Automated and accurate segmentation of the infected regions in computed tomography (CT) images is critical for the prediction of the pathological stage and treatment response of COVID-19. Several deep convolutional neural networks (DCNNs) have been designed for this task, whose performance, however, tends to be suppressed by their limited local receptive fields and insufficient global reasoning ability. In this paper, we propose a pixel-wise sparse graph reasoning (PSGR) module and insert it into a segmentation network to enhance the modeling of long-range dependencies for COVID-19 infected region segmentation in CT images. In the PSGR module, a graph is first constructed by projecting each pixel on a node based on the features produced by the segmentation backbone, and then converted into a sparsely-connected graph by keeping only K strongest connections to each uncertain pixel. The long-range information reasoning is performed on the sparsely-connected graph to generate enhanced features. The advantages of this module are two-fold: (1) the pixel-wise mapping strategy not only avoids imprecise pixel-to-node projections but also preserves the inherent information of each pixel for global reasoning; and (2) the sparsely-connected graph construction results in effective information retrieval and reduction of the noise propagation. The proposed solution has been evaluated against four widely-used segmentation models on three public datasets. The results show that the segmentation model equipped with our PSGR module can effectively segment COVID-19 infected regions in CT images, outperforming all other competing models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑