Unsupervised segmentation of dynamic pulmonary MRI using cross-modality adaptation with annotated CT images.
Authors
Affiliations (6)
Affiliations (6)
- National Engineering Research Center of Advanced Magnetic Resonance Technologies for Diagnosis and Therapy, School of Biomedical Engineering, Shanghai Jiao Tong University, Shanghai, China.
- Institute of Medical Robotics, Shanghai Jiao Tong University, Shanghai, China.
- Department of Radiology, Second Affiliated Hospital of Naval Medical University, Shanghai, China.
- Department of Radiology, Second Affiliated Hospital of Naval Medical University, Shanghai, China. [email protected].
- National Engineering Research Center of Advanced Magnetic Resonance Technologies for Diagnosis and Therapy, School of Biomedical Engineering, Shanghai Jiao Tong University, Shanghai, China. [email protected].
- Institute of Medical Robotics, Shanghai Jiao Tong University, Shanghai, China. [email protected].
Abstract
Accurate segmentation of lung parenchyma in dynamic pulmonary magnetic resonance imaging (MRI) is required for clinical diagnosis and treatment planning. However, supervised deep learning algorithms rely on annotated datasets, which are scarce for pulmonary MRI. This study aims to leverage existing annotated computed tomography (CT) data to enable unsupervised segmentation of pulmonary MRI. A new framework was proposed for unsupervised segmentation of pulmonary MRI. First, a masked autoencoder is pretrained to learn modality-invariant features. Next, an initial segmenter is trained using labeled CT images, combined with a temporal consistency loss on 4D MR images. The initial segmenter generates predictions for MR images, which are further processed through a select-and-refine pipeline to produce high-quality pseudolabels. Finally, a final segmenter is trained using the pseudolabeled MRI, combined with the temporal consistency constraint. The model was trained using 31 unlabeled 4D MR images and 30 labeled CT images, and evaluated on 20 and 12 4D MR images acquired from two different centers. The proposed method achieves accurate and robust segmentation of lung parenchyma and outperforms state-of-the-art cross-modality methods, with Dice scores of 97.75 ± 0.57% and 97.72 ± 0.55%, and average surface distances of 1.80 ± 1.40 mm and 1.34 ± 0.69 mm across the two test sets. The proposed method effectively transfers segmentation knowledge from CT to MRI, enabling accurate segmentation of lung parenchyma. By eliminating the dependency on MRI annotations, our technique offers a practical and promising solution for segmentation of dynamic pulmonary MRI.