English

Diffusion Features to Bridge Domain Gap for Semantic Segmentation

Computer Vision and Pattern Recognition 2024-11-22 v2 Artificial Intelligence

Abstract

Pre-trained diffusion models have demonstrated remarkable proficiency in synthesizing images across a wide range of scenarios with customizable prompts, indicating their effective capacity to capture universal features. Motivated by this, our study delves into the utilization of the implicit knowledge embedded within diffusion models to address challenges in cross-domain semantic segmentation. This paper investigates the approach that leverages the sampling and fusion techniques to harness the features of diffusion models efficiently. We propose DIffusion Feature Fusion (DIFF) as a backbone use for extracting and integrating effective semantic representations through the diffusion process. By leveraging the strength of text-to-image generation capability, we introduce a new training framework designed to implicitly learn posterior knowledge from it. Through rigorous evaluation in the contexts of domain generalization semantic segmentation, we establish that our methodology surpasses preceding approaches in mitigating discrepancies across distinct domains and attains the state-of-the-art (SOTA) benchmark.

Keywords

Cite

@article{arxiv.2406.00777,
  title  = {Diffusion Features to Bridge Domain Gap for Semantic Segmentation},
  author = {Yuxiang Ji and Boyong He and Chenyuan Qu and Zhuoyue Tan and Chuan Qin and Liaoni Wu},
  journal= {arXiv preprint arXiv:2406.00777},
  year   = {2024}
}

Comments

The code is released at https://github.com/Yux1angJi/DIFF

R2 v1 2026-06-28T16:50:10.673Z