Causal Evidence Learning for Trusted Open Set Recognition Under Covariate Shift

被引:0
|
作者
Bao, Qingsen [1 ]
Chen, Lei [1 ]
Zhang, Feng [1 ]
Wang, Jun [2 ]
Zhang, Changqing [3 ]
机构
[1] Nanjing Univ Posts & Telecommun, Sch Comp Sci, Nanjing 210023, Peoples R China
[2] Shanghai Univ, Sch Commun & Informat Engn, Shanghai 200444, Peoples R China
[3] Tianjin Univ, Coll Intelligence & Comp, Tianjin 300350, Peoples R China
基金
中国国家自然科学基金;
关键词
Uncertainty; Training; Testing; Estimation; Correlation; Predictive models; Semantics; Open set recognition; uncertainty estimation; covariate shift; data augmentation;
D O I
10.1109/TCSVT.2024.3445182
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Trusted open set recognition aims to classify known classes and reject unknown ones, as well as outputs an uncertainty estimate to measure the reliability of recognition results, thus extending the application scenarios of traditional open set recognition methods to risk-sensitive fields. Current methods assume that the covariate distribution of the known classes remains constant during training and testing. However, due to the common occurrence of covariate shift in practical applications, existing methods often suffer from limited generalization. To this end, a causal evidence learning framework, highlighted by the controllable Evidential Uncertainty Guided Adversarial Data Augmentation (EUG-ADA) and Causal Adversarial Disentanglement (CausalAD) strategies, is proposed to support trusted open set recognition under covariate shift. Specifically, EUG-ADA generates high-quality augmentation samples to increase training data diversity, guided by controllable evidential uncertainty and constrained by semantic consistency. Moreover, it is complemented by the CausalAD, which learns causal representations through causal intervention, mitigating the risk of misrecognition of unknown classes caused by the model's reliance on shortcuts for prediction. The combined effect of EUG-ADA and CausalAD enables the model to learn more generalized and robust causal evidence for trusted open set recognition. Finally, extensive experimental results on both real-world and synthetic data validate the effectiveness of the proposed method, demonstrating that it improves not only open set recognition performance under covariate shift but also the reliability of uncertainty estimates. The code is released on https://github.com/ScorpioBao/CEL-OSR.
引用
收藏
页码:13085 / 13098
页数:14
相关论文
共 50 条
  • [21] Robust Learning under Uncertain Test Distributions: Relating Covariate Shift to Model Misspecification
    Wen, Junfeng
    Yu, Chun-Nam
    Greiner, Russell
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 32 (CYCLE 2), 2014, 32 : 631 - 639
  • [22] Gait Recognition under Different Covariate Conditions using Deep Learning Technique
    Junaid, Iman
    Ari, Samit
    2022 IEEE INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING AND COMMUNICATIONS, SPCOM, 2022,
  • [23] Open-World Learning Under Dataset Shift
    Srey, Ponhvoan
    Zhang, Yuhui
    Kanamori, Takafumi
    2024 IEEE CONFERENCE ON ARTIFICIAL INTELLIGENCE, CAI 2024, 2024, : 1040 - 1042
  • [24] Open Set Recognition With Incremental Learning for SAR Target Classification
    Ma, Xiaojie
    Ji, Kefeng
    Feng, Sijia
    Zhang, Linbin
    Xiong, Boli
    Kuang, Gangyao
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [25] Classification-Reconstruction Learning for Open-Set Recognition
    Yoshihashi, Ryota
    Shao, Wen
    Kawakami, Rei
    You, Shaodi
    Iida, Makoto
    Naemura, Takeshi
    2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 4011 - 4020
  • [26] Open-Set Plankton Recognition Using Similarity Learning
    Mohamed, Ola Badreldeen Bdawy
    Eerola, Thomas
    Kraft, Kaisa
    Lensu, Lasse
    Kalviainen, Heikki
    ADVANCES IN VISUAL COMPUTING, ISVC 2022, PT I, 2022, 13598 : 174 - 183
  • [27] Learning Discriminative Feature Representation for Open Set Action Recognition
    Zhang, Hongjie
    Liu, Yi
    Wang, Yali
    Wang, Limin
    Qiao, Yu
    PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2023, 2023, : 7696 - 7705
  • [28] Open-set iris recognition based on deep learning
    Sun, Jie
    Zhao, Shipeng
    Miao, Sheng
    Wang, Xuan
    Yu, Yanan
    IET IMAGE PROCESSING, 2022, 16 (09) : 2361 - 2372
  • [29] IMPROVING OPEN-SET RECOGNITION WITH BAYESIAN METRIC LEARNING
    Chen, Tong
    Feng, Guanchao
    Djuric, Petar M.
    2024 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING, ICASSP 2024, 2024, : 6185 - 6189
  • [30] Learning to Generate the Unknowns as a Remedy to the Open-Set Domain Shift
    Baktashmotlagh, Mahsa
    Chen, Tianle
    Salzmann, Mathieu
    2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, : 3737 - 3746