A Domain-Shift Invariant CNN Framework for Cardiac MRI Segmentation Across Unseen Domains

Sanjeet S. Patil, Manojkumar Ramteke, Mansi Verma, Sandeep Seth, Rohit Bhargava, Shachi Mittal, Anurag S. Rathore

Research output: Contribution to journalArticlepeer-review


The emergence of various deep learning approaches in diagnostic medical image segmentation has made machines capable of accomplishing human-level accuracy. However, the generalizability of these architectures across patients from different countries, Magnetic Resonance Imaging (MRI) scans from distinct vendors, and varying imaging conditions remains questionable. In this work, we propose a translatable deep learning framework for diagnostic segmentation of cine MRI scans. This study aims to render the available SOTA (state-of-the-art) architectures domain-shift invariant by utilizing the heterogeneity of multi-sequence cardiac MRI. To develop and test our approach, we curated a diverse group of public datasets and a dataset obtained from private source. We evaluated 3 SOTA CNN (Convolution neural network) architectures i.e., U-Net, Attention-U-Net, and Attention-Res-U-Net. These architectures were first trained on a combination of three different cardiac MRI sequences. Next, we examined the M&M (multi-center & mutli-vendor) challenge dataset to investigate the effect of different training sets on translatability. The U-Net architecture, trained on the multi-sequence dataset, proved to be the most generalizable across multiple datasets during validation on unseen domains. This model attained mean dice scores of 0.81, 0.85, and 0.83 for myocardial wall segmentation after testing on unseen MyoPS (Myocardial Pathology Segmentation) 2020 dataset, AIIMS (All India Institute of Medical Sciences) dataset and M&M dataset, respectively. Our framework achieved Pearson’s correlation values of 0.98, 0.99, and 0.95 between the observed and predicted parameters of end diastole volume, end systole volume, and ejection fraction, respectively, on the unseen Indian population dataset.

Original languageEnglish (US)
Pages (from-to)2148-2163
Number of pages16
JournalJournal of Digital Imaging
Issue number5
StatePublished - Oct 2023


  • Cardiac magnetic resonance imaging
  • Convolutional neural networks
  • Deep learning
  • Generalizable architecture
  • Semantic segmentation

ASJC Scopus subject areas

  • Radiological and Ultrasound Technology
  • Radiology Nuclear Medicine and imaging
  • Computer Science Applications


Dive into the research topics of 'A Domain-Shift Invariant CNN Framework for Cardiac MRI Segmentation Across Unseen Domains'. Together they form a unique fingerprint.

Cite this