vix.ing · top · new · best · stats

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

2020/07/02 by Ziqiang Li, Li, Ziqiang, Hong Pan +5
Computer Science · Engineering · Medicine · #Advanced Neural Network Applications #Artificial intelligence #Artificial neural network #Computer Vision and Pattern Recognition (cs.CV) #Computer science #Computer vision #FOS: Computer and information sciences #FOS: Electrical engineering #Image (mathematics) #Image and Video Processing (eess.IV) #Medical Image Segmentation Techniques #Neural and Evolutionary Computing (cs.NE) #Noise (video) #Overfitting #Pattern recognition (psychology) #Pooling #Position (finance) #Process (computing) #Radiomics and Machine Learning in Medical Imaging #Segmentation #Task (project management) #cs.CV #cs.NE #eess.IV #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2007.01001

published in arXiv (Cornell University) (Cornell University) · Accepted by the 2020 International Joint Conference on Neural Networks (IJCNN 2020)

arxiv created 2020/07/02 · openalex publication_date 2020/07/02 · arxiv updated 2020/07/03 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

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.

Citations

Related