3D asymmetric expectation-maximization attention network for brain tumor segmentation

Area of Science:

• Medical Imaging
• Artificial Intelligence
• Neuroscience

Background:

• Accurate brain tumor segmentation on MRI is crucial for diagnosis and treatment planning.
• 3D deep neural networks offer improved accuracy over 2D methods for brain tumor segmentation.
• Existing 3D models often face high computational demands.

Purpose of the Study:

• To introduce a novel 3D asymmetric expectation-maximization attention network (AEMA-Net) for automatic brain tumor segmentation.
• To enhance feature extraction and long-range context capture in brain tumor segmentation models.
• To address the high computational cost associated with 3D segmentation models.

Main Methods:

• Developed AEMA-Net, an encoder-decoder neural network modifying the 3D dilated multi-fiber network (DMF-Net).
• Incorporated an asymmetric convolution block into multi-fiber and dilated multi-fiber units for enhanced feature learning.
• Integrated an expectation-maximization attention (EMA) module to capture long-range contextual dependencies.

Main Results:

• AEMA-Net demonstrated superior performance compared to 3D U-Net and DMF-Net on BraTS 2018, 2019, and 2020 datasets.
• The model achieved competitive results against state-of-the-art brain tumor segmentation techniques.
• Experimental evaluations confirmed the effectiveness of the proposed modifications for improved segmentation accuracy.

Conclusions:

• AEMA-Net offers an effective solution for automatic brain tumor segmentation on MRI.
• The network successfully balances accuracy improvements with computational efficiency.
• AEMA-Net represents a significant advancement in automated neuro-oncology imaging analysis.