Enhancing Cross-Modal Medical Image Segmentation through Compositionality
Preprint
Abstract
Cross-modal medical image segmentation presents a significant challenge, as different imaging modalities produce images with varying resolutions, contrasts, and appearances of anatomical structures. We introduce compositionality as an inductive bias in a cross-modal segmentation network to improve segmentation performance and interpretability while reducing complexity. The proposed network is an end-to-end cross-modal segmentation framework that enforces compositionality on the learned representations using learnable von Mises-Fisher kernels. These kernels facilitate content-style disentanglement in the learned representations, resulting in compositional content representations that are inherently interpretable and effectively disentangle different anatomical structures. The experimental results demonstrate enhanced segmentation performance and reduced computational costs on multiple medical datasets. Additionally, we demonstrate the interpretability of the learned compositional features. Code and checkpoints will be publicly available at: https://github.com/Trustworthy-AI-UU-NKI/Cross-Modal-Segmentation.Keywords
All Related Versions
This publication has 11 references indexed in Scilit:
- Unsupervised Domain Adaptation for Medical Image Segmentation by Disentanglement Learning and Self-TrainingIEEE Transactions on Medical Imaging, 2022
- Learning disentangled representations in the imaging domainMedical Image Analysis, 2022
- CyCMIS: Cycle-consistent Cross-domain Medical Image Segmentation via diverse image augmentationMedical Image Analysis, 2022
- Beyond Mutual Information: Generative Adversarial Network for Domain Adaptation Using Information Bottleneck ConstraintIEEE Transactions on Medical Imaging, 2021
- Diverse data augmentation for learning image segmentation with cross-modality annotationsMedical Image Analysis, 2021
- Disentangle domain features for cross-modality cardiac image segmentationMedical Image Analysis, 2021
- CHAOS Challenge - combined (CT-MR) healthy abdominal organ segmentationMedical Image Analysis, 2021
- CF Distance: A New Domain Discrepancy Metric and Application to Explicit Domain Adaptation for Cross-Modality Cardiac Image SegmentationIEEE Transactions on Medical Imaging, 2020
- DRIT++: Diverse Image-to-Image Translation via Disentangled RepresentationsInternational Journal of Computer Vision, 2020
- Evaluation of algorithms for Multi-Modality Whole Heart Segmentation: An open-access grand challengeMedical Image Analysis, 2019