论文标题

PGD​​-UNET:一个位置引导的可变形网络,用于同时分割器官和肿瘤

PGD-UNet: A Position-Guided Deformable Network for Simultaneous Segmentation of Organs and Tumors

论文作者

Li, Ziqiang, Pan, Hong, Zhu, Yaping, Qin, A. K.

论文摘要

器官和肿瘤的精确分割在临床应用中起着至关重要的作用。由于不规则的形状和各种尺寸的器官和肿瘤以及感兴趣的解剖结构(AOI)和背景区域之间的显着类不平衡,这是一项具有挑战性的任务。此外,在大多数情况下,肿瘤和正常器官通常在医学图像中重叠,但是当前的方法无法准确描绘肿瘤和器官。为了应对此类挑战,我们提出了一种位置引导的可变形UNET,即PGD-UNET,该挑战利用了可变形卷积的空间变形能力来处理器官和肿瘤的几何变化。位置信息明确编码到网络中,以增强变形的功能。同时,我们引入了一个新的池化模块,以保留传统的最大通行操作中丢失的位置信息。此外,由于不同结构之间的界限以及注释的主观性不清,标签不一定对于医疗图像分割任务而言是准确的。由于标签噪声,这可能会导致训练有素的网络过度拟合。为了解决这个问题,我们制定了一种新颖的损失功能,以抑制潜在标签噪声对训练过程的影响。我们的方法对两个具有挑战性的分割任务进行了评估,并在这两个任务中都达到了非常有希望的分割精度。

Precise segmentation of organs and tumors plays a crucial role in clinical applications. It is a challenging task due to the irregular shapes and various sizes of organs and tumors as well as the significant class imbalance between the anatomy of interest (AOI) and the background region. In addition, in most situation tumors and normal organs often overlap in medical images, but current approaches fail to delineate both tumors and organs accurately. To tackle such challenges, we propose a position-guided deformable UNet, namely PGD-UNet, which exploits the spatial deformation capabilities of deformable convolution to deal with the geometric transformation of both organs and tumors. Position information is explicitly encoded into the network to enhance the capabilities of deformation. Meanwhile, we introduce a new pooling module to preserve position information lost in conventional max-pooling operation. Besides, due to unclear boundaries between different structures as well as the subjectivity of annotations, labels are not necessarily accurate for medical image segmentation tasks. It may cause the overfitting of the trained network due to label noise. To address this issue, we formulate a novel loss function to suppress the influence of potential label noise on the training process. Our method was evaluated on two challenging segmentation tasks and achieved very promising segmentation accuracy in both tasks.

扫码加入交流群

加入微信交流群

微信交流群二维码

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