Unsupervised Domain Adaptation for Medical Image Segmentation with Conditional Diffusion Model

Yang Zhao,Ke Lu,Shuhua Wang,Jian Lu,Jian Xue
DOI: https://doi.org/10.1109/isbi56570.2024.10635862
2024-01-01
Abstract:Unsupervised Domain Adaptation (UDA) has received widespread attention in medical image processing, as it can efficiently handle cross-domain data while diminishing the algorithm’s reliance on data annotations. The existing UDA approaches based on image alignment have widely used adversarial learning to minimize the appearance differences of cross-domain data, but are prone to corrupting the original structural semantics. Conditional Diffusion Model (CDM) possesses the ability to generate images that adhere to specific distributions, showcasing its tremendous potential in various research areas. In this paper, we propose a UDA method with CDM for magnetic resonance (MR) images that generates prediction masks for target domain data. To enhance the generalization ability of the model to target domain images, a Cross-domain Frequency-mixed Consistency (CFC) method is proposed, which fuses portions of the spectrum from both source and target domain images. Additionally, we propose an Intra-domain Structural Consistency (ISC) method, which utilizes the CutMix strategy to introduce structure-level perturbations to the frequency-mixed image, thereby improving structural consistency in predicting target domain data. Extensive experiments on two MRI datasets demonstrate that our method outperforms other state-of-the-art UDA methods, and also showcased the promising potential of the CDM method for handling medical images for the first time.
What problem does this paper attempt to address?