Uncertainty-Aware Optimal Transport for Semantically Coherent Out-of-Distribution Detection

被引:2
|
作者
Lu, Fan [1 ]
Zhu, Kai [1 ]
Zhai, Wei
Zheng, Kecheng [2 ]
Cao, Yang [1 ,3 ]
机构
[1] Univ Sci & Technol China, Hefei, Peoples R China
[2] Ant Grp, Sydney, NSW, Australia
[3] Hefei Comprehens Natl Sci Ctr, Inst Artificial Intelligence, Hefei, Peoples R China
基金
国家重点研发计划;
关键词
D O I
10.1109/CVPR52729.2023.00320
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semantically coherent out-of-distribution (SCOOD) detection aims to discern outliers from the intended data distribution with access to unlabeled extra set. The coexistence of in-distribution and out-of-distribution samples will exacerbate the model overfitting when no distinction is made. To address this problem, we propose a novel uncertainty-aware optimal transport scheme. Our scheme consists of an energy-based transport (ET) mechanism that estimates the fluctuating cost of uncertainty to promote the assignment of semantic-agnostic representation, and an inter-cluster extension strategy that enhances the discrimination of semantic property among different clusters by widening the corresponding margin distance. Furthermore, a T-energy score is presented to mitigate the magnitude gap between the parallel transport and classifier branches. Extensive experiments on two standard SCOOD benchmarks demonstrate the above-par OOD detection performance, outperforming the state-of-the-art methods by a margin of 27.69% and 34.4% on FPR@95, respectively. Code is available at https://github.com/LuFan31/ET-OOD.
引用
收藏
页码:3282 / 3291
页数:10
相关论文
共 50 条
  • [1] Semantically Coherent Out-of-Distribution Detection
    Yang, Jingkang
    Wang, Haoqi
    Feng, Litong
    Yan, Xiaopeng
    Zheng, Huabin
    Zhang, Wayne
    Liub, Ziwei
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 8281 - 8289
  • [2] Out-of-distribution detection-assisted trustworthy machinery fault diagnosis approach with uncertainty-aware deep ensembles
    Han, Te
    Li, Yan-Fu
    [J]. RELIABILITY ENGINEERING & SYSTEM SAFETY, 2022, 226
  • [3] Out-of-distribution Detection with Boundary Aware Learning
    Pei, Sen
    Zhang, Xin
    Fan, Bin
    Meng, Gaofeng
    [J]. COMPUTER VISION, ECCV 2022, PT XXIV, 2022, 13684 : 235 - 251
  • [4] Boosting Out-of-Distribution Image Detection With Epistemic Uncertainty
    Oh, Dokwan
    Ji, Daehyun
    Kwon, Ohmin
    Hyun, Yoonsuk
    [J]. IEEE ACCESS, 2022, 10 : 109289 - 109298
  • [5] Out-of-Distribution Detection with Uncertainty Enhanced Attention Maps
    Gao, Yue
    Su, Qinliang
    [J]. 2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [6] Characterizing Out-of-Distribution Error via Optimal Transport
    Lu, Yuzhe
    Qin, Yilong
    Zhai, Runtian
    Shen, Andrew
    Chen, Ketong
    Wang, Zhenlin
    Kolouri, Soheil
    Stepputtis, Simon
    Campbell, Joseph
    Sycara, Katia
    [J]. ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [7] Optimal Parameter and Neuron Pruning for Out-of-Distribution Detection
    Chen, Chao
    Fu, Zhihang
    Liu, Kai
    Chen, Ze
    Tao, Mingyuan
    Ye, Jieping
    [J]. ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [8] Out-of-distribution Detection Learning with Unreliable Out-of-distribution Sources
    Zheng, Haotian
    Wang, Qizhou
    Fang, Zhen
    Xia, Xiaobo
    Liu, Feng
    Liu, Tongliang
    Han, Bo
    [J]. ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [9] Joint Out-of-Distribution Detection and Uncertainty Estimation for Trajectory Prediction
    Wiederer, Julian
    Schmidt, Julian
    Kressel, Ulrich
    Dietmayer, Klaus
    Belagiannis, Vasileios
    [J]. 2023 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS), 2023, : 5487 - 5494
  • [10] Predictive uncertainty estimation for out-of-distribution detection in digital pathology
    Linmans, Jasper
    Elfwing, Stefan
    van der Laak, Jeroen
    Litjens, Geert
    [J]. MEDICAL IMAGE ANALYSIS, 2023, 83