Brain Segmentation from k-Space with End-to-End Recurrent Attention Network

The task of medical image segmentation commonly involves an image reconstruction step to convert acquired raw data to images before any analysis. However, noises, artifacts and loss of information due to the reconstruction process are almost inevitable, which compromises the final performance of segmentation. We present a novel learning framework that performs magnetic resonance brain image segmentation directly from k-space data. The end-to-end framework consists of a unique task-driven attention module that recurrently utilizes intermediate segmentation estimation to facilitate image-domain feature extraction from the raw data, thus closely bridging the reconstruction and the segmentation tasks. In addition, to address the challenge of manual labeling, we introduce a novel workflow to generate labeled training data for segmentation by exploiting imaging modality simulators and digital phantoms. Extensive experimental results show that the proposed method outperforms several state-of-the-art methods.

[1]  Dimitris N. Metaxas,et al.  MRI Reconstruction Via Cascaded Channel-Wise Attention Network , 2018, 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019).

[2]  Daniel Rueckert,et al.  A Deep Cascade of Convolutional Neural Networks for MR Image Reconstruction , 2017, IPMI.

[3]  Liyan Sun,et al.  Joint CS-MRI Reconstruction and Segmentation with a Unified Deep Network , 2019, IPMI.

[4]  Yun Fu,et al.  Tell Me Where to Look: Guided Attention Inference Network , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[5]  Daniel Rueckert,et al.  Application-Driven MRI: Joint Reconstruction and Segmentation from Undersampled MRI Data , 2014, MICCAI.

[6]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[7]  Daniel C. Castro,et al.  Cardiac MR Segmentation from Undersampled k-space Using Deep Latent Representation Learning , 2018, MICCAI.

[8]  Daniel Rueckert,et al.  A Deep Cascade of Convolutional Neural Networks for Dynamic MR Image Reconstruction , 2017, IEEE Transactions on Medical Imaging.

[9]  Richard Kijowski,et al.  Fast Realistic MRI Simulations Based on Generalized Multi-Pool Exchange Tissue Model , 2016, IEEE Transactions on Medical Imaging.

[10]  Dimitris N. Metaxas,et al.  FR-Net: Joint Reconstruction and Segmentation in Compressed Sensing Cardiac MRI , 2019, FIMH.

[11]  Dit-Yan Yeung,et al.  Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting , 2015, NIPS.