arXiv:2410.21708cs.CV2024-10NeurIPS被引 7

用文生图扩散模型实现多模态语义分割无监督适配

Unsupervised Modality Adaptation with Text-to-Image Diffusion Models for Semantic Segmentation

  • 基于扩散模型生成稳定伪标签,缓解跨模态差异导致的精度下降
  • 通过标签调色板与潜空间回归,恢复细粒度特征,提升重建质量
  • 在图像→深度/红外/事件模态任务中表现领先,适合多传感器场景

尽管已有进展,当前无监督域适应方法主要针对图像域间迁移,未有效利用深度、红外、事件等丰富的视觉模态,限制了其在真实多模态场景中的应用。为此,本文提出面向语义分割的文生图扩散模型模态适配方法(MADM),利用大规模图文对预训练的文生图扩散模型增强跨模态能力。MADM包含两个互补组件:首先,为应对大模态差距导致的伪标签精度下降问题,设计基于扩散的伪标签生成机制,通过添加潜空间噪声稳定标签;其次,针对扩散模型潜空间分辨率低的问题,引入标签调色板与潜空间回归,将独热编码标签转为RGB形式并在潜空间中回归,确保预训练解码器上采样获得精细特征。大量实验表明,MADM在图像到深度、红外、事件模态等多种任务中均达到最新性能。代码与模型已开源:https://github.com/XiaRho/MADM。

原文摘要 · Abstract (English)

Despite their success, unsupervised domain adaptation methods for semantic segmentation primarily focus on adaptation between image domains and do not utilize other abundant visual modalities like depth, infrared and event. This limitation hinders their performance and restricts their application in real-world multimodal scenarios. To address this issue, we propose Modality Adaptation with text-to-image Diffusion Models (MADM) for semantic segmentation task which utilizes text-to-image diffusion models pre-trained on extensive image-text pairs to enhance the model's cross-modality capabilities. Specifically, MADM comprises two key complementary components to tackle major challenges. First, due to the large modality gap, using one modal data to generate pseudo labels for another modality suffers from a significant drop in accuracy. To address this, MADM designs diffusion-based pseudo-label generation which adds latent noise to stabilize pseudo-labels and enhance label accuracy. Second, to overcome the limitations of latent low-resolution features in diffusion models, MADM introduces the label palette and latent regression which converts one-hot encoded labels into the RGB form by palette and regresses them in the latent space, thus ensuring the pre-trained decoder for up-sampling to obtain fine-grained features. Extensive experimental results demonstrate that MADM achieves state-of-the-art adaptation performance across various modality tasks, including images to depth, infrared, and event modalities. We open-source our code and models at https://github.com/XiaRho/MADM.

语义分割多模态扩散模型无监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。