Segmentation of Brain Metastases in MRI: A Two-Stage Deep Learning Approach with Modality Impact Study

Yousef Sadegheih,Dorit Merhof
2024-07-19
Abstract:Brain metastasis segmentation poses a significant challenge in medical imaging due to the complex presentation and variability in size and location of metastases. In this study, we first investigate the impact of different imaging modalities on segmentation performance using a 3D U-Net. Through a comprehensive analysis, we determine that combining all available modalities does not necessarily enhance performance. Instead, the combination of T1-weighted with contrast enhancement (T1c), T1-weighted (T1), and FLAIR modalities yields superior results. Building on these findings, we propose a two-stage detection and segmentation model specifically designed to accurately segment brain metastases. Our approach demonstrates that leveraging three key modalities (T1c, T1, and FLAIR) achieves significantly higher accuracy compared to single-pass deep learning models. This targeted combination allows for precise segmentation, capturing even small metastases that other models often miss. Our model sets a new benchmark in brain metastasis segmentation, highlighting the importance of strategic modality selection and multi-stage processing in medical imaging. Our implementation is freely accessible to the research community on \href{<a class="link-external link-https" href="https://github.com/xmindflow/Met-Seg" rel="external noopener nofollow">this https URL</a>}{GitHub}.
Image and Video Processing
What problem does this paper attempt to address?
This paper attempts to address the issue of brain metastasis segmentation in MRI. Specifically, the authors focus on the following points: 1. **The impact of different imaging modalities on segmentation performance**: The study investigated various MRI modalities (such as T1-weighted, T1-weighted enhanced, T2-weighted, and FLAIR) on the segmentation performance of brain metastases, finding that not all modality combinations improve performance. The combination of T1-weighted enhanced (T1c), T1-weighted (T1), and FLAIR modalities achieved the best results. 2. **Proposing a two-stage detection and segmentation model**: Based on the results of the above modality selection, a two-stage deep learning model was proposed. The first stage uses DenseNet121 to detect regions that may contain metastases, and the second stage uses a modified 3D U-Net for detailed segmentation. This approach can more accurately identify and segment small metastases, improving the overall segmentation accuracy and efficiency. 3. **Improving the limitations of existing methods**: Existing single-pass deep learning models face challenges in handling small metastases or distinguishing metastases from surrounding tissues. The method proposed in this paper significantly improves the segmentation capability of small and heterogeneous brain metastases through multimodal selection and two-stage processing. In summary, this paper aims to improve the automatic segmentation accuracy of brain metastases through strategic modality selection and multi-stage processing, thereby providing more reliable support for clinical diagnosis and treatment.