用单步扩散模型提升检测器在跨域场景下的性能、泛化与迁移能力。
Boosting Domain Generalized and Adaptive Detection with Diffusion Models: Fitness, Generalization, and Transferability
- 从单步扩散过程提取中间特征,降低75%推理时间。
- 构建物体中心辅助分支,实现对目标的鲁棒特征提取。
- 统一框架下融合特征与对象级对齐,兼顾性能与跨域适应性。
检测器常因训练与测试数据间的领域差异导致性能下降。现有方法虽尝试将扩散模型用于领域泛化(DG)与自适应(DA),但仍面临推理成本高、未充分挖掘扩散模型潜力等问题。本文通过提取单步扩散过程中的中间特征,优化特征采集与融合,使推理时间减少75%,同时提升源域性能(即性能适配性)。进一步地,利用带框掩码图像与类别提示构建物体中心的辅助分支,提取聚焦于物体的鲁棒且领域不变特征,并引入一致性损失对齐辅助分支与主分支,平衡性能与泛化能力,防止过拟合,提升目标域表现(即泛化性)。此外,在统一框架中,标准检测器通过源域(用于DG)与无标签目标域(用于DA)的特征级与对象级对齐,受扩散检测器引导,显著改善跨域检测性能(即可迁移性)。该方法在3个DA基准与5个DG基准上取得竞争性结果。在COCO泛化基准上的实验表明,其在大领域偏移与低数据场景下仍保持显著优势,展现出卓越效率。工作验证了扩散模型在领域泛化与自适应检测任务中的优越性,为多样领域下的视觉感知提供重要启示。代码已开源。
原文摘要 · Abstract (English)
Detectors often suffer from performance drop due to domain gap between training and testing data. Recent methods explore diffusion models applied to domain generalization (DG) and adaptation (DA) tasks, but still struggle with large inference costs and have not yet fully leveraged the capabilities of diffusion models. We propose to tackle these problems by extracting intermediate features from a single-step diffusion process, improving feature collection and fusion to reduce inference time by 75% while enhancing performance on source domains (i.e., Fitness). Then, we construct an object-centered auxiliary branch by applying box-masked images with class prompts to extract robust and domain-invariant features that focus on object. We also apply consistency loss to align the auxiliary and ordinary branch, balancing fitness and generalization while preventing overfitting and improving performance on target domains (i.e., Generalization). Furthermore, within a unified framework, standard detectors are guided by diffusion detectors through feature-level and object-level alignment on source domains (for DG) and unlabeled target domains (for DA), thereby improving cross-domain detection performance (i.e., Transferability). Our method achieves competitive results on 3 DA benchmarks and 5 DG benchmarks. Additionally, experiments on COCO generalization benchmark demonstrate that our method maintains significant advantages and show remarkable efficiency in large domain shifts and low-data scenarios. Our work shows the superiority of applying diffusion models to domain generalized and adaptive detection tasks and offers valuable insights for visual perception tasks across diverse domains. The code is available at \href{https://github.com/heboyong/Fitness-Generalization-Transferability}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。