解决类别不平衡下的跨域异常检测难题,提升真实场景下模型鲁棒性。
Your Data Is Not Perfect: Towards Cross-Domain Out-of-Distribution Detection in Class-Imbalanced Data
- 基于原型对齐实现源域到目标域的语义对齐,缓解域差异
- 自适应生成样本级阈值,精准识别异常样本,性能超越当前最优
- 引入不确定性聚类处理类别不平衡,适合工业部署中的真实数据
以往的分布外(OOD)检测系统仅关注已知类(ID)与异常类(OOD)之间的语义差异。然而,实际中还存在两个额外挑战:源域与目标域间的域差异,以及不同类别间存在的类别不平衡问题。事实上,不同域中相似物体应属于同一类别。本文提出一种更贴近现实但更具挑战性的设置——类别不平衡跨域OOD检测(CCOD),即使用少量有标注的源数据集进行训练,并在大量无标注的目标数据集上进行检测。目标域不假设只含异常类或类别平衡,其类别分布可与源域不同。为应对这一挑战,我们提出一种新的不确定性感知自适应语义对齐(UASA)网络,采用基于原型的对齐策略。首先在源域构建标签驱动的原型,用于目标域分类以缩小域差距;不使用固定阈值,而是生成自适应样本级阈值以应对语义差异;最后通过不确定性感知聚类将语义相似的目标样本分组,缓解类别不平衡问题。在三个挑战性基准上的大量实验表明,所提UASA方法显著优于现有最先进方法。
原文摘要 · Abstract (English)
Previous OOD detection systems only focus on the semantic gap between ID and OOD samples. Besides the semantic gap, we are faced with two additional gaps: the domain gap between source and target domains, and the class-imbalance gap between different classes. In fact, similar objects from different domains should belong to the same class. In this paper, we introduce a realistic yet challenging setting: class-imbalanced cross-domain OOD detection (CCOD), which contains a well-labeled (but usually small) source set for training and conducts OOD detection on an unlabeled (but usually larger) target set for testing. We do not assume that the target domain contains only OOD classes or that it is class-balanced: the distribution among classes of the target dataset need not be the same as the source dataset. To tackle this challenging setting with an OOD detection system, we propose a novel uncertainty-aware adaptive semantic alignment (UASA) network based on a prototype-based alignment strategy. Specifically, we first build label-driven prototypes in the source domain and utilize these prototypes for target classification to close the domain gap. Rather than utilizing fixed thresholds for OOD detection, we generate adaptive sample-wise thresholds to handle the semantic gap. Finally, we conduct uncertainty-aware clustering to group semantically similar target samples to relieve the class-imbalance gap. Extensive experiments on three challenging benchmarks demonstrate that our proposed UASA outperforms state-of-the-art methods by a large margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。