ALA-Net: Adaptive Lesion-Aware Attention Network for 3D Colorectal Tumor Segmentation

IEEE Trans Med Imaging. 2021 Dec;40(12):3627-3640. doi: 10.1109/TMI.2021.3093982. Epub 2021 Nov 30.

Abstract

Accurate and reliable segmentation of colorectal tumors and surrounding colorectal tissues on 3D magnetic resonance images has critical importance in preoperative prediction, staging, and radiotherapy. Previous works simply combine multilevel features without aggregating representative semantic information and without compensating for the loss of spatial information caused by down-sampling. Therefore, they are vulnerable to noise from complex backgrounds and suffer from misclassification and target incompleteness-related failures. In this paper, we address these limitations with a novel adaptive lesion-aware attention network (ALA-Net) which explicitly integrates useful contextual information with spatial details and captures richer feature dependencies based on 3D attention mechanisms. The model comprises two parallel encoding paths. One of these is designed to explore global contextual features and enlarge the receptive field using a recurrent strategy. The other captures sharper object boundaries and the details of small objects that are lost in repeated down-sampling layers. Our lesion-aware attention module adaptively captures long-range semantic dependencies and highlights the most discriminative features, improving semantic consistency and completeness. Furthermore, we introduce a prediction aggregation module to combine multiscale feature maps and to further filter out irrelevant information for precise voxel-wise prediction. Experimental results show that ALA-Net outperforms state-of-the-art methods and inherently generalizes well to other 3D medical images segmentation tasks, providing multiple benefits in terms of target completeness, reduction of false positives, and accurate detection of ambiguous lesion regions.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Colorectal Neoplasms* / diagnostic imaging
  • Humans
  • Image Processing, Computer-Assisted
  • Magnetic Resonance Imaging
  • Neural Networks, Computer*
  • Semantics