AISep 26, 2024

Dirichlet-Based Coarse-to-Fine Example Selection For Open-Set Annotation

arXiv:2409.17607v1h-index: 5
Originality Incremental advance
AI Analysis

This addresses the challenge of open-set noise in active learning for real-world applications, representing an incremental improvement.

The paper tackles the problem of active learning deteriorating in open-set annotation scenarios by proposing a Dirichlet-based coarse-to-fine selection strategy, which achieves state-of-the-art performance across various openness ratio datasets.

Active learning (AL) has achieved great success by selecting the most valuable examples from unlabeled data. However, they usually deteriorate in real scenarios where open-set noise gets involved, which is studied as open-set annotation (OSA). In this paper, we owe the deterioration to the unreliable predictions arising from softmax-based translation invariance and propose a Dirichlet-based Coarse-to-Fine Example Selection (DCFS) strategy accordingly. Our method introduces simplex-based evidential deep learning (EDL) to break translation invariance and distinguish known and unknown classes by considering evidence-based data and distribution uncertainty simultaneously. Furthermore, hard known-class examples are identified by model discrepancy generated from two classifier heads, where we amplify and alleviate the model discrepancy respectively for unknown and known classes. Finally, we combine the discrepancy with uncertainties to form a two-stage strategy, selecting the most informative examples from known classes. Extensive experiments on various openness ratio datasets demonstrate that DCFS achieves state-of-art performance.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes