How to Exploit Hyperspherical Embeddings for Out-of-Distribution Detection?
Yifei Ming, Yiyou Sun, Ousmane Dia, Yixuan Li
Abstract
Out-of-distribution (OOD) detection is a critical task for reliable machine learning. Recent advances in representation learning give rise to distance-based OOD detection, where testing samples are detected as OOD if they are relatively far away from the centroids or prototypes of in-distribution (ID) classes. However, prior methods directly take off-the-shelf contrastive losses that suffice for classifying ID samples, but are not optimally designed when test inputs contain OOD samples. In this work, we propose CIDER, a novel representation learning framework that exploits hyperspherical embeddings for OOD detection. CIDER jointly optimizes two losses to promote strong ID-OOD separability: a dispersion loss that promotes large angular distances among different class prototypes, and a compactness loss that encourages samples to be close to their class prototypes. We analyze and establish the unexplored relationship between OOD detection performance and the embedding properties in the hyperspherical space, and demonstrate the importance of dispersion and compactness. CIDER establishes superior performance, outperforming the latest rival by 13.33% in FPR95. Code is available at https://github.com/deeplearning-wisc/cider .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext fda7a207-cc15-4378-8947-afe65badcfb7Cited by top-tier papers78
- Dream the Impossible: Outlier Imagination with Diffusion ModelsXuefeng Du, Yiyou Sun, Jerry Zhu, Yixuan LiNeurIPS 2023 · 114 citations
- Scaling for Training Time and Post-hoc Out-of-distribution Detection EnhancementKai Xu, Rongyu Chen, Gianni Franchi, Angela YaoICLR 2024 · 81 citations
- Learning to Augment Distributions for Out-of-distribution DetectionQizhou Wang, Zhen Fang, Yonggang Zhang, Feng Liu et al.NeurIPS 2023 · 59 citations
- Learning with Mixture of Prototypes for Out-of-Distribution DetectionHaodong Lu, Dong Gong, Shuo Wang, Jason Xue et al.ICLR 2024 · 55 citations
- Out-of-distribution Detection Learning with Unreliable Out-of-distribution SourcesHaotian Zheng, Qizhou Wang, Zhen Fang, Xiaobo Xia et al.NeurIPS 2023 · 53 citations
Builds on28
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Supervised Contrastive LearningPrannay Khosla, Piotr Teterwak, Chen Wang, Aaron Sarna et al.NeurIPS 2020 · 7,049 citations
- Understanding Contrastive Representation Learning through Alignment and Uniformity on the HypersphereTongzhou Wang, Phillip IsolaICML 2020 · 2,360 citations
- Energy-based Out-of-distribution DetectionWeitang Liu, Xiaoyun Wang, John D. Owens, Yixuan LiNeurIPS 2020 · 2,213 citations
- Contrastive Learning with Hard Negative SamplesJoshua David Robinson, Ching-Yao Chuang, Suvrit Sra, Stefanie JegelkaICLR 2021 · 999 citations
Related papers
- Provable Discriminative Hyperspherical Embedding for Out-of-Distribution DetectionZhipeng Zou, Sheng Wan, Guangyu Li, Bo Han et al.AAAI 2025 · 1 citation
- HYPO: Hyperspherical Out-Of-Distribution GeneralizationHaoyue Bai, Yifei Ming, Julian Katz-Samuels, Yixuan LiICLR 2024 · 13 citations
- SIREN: Shaping Representations for Detecting Out-of-Distribution ObjectsXuefeng Du, Gabriel Gozum, Yifei Ming, Yixuan LiNeurIPS 2022 · 105 citations
- How to Overcome Curse-of-Dimensionality for Out-of-Distribution Detection?Soumya Suvra Ghosal, Yiyou Sun, Yixuan LiAAAI 2024 · 26 citations
- Contrastive Out-of-Distribution Detection for Pretrained TransformersWenxuan Zhou, Fangyu Liu, Muhao ChenEMNLP 2021 · 63 citations
