Segmentation-Aware MRI Reconstruction


Acar M., ÇUKUR T., Öksüz İ.

5th Workshop on Machine Learning for Medical Image Reconstruction (MLMIR) held as part of the 25th Medical Image Computing and Computer Assisted Intervention (MICCAI), Singapore, Singapur, 22 Eylül 2022, cilt.13587, ss.53-61 identifier identifier

  • Yayın Türü: Bildiri / Tam Metin Bildiri
  • Cilt numarası: 13587
  • Doi Numarası: 10.1007/978-3-031-17247-2_6
  • Basıldığı Şehir: Singapore
  • Basıldığı Ülke: Singapur
  • Sayfa Sayıları: ss.53-61
  • Anahtar Kelimeler: Cardiac MRI, Reconstruction, Segmentation, Convolutional neural networks, NEURAL-NETWORK
  • İstanbul Teknik Üniversitesi Adresli: Evet

Özet

Deep learning models have been broadly adopted for accelerating MRI acquisitions in recent years. A common approach is to train deep models based on loss functions that place equal emphasis on reconstruction errors across the field-of-view. This homogeneous weighting of loss contributions might be undesirable in cases where the diagnostic focus is on tissues in a specific subregion of the image. In this paper, we propose a framework for segmentation-aware reconstruction based on segmentation as a proxy task. We leverage an end-to-end model comprising reconstruction and segmentation networks; and leverage backpropagation of segmentation error to devise a pseudo-attention effect to focus the reconstruction network. We introduce a novel stabilization method to prevent convergence onto a local minima with unacceptably poor reconstruction or segmentation performance. Our stabilization approach initiates learning on fully-sampled acquisitions, and gradually increases the undersampling rate assumed in the training set to its desired value. We validate our approach for cardiac MR reconstruction on the publicly available OCMR dataset. Segmentation-aware reconstruction significantly outperforms vanilla reconstruction for cardiac imaging.