Adaptive cascaded transformer U-Net for MRI brain tumor segmentation

© 2024 Institute of Physics and Engineering in Medicine..

OBJECTIVE: Brain tumor segmentation on magnetic resonance imaging (MRI) plays an important role in assisting the diagnosis and treatment of cancer patients. Recently, cascaded U-Net models have achieved excellent performance via conducting coarse-to-fine segmentation of MRI brain tumors. However, they still suffer from obvious global and local differences among various brain tumors, which are difficult to solve with conventional convolutions.

APPROACH: To address the issue, this work proposes a novel Adaptive Cascaded Transformer U-Net (ACTransU-Net) for MRI brain tumor segmentation, which simultaneously integrates Transformer and dynamic convolution into a single cascaded U-Net architecture to adaptively capture global information and local details of brain tumors. ACTransU-Net first cascades two 3D U-Nets into a two-stage network to segment brain tumors from coarse to fine. Subsequently, it integrates omni-dimensional dynamic convolution modules into the second-stage shallow encoder and decoder, thereby enhancing the local detail representation of various brain tumors through dynamically adjusting convolution kernel parameters. Moreover, 3D Swin-Transformer modules are introduced into the second-stage deep encoder and decoder to capture image long-range dependencies, which helps adapt the global representation of brain tumors.

MAIN RESULTS: Extensive experiment results evaluated on the public BraTS 2020 and BraTS 2021 brain tumor datasets demonstrate the effectiveness of ACTransU-Net, with average DSC of 84.96% and 91.37%, and HD95 of 10.81 mm and 7.31 mm, proving competitiveness with the state-of-the-art methods.

SIGNIFICANCE: The proposed method focuses on adaptively capturing both global information and local details of brain tumors, aiding physicians in their accurate diagnosis. Additionally, it has the potential to extend ACTransU-Net for segmenting other types of lesions. The source code is available at: https://github.com/chenbn266/ACTransUnet.

Medienart:

E-Artikel

Erscheinungsjahr:

2024

Erschienen:

2024

Enthalten in:

Zur Gesamtaufnahme - year:2024

Enthalten in:

Physics in medicine and biology - (2024) vom: 18. Apr.

Sprache:

Englisch

Beteiligte Personen:

Chen, Bonian [VerfasserIn]
Sun, Qiule [VerfasserIn]
Han, Yutong [VerfasserIn]
Liu, Bin [VerfasserIn]
Zhang, Jianxin [VerfasserIn]
Zhang, Qiang [VerfasserIn]

Links:

Volltext

Themen:

Brain tumor segmentation
Cascaded network
Dynamic convolution
Journal Article
Magnetic resonance imaging
Transformer
U-Net

Anmerkungen:

Date Revised 18.04.2024

published: Print-Electronic

Citation Status Publisher

doi:

10.1088/1361-6560/ad4081

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM371254876