Adaptive cascaded transformer U-Net for MRI brain tumor segmentation

Phys Med Biol. 2024 Apr 18. doi: 10.1088/1361-6560/ad4081. Online ahead of print.ABSTRACTOBJECTIVE: 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 pu...
Source: Physics in Medicine and Biology - Category: Physics Authors: Source Type: research