arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.15580cs.CV

层次化视觉类别建模:一种用于分布外检测的联合表示学习与密度估计框架

Hierarchical Visual Categories Modeling: A Joint Representation Learning and Density Estimation Framework for Out-of-Distribution Detection

Academy for Engineering and Technology, Fudan University(复旦大学工程技术研究院);School of Computer Science, Fudan University(复旦大学计算机科学学院)

Jinglun Li, Xinyu Zhou, Pinxue Guo, Yixuan Sun, Yiwen Huang, Weifeng Ge, Wenqiang Zhang

AI总结:

本文提出一种层次化视觉类别建模方案,通过联合表示学习与高斯混合模型统计建模,并聚合基于马氏距离的度量计算分布内评分,无需辅助离群点数据即可有效检测分布外样本,且不削弱视觉识别模型的判别能力。

摘要:

检测视觉识别模型的分布外输入已成为安全深度学习中的关键问题。本文提出了一种新颖的层次化视觉类别建模方案,通过联合表示学习和统计建模将分布外数据与分布内数据分离。我们为每个分布内类别学习一个高斯混合模型。有许多高斯混合模型用于建模不同的视觉类别。基于这些高斯模型,我们通过聚合多个基于马氏距离的度量设计了一个分布内评分函数。我们不使用任何辅助离群点数据作为训练样本,因为这可能损害分布外检测算法的泛化能力。我们将ImageNet-1k数据集随机划分为十折。我们使用其中一折作为分布内数据集,其余作为分布外数据集来评估所提出的方法。我们还在七个流行基准上进行了实验,包括CIFAR、iNaturalist、SUN、Places、Textures、ImageNet-O和OpenImage-O。大量实验表明,所提出的方法明显优于最先进的算法。同时,我们发现与经典方法学习的特征相比,我们的视觉表示具有竞争力的性能。这些结果表明,所提出的方法没有削弱视觉识别模型的判别能力,并在检测分布外样本方面保持高效率。

英文摘要:

Detecting out-of-distribution inputs for visual recognition models has become critical in safe deep learning. This paper proposes a novel hierarchical visual category modeling scheme to separate out-of-distribution data from in-distribution data through joint representation learning and statistical modeling. We learn a mixture of Gaussian models for each in-distribution category. There are many Gaussian mixture models to model different visual categories. With these Gaussian models, we design an in-distribution score function by aggregating multiple Mahalanobis-based metrics. We don't use any auxiliary outlier data as training samples, which may hurt the generalization ability of out-of-distribution detection algorithms. We split the ImageNet-1k dataset into ten folds randomly. We use one fold as the in-distribution dataset and the others as out-of-distribution datasets to evaluate the proposed method. We also conduct experiments on seven popular benchmarks, including CIFAR, iNaturalist, SUN, Places, Textures, ImageNet-O, and OpenImage-O. Extensive experiments indicate that the proposed method outperforms state-of-the-art algorithms clearly. Meanwhile, we find that our visual representation has a competitive performance when compared with features learned by classical methods. These results demonstrate that the proposed method hasn't weakened the discriminative ability of visual recognition models and keeps high efficiency in detecting out-of-distribution samples.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑