Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Duo Peng, Ping Hu, Qiuhong Ke, Jun Liu• 2023

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationAbdominal Abd MRI -> CT (test)
Liver Score41.52
24
Medical Image SegmentationAbdominal Abd CT -> MRI (test)
Liver Score29.15
24
Brain Tumor SegmentationBraTS T1 Target Domain (test)
WT Score46.11
14
Brain Tumor SegmentationBraTS T1CE Target Domain (test)
WT Score37.69
14
Medical Image SegmentationBraTS Target domain T1
95HD (WT)43.41
14
Cardiac Image SegmentationMM-WHS Cardiac MRI → Cardiac CT
95HD (LVM)17.12
14
Cardiac Image SegmentationMM-WHS Cardiac CT → Cardiac MRI
95HD (LVM)53.27
14
Medical Image SegmentationMM-WHS Cardiac CT → Cardiac MRI
Dice (LVM)57.3
14
Medical Image SegmentationAbdominal Multi-Organ MRI → CT
95HD (Liver)62.85
14
Medical Image SegmentationAbdominal Multi-Organ CT → MRI
95HD (Liver)70.19
14
Showing 10 of 11 rows

Other info

Follow for update