Unsupervised Cross-Modality MR Image Segmentation Via Prompt-Driven Foundation Model
Obtaining pixel-level expert annotations is expensive and labor-intensive in the medical imaging field, especially for multi-modality imaging data like MR. Most conventional cross-modality segmentation methods rely on unsupervised domain adaptation to achieve efficient cross-domain segmentation. However, these methods are often hindered by discrepancies between the source and target domains. In this paper, we propose a new scheme for cross-modality segmentation based on foundation models, which uses spatial consistency across multiple modalities and is not affected by discrepancies between the source and target domains. This scheme allows us to use annotated data from one imaging modality to train a network capable of performing accurate segmentation on other target imaging modalities, without the need for labels or registration processes. Specifically, we propose using a SAM-based model that uses segmentation results from one imaging modality as pseudo labels and prompts to guide training and testing in the target imaging modality. Moreover, we introduce consistency-based prompt tuning and hybrid representation learning to address potential unregistered issues and noisy label problems that may arise in cross-modality segmentation. We conducted extensive validation experiments on two internal datasets and one public dataset, including liver lesion segmentation and liver segmentation. Our method demonstrates significant improvement compared to current state-of-the-art approaches.
Authors
- Huimao Zhang (ORCID: https://orcid.org/0000-0003-0443-2831)
- Wanyun Ma (ORCID: https://orcid.org/0000-0002-0712-5565)
- Kan He (ORCID: https://orcid.org/0000-0002-4931-2115)
- Qi Dou (ORCID: https://orcid.org/0000-0002-3416-9950)
- Hongbin Wang (ORCID: https://orcid.org/0000-0003-2176-2998)
- Lei Zhang (ORCID: https://orcid.org/0000-0003-2343-084X)
- Qi Yang (ORCID: https://orcid.org/0000-0001-6969-8724)
- Jingfeng Zhang (ORCID: https://orcid.org/0000-0002-8206-6235)
- Wen Shen (ORCID: https://orcid.org/0000-0002-8473-5092)
- Lisa Wong
- Kun Zhang (ORCID: https://orcid.org/0000-0002-1088-5813)
Institutions
- Chinese University of Hong Kong (HK)
- Jilin University (CN)
- Nankai University (CN)
- Tianjin First Center Hospital (CN)
- Ningbo No. 2 Hospital (CN)
- First Hospital of Jilin University (CN)
- Tianjin Medical University (CN)
Publication Details
- Journal
- IEEE Transactions on Biomedical Engineering
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1109/tbme.2025.3628499
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00