一种基于余弦相似度的分布外检测方法
机器学习
2023-06-28 v1
摘要
检测 OOD 数据的能力是实际机器学习应用的关键方面。在这项工作中，我们表明测试特征与典型 ID 特征之间的余弦相似度是 OOD 数据的一个良好指标。我们提出类别典型匹配（CTM），一种使用后验余弦相似度评分函数的后处理 OOD 检测算法。在多个基准上的大量实验表明，CTM 优于现有的后处理 OOD 检测方法。
引用
@article{arxiv.2306.14920,
title = {A Cosine Similarity-based Method for Out-of-Distribution Detection},
author = {Nguyen Ngoc-Hieu and Nguyen Hung-Quang and The-Anh Ta and Thanh Nguyen-Tang and Khoa D Doan and Hoang Thanh-Tung},
journal= {arXiv preprint arXiv:2306.14920},
year = {2023}
}
备注
Accepted paper at ICML 2023 Workshop on Spurious Correlations, Invariance, and Stability. 10 pages (4 main + appendix)