Gradient-Map-Guided Adaptive Domain Generalization for Cross Modality MRI Segmentation
Abstract
Cross-modal MRI segmentation is of great value for computer-aided medical diagnosis, enabling flexible data acquisition and model generalization. However, most existing methods have difficulty in handling local variations in domain shift and typically require a significant amount of data for training, which hinders their usage in practice. To address these problems, we propose a novel adaptive domain generalization framework, which integrates a learning-free cross-domain representation based on image gradient maps and a class prior-informed test-time adaptation strategy for mitigating local domain shift. We validate our approach on two multi-modal MRI datasets with six cross-modal segmentation tasks. Across all the task settings, our method consistently outperforms competing approaches and shows a stable performance even with limited training data.
- Publication:
-
arXiv e-prints
- Pub Date:
- November 2023
- DOI:
- 10.48550/arXiv.2311.09737
- arXiv:
- arXiv:2311.09737
- Bibcode:
- 2023arXiv231109737L
- Keywords:
-
- Computer Science - Computer Vision and Pattern Recognition
- E-Print:
- 9 pages, Machine Learning for Health (ML4H) 2023