Low-contrast medical image segmentation via transformer and boundary perception

Yinglin Zhang, Ruiling Xi, Wei Wang, Heng Li, Lingxi Hu, Huiyan Lin, Dave Towey, Ruibin Bai, Huazhu Fu, Risa Higashita, Jiang Liu

Research output: Journal PublicationArticlepeer-review

1 Citation (Scopus)

Abstract

Low-contrast medical image segmentation is a challenging task that requires full use of local details and global context. However, existing convolutional neural networks (CNNs) cannot fully exploit global information due to limited receptive fields and local weight sharing. On the other hand, the transformer effectively establishes long-range dependencies but lacks desirable properties for modeling local details. This paper proposes a Transformer-embedded Boundary perception Network (TBNet) that combines the advantages of transformer and convolution for low-contrast medical image segmentation. Firstly, the transformer-embedded module uses convolution at the low-level layer to model local details and uses the Enhanced TRansformer (ETR) to capture long-range dependencies at the high-level layer. This module can extract robust features with semantic contexts to infer the possible target location and basic structure in low-contrast conditions. Secondly, we utilize the decoupled body-edge branch to promote general feature learning and precept precise boundary locations. The ETR establishes long-range dependencies across the whole feature map range and is enhanced by introducing local information. We implement it in a parallel mode, i.e., the group of self-attention with multi-head captures the global relationship, and the group of convolution retains local details. We compare TBNet with other state-of-the-art (SOTA) methods on the cornea endothelial cell, ciliary body, and kidney segmentation tasks. The TBNet improves segmentation performance, proving its effectiveness and robustness.

Original languageEnglish
Pages (from-to)1-13
Number of pages13
JournalIEEE Transactions on Emerging Topics in Computational Intelligence
Volume8
Issue number3
DOIs
Publication statusPublished - Jun 2024

Keywords

  • Low-contrast medical image
  • transformer
  • segmentation
  • deep learning

ASJC Scopus subject areas

  • Computer Science Applications
  • Control and Optimization
  • Computational Mathematics
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Low-contrast medical image segmentation via transformer and boundary perception'. Together they form a unique fingerprint.

Cite this