arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02653cs.LGcs.CV
收藏

基于原型混合学习的分布外检测

Learning with Mixture of Prototypes for Out-of-Distribution Detection

University of New South Wales(新南威尔士大学);CSIRO’s Data61(CSIRO Data61)

Haodong Lu, Dong Gong, Shuo Wang, Jason Xue, Lina Yao, Kristen Moore

AI总结:

提出PALM方法,用多个原型建模每类数据以捕捉多样性,通过MLE和对比损失学习紧凑嵌入,在CIFAR-100上达到93.82的AUROC,并支持无标签ID数据。

摘要:

分布外(OOD)检测旨在检测远离分布内(ID)训练数据的测试样本,这对于机器学习模型在现实世界中的安全部署至关重要。基于距离的OOD检测方法随着深度表示学习的增强而出现。它们通过测量样本与ID类中心或原型之间的距离来识别未见过的OOD样本。然而,现有方法依赖于过度简化的数据假设来学习表示,例如,用每个类的一个中心原型对ID数据进行建模,或使用并非为OOD检测设计的损失函数,这忽略了数据内部的自然多样性。强行将每个类的数据样本压缩到仅一个原型周围,导致对现实数据的建模不足和性能受限。为解决这些问题,我们提出了原型混合学习(PALM),该方法使用多个原型对每个类进行建模,以捕捉样本多样性,并学习更忠实、更紧凑的样本嵌入以增强OOD检测。我们的方法自动识别并动态更新原型,通过互邻软分配权重将每个样本分配给原型的子集。PALM优化最大似然估计(MLE)损失,鼓励样本嵌入围绕相关原型紧凑分布,同时对所有原型施加对比损失,以增强原型级别的类内紧凑性和类间判别性。此外,原型的自动估计使我们的方法能够扩展到具有无标签ID数据的挑战性OOD检测任务。大量实验证明了PALM的优越性,在具有挑战性的CIFAR-100基准上实现了最先进的平均AUROC性能93.82。代码可在https://github.com/jeff024/PALM获取。

英文摘要:

Out-of-distribution (OOD) detection aims to detect testing samples far away from the in-distribution (ID) training data, which is crucial for the safe deployment of machine learning models in the real world. Distance-based OOD detection methods have emerged with enhanced deep representation learning. They identify unseen OOD samples by measuring their distances from ID class centroids or prototypes. However, existing approaches learn the representation relying on oversimplified data assumptions, e.g, modeling ID data of each class with one centroid class prototype or using loss functions not designed for OOD detection, which overlook the natural diversities within the data. Naively enforcing data samples of each class to be compact around only one prototype leads to inadequate modeling of realistic data and limited performance. To tackle these issues, we propose PrototypicAl Learning with a Mixture of prototypes (PALM) which models each class with multiple prototypes to capture the sample diversities, and learns more faithful and compact samples embeddings to enhance OOD detection. Our method automatically identifies and dynamically updates prototypes, assigning each sample to a subset of prototypes via reciprocal neighbor soft assignment weights. PALM optimizes a maximum likelihood estimation (MLE) loss to encourage the sample embeddings to be compact around the associated prototypes, as well as a contrastive loss on all prototypes to enhance intra-class compactness and inter-class discrimination at the prototype level. Moreover, the automatic estimation of prototypes enables our approach to be extended to the challenging OOD detection task with unlabelled ID data. Extensive experiments demonstrate the superiority of PALM, achieving state-of-the-art average AUROC performance of 93.82 on the challenging CIFAR-100 benchmark. Code is available at https://github.com/jeff024/PALM.

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

↑