Diffusion-based Image Translation with Label Guidance for Domain Adaptive Semantic Segmentation
About
Translating images from a source domain to a target domain for learning target models is one of the most common strategies in domain adaptive semantic segmentation (DASS). However, existing methods still struggle to preserve semantically-consistent local details between the original and translated images. In this work, we present an innovative approach that addresses this challenge by using source-domain labels as explicit guidance during image translation. Concretely, we formulate cross-domain image translation as a denoising diffusion process and utilize a novel Semantic Gradient Guidance (SGG) method to constrain the translation process, conditioning it on the pixel-wise source labels. Additionally, a Progressive Translation Learning (PTL) strategy is devised to enable the SGG method to work reliably across domains with large gaps. Extensive experiments demonstrate the superiority of our approach over state-of-the-art methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Medical Image Segmentation | Abdominal Abd MRI -> CT (test) | Liver Score41.52 | 24 | |
| Medical Image Segmentation | Abdominal Abd CT -> MRI (test) | Liver Score29.15 | 24 | |
| Brain Tumor Segmentation | BraTS T1 Target Domain (test) | WT Score46.11 | 14 | |
| Brain Tumor Segmentation | BraTS T1CE Target Domain (test) | WT Score37.69 | 14 | |
| Medical Image Segmentation | BraTS Target domain T1 | 95HD (WT)43.41 | 14 | |
| Cardiac Image Segmentation | MM-WHS Cardiac MRI → Cardiac CT | 95HD (LVM)17.12 | 14 | |
| Cardiac Image Segmentation | MM-WHS Cardiac CT → Cardiac MRI | 95HD (LVM)53.27 | 14 | |
| Medical Image Segmentation | MM-WHS Cardiac CT → Cardiac MRI | Dice (LVM)57.3 | 14 | |
| Medical Image Segmentation | Abdominal Multi-Organ MRI → CT | 95HD (Liver)62.85 | 14 | |
| Medical Image Segmentation | Abdominal Multi-Organ CT → MRI | 95HD (Liver)70.19 | 14 |