解决开放集识别中未知样本误判问题,提升模型判别能力。
Known Meets Unknown: Mitigating Overconfidence in Open Set Recognition
- 通过参数扰动估计预测不确定性,量化模型置信度
- 采用双阶段分类器增强已知与未知类的区分能力
- 在多个公开数据集上优于现有方法,尤其对语义相似未知样本有效
开放集识别(OSR)要求模型不仅能准确分类已知类别,还需有效拒绝未知样本。然而当未知样本在语义上与已知类别接近时,特征空间中的类间重叠会导致模型对其赋予过高的置信度,从而错误地将其归为已知类别,即产生过自信现象。这种过自信模糊了已知与未知类之间的决策边界,削弱了OSR性能。为此,本文提出一种显式缓解因类间重叠导致过自信的框架。该框架包含两个组件:基于扰动的不确定性估计模块,通过可控参数扰动生成多样化预测并量化预测不确定性;以及具有独立学习分类器的未知检测模块,采用两阶段流程,利用估计的不确定性提升已知与未知类的判别能力,从而增强整体OSR表现。在三个公开数据集上的实验表明,所提框架在性能上显著优于现有OSR方法。
原文摘要 · Abstract (English)
Open Set Recognition (OSR) requires models not only to accurately classify known classes but also to effectively reject unknown samples. However, when unknown samples are semantically similar to known classes, inter-class overlap in the feature space often causes models to assign unjustifiably high confidence to them, leading to misclassification as known classes -- a phenomenon known as overconfidence. This overconfidence undermines OSR by blurring the decision boundary between known and unknown classes. To address this issue, we propose a framework that explicitly mitigates overconfidence caused by inter-class overlap. The framework consists of two components: a perturbation-based uncertainty estimation module, which applies controllable parameter perturbations to generate diverse predictions and quantify predictive uncertainty, and an unknown detection module with distinct learning-based classifiers, implemented as a two-stage procedure, which leverages the estimated uncertainty to improve discrimination between known and unknown classes, thereby enhancing OSR performance. Experimental results on three public datasets show that the proposed framework achieves superior performance over existing OSR methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。