Optimal Transport-Induced Samples against Out-of-Distribution Overconfidence
Keke Tang, Ziyong Du, Xiaofei Wang, Weilong Peng, Peican Zhu, Zhihong Tian
Abstract
Deep neural networks (DNNs) often produce overconfident predictions on out-of-distribution (OOD) inputs, undermining their reliability in open-world environments. Singularities in semi-discrete optimal transport (OT) mark regions of semantic ambiguity, where classifiers are particularly prone to unwarranted high-confidence predictions. Motivated by this observation, we propose a principled framework to mitigate OOD overconfidence by leveraging the geometry of OT-induced singular boundaries. Specifically, we formulate an OT problem between a continuous base distribution and the latent embeddings of training data, and identify the resulting singular boundaries. By sampling near these boundaries, we construct a class of OOD inputs, termed optimal transport-induced OOD samples (OTIS), which are geometrically grounded and inherently semantically ambiguous. During training, a confidence suppression loss is applied to OTIS to guide the model toward more calibrated predictions in structurally uncertain regions. Extensive experiments show that our method significantly alleviates OOD overconfidence and outperforms state-of-the-art methods.
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 0c26be3f-9735-4c53-9fc3-df8002871a01Builds on11
- Energy-based Out-of-distribution DetectionWeitang Liu, Xiaoyun Wang, John D. Owens, Yixuan LiNeurIPS 2020 · 2,213 citations
- ReAct: Out-of-distribution Detection With Rectified ActivationsYiyou Sun, Chuan Guo, Yixuan LiNeurIPS 2021 · 733 citations
- VOS: Learning What You Don't Know by Virtual Outlier SynthesisXuefeng Du, Zhaoning Wang, Mu Cai, Yixuan LiICLR 2022 · 417 citations
- Towards neural networks that provably know when they don't knowAlexander Meinke, Matthias HeinICLR 2020 · 151 citations
- Ae-OT: a New Generative Model based on Extended Semi-discrete Optimal transportDongsheng An, Yang Guo, Na Lei, Zhongxuan Luo et al.ICLR 2020 · 68 citations
Related papers
- CODEs: Chamfer Out-of-Distribution Examples against Overconfidence IssueKeke Tang, Dingruibo Miao, Weilong Peng, Jianpeng Wu et al.ICCV 2021 · 35 citations
- OpenMix: Exploring Outlier Samples for Misclassification DetectionFei Zhu, Zhen Cheng, Xu-Yao Zhang, Cheng-Lin LiuCVPR 2023
- Nearest Neighbor Guidance for Out-of-Distribution DetectionJaewoo Park, Yoon Gyo Jung, Andrew Beng Jin TeohICCV 2023 · 74 citations
- Bypassing the Transport Plan: Dynamic Reweighting for Out-of-Distribution Detection with Optimal TransportYang Xiao, Weiming Liu, Jun Dan, Tengyue Xu et al.CVPR 2026
- Making transport more robust and interpretable by moving data through a small number of anchor pointsChi-Heng Lin, Mehdi Azabou, Eva L. DyerICML 2021 · 26 citations
