AX-Unet: A Deep Learning Framework for Image Segmentation to Assist Pancreatic Tumor Diagnosis

Minqiang Yang, Yuhong Zhang, Haoning Chen, Wei Wang, Haixu Ni*, Xinlong Chen, Zhuoheng Li, Chengsheng Mao

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Image segmentation plays an essential role in medical imaging analysis such as tumor boundary extraction. Recently, deep learning techniques have dramatically improved performance for image segmentation. However, an important factor preventing deep neural networks from going further is the information loss during the information propagation process. In this article, we present AX-Unet, a deep learning framework incorporating a modified atrous spatial pyramid pooling module to learn the location information and to extract multi-level contextual information to reduce information loss during downsampling. We also introduce a special group convolution operation on the feature map at each level to achieve information decoupling between channels. In addition, we propose an explicit boundary-aware loss function to tackle the blurry boundary problem. We evaluate our model on two public Pancreas-CT datasets, NIH Pancreas-CT dataset, and the pancreas part in medical segmentation decathlon (MSD) medical dataset. The experimental results validate that our model can outperform the state-of-the-art methods in pancreas CT image segmentation. By comparing the extracted feature output of our model, we find that the pancreatic region of normal people and patients with pancreatic tumors shows significant differences. This could provide a promising and reliable way to assist physicians for the screening of pancreatic tumors.

Original languageEnglish (US)
Article number894970
JournalFrontiers in Oncology
Volume12
DOIs
StatePublished - Jun 2 2022

Keywords

  • atrous spatial pyramid pooling
  • boundary-aware loss function
  • group convolution
  • image segmentation
  • pancreas CT

ASJC Scopus subject areas

  • Oncology
  • Cancer Research

Fingerprint

Dive into the research topics of 'AX-Unet: A Deep Learning Framework for Image Segmentation to Assist Pancreatic Tumor Diagnosis'. Together they form a unique fingerprint.

Cite this