arXiv:2410.08023cs.CVcs.AI2024-10

通过遮蔽与去噪提升目标域特征,让模型更准分类。

GrabDAE: An Innovative Framework for Unsupervised Domain Adaptation Utilizing Grab-Mask and Denoise Auto-Encoder

  • 用Grab-Mask模糊背景,让模型专注关键特征。
  • 用去噪自编码器增强特征对齐,提升分类准确率。
  • 适合图像分类中的无监督领域自适应任务。

无监督领域自适应(UDA)旨在通过解决领域偏移问题,将源域标注数据训练的模型适配到无标签目标域。现有方法常未能充分挖掘目标域上下文信息,导致源域与目标域对齐时决策边界分离不佳。为此,我们提出GrabDAE框架,用于视觉分类任务中的领域偏移应对。其核心创新包括:Grab-Mask模块,通过模糊目标域图像背景,使模型在对比学习中聚焦于关键、领域相关特征;以及去噪自编码器(DAE),通过重建特征并过滤噪声,增强特征对齐,提升对目标域的鲁棒适应能力。在VisDA-2017、Office-Home和Office31等基准数据集上的大量实验表明,GrabDAE持续超越现有先进UDA方法,创下新性能基准。该框架通过新颖的特征掩蔽与去噪策略,为领域自适应提供了显著的理论与实践进步。

原文摘要 · Abstract (English)

Unsupervised Domain Adaptation (UDA) aims to adapt a model trained on a labeled source domain to an unlabeled target domain by addressing the domain shift. Existing Unsupervised Domain Adaptation (UDA) methods often fall short in fully leveraging contextual information from the target domain, leading to suboptimal decision boundary separation during source and target domain alignment. To address this, we introduce GrabDAE, an innovative UDA framework designed to tackle domain shift in visual classification tasks. GrabDAE incorporates two key innovations: the Grab-Mask module, which blurs background information in target domain images, enabling the model to focus on essential, domain-relevant features through contrastive learning; and the Denoising Auto-Encoder (DAE), which enhances feature alignment by reconstructing features and filtering noise, ensuring a more robust adaptation to the target domain. These components empower GrabDAE to effectively handle unlabeled target domain data, significantly improving both classification accuracy and robustness. Extensive experiments on benchmark datasets, including VisDA-2017, Office-Home, and Office31, demonstrate that GrabDAE consistently surpasses state-of-the-art UDA methods, setting new performance benchmarks. By tackling UDA's critical challenges with its novel feature masking and denoising approach, GrabDAE offers both significant theoretical and practical advancements in domain adaptation.

领域自适应去噪自编码器图像分类

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