FANCL: Feature-Guided Attention Network with Curriculum Learning for Brain Metastases Segmentation

Zijiang Liu,Xiaoyu Liu,Linhao Qu,Yonghong Shi
2024-10-29
Abstract:Accurate segmentation of brain metastases (BMs) in MR image is crucial for the diagnosis and follow-up of patients. Methods based on deep convolutional neural networks (CNNs) have achieved high segmentation performance. However, due to the loss of critical feature information caused by convolutional and pooling operations, CNNs still face great challenges in small BMs segmentation. Besides, BMs are irregular and easily confused with healthy tissues, which makes it difficult for the model to effectively learn tumor structure during training. To address these issues, this paper proposes a novel model called feature-guided attention network with curriculum learning (FANCL). Based on CNNs, FANCL utilizes the input image and its feature to establish the intrinsic connections between metastases of different sizes, which can effectively compensate for the loss of high-level feature from small tumors with the information of large tumors. Furthermore, FANCL applies the voxel-level curriculum learning strategy to help the model gradually learn the structure and details of BMs. And baseline models of varying depths are employed as curriculum-mining networks for organizing the curriculum progression. The evaluation results on the BraTS-METS 2023 dataset indicate that FANCL significantly improves the segmentation performance, confirming the effectiveness of our method.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the accurate segmentation of brain metastases (BMs) in MRI images. Specifically, the paper points out that although methods based on deep convolutional neural networks (CNNs) have achieved high precision in medical image segmentation tasks, due to the loss of key feature information caused by convolution and pooling operations, these methods still face great challenges in the segmentation of cerebellar metastases. In addition, because brain metastases have irregular shapes and are easily confused with healthy tissues, it is difficult for the model to effectively learn the tumor structure during the training process. Therefore, the paper proposes a new model - Feature - Guided Attention Network with Curriculum Learning (FANCL), aiming to solve the above problems. FANCL improves the segmentation performance of brain metastases in the following ways: 1. **Feature - guided attention mechanism**: Use the input image and its features to establish the internal relationship between metastases of different sizes, effectively compensating for the high - level feature information lost by small tumors due to convolution operations. 2. **Voxel - level curriculum learning strategy**: By gradually learning the structure and details of brain metastases, help the model gradually adapt to complex segmentation tasks. 3. **Baseline model as curriculum mining network**: Use baseline models of different depths (such as nnU - Net) to organize the curriculum progress, ensuring that the curriculum content conforms to the model performance and data characteristics. Through these innovations, the evaluation results of FANCL on the BraTS - METS 2023 dataset show that this model significantly improves the segmentation performance and verifies the effectiveness of the method.