提出双阶段能量模型,让主动学习在未知类别存在时仍能精准选样。
Energy-Based Open-Set Active Learning for Object Classification

- 用两个能量模型分别区分已知/未知样本并评估样本价值。
- 在CIFAR-10、ModelNet40等数据集上显著提升标注效率与分类精度。
- 适合处理现实场景中存在未知类别的对象分类任务。
主动学习通过从大量未标注数据中选择最有价值的样本进行标注,有效降低深度学习的标注成本。传统主动学习基于封闭集假设,即所有类别均已知且一致,但在真实场景中常面临开放集环境,即未标注数据包含已知和未知类别。此时,标准方法易误选未知类别样本,导致标注预算浪费。本文提出一种新型双阶段能量基框架用于开放集主动学习。方法采用两个专用的能量基模型(EBMs):第一个为已知/未知分离器,过滤掉可能属于未知类别的样本;第二个为样本评分器,评估筛选后已知类样本的信息量。利用能量景观,模型为已知样本分配较低能量,未知样本分配较高能量,确保仅选择目标类别样本进行标注。该框架实现高效精准的样本选择,在2D(CIFAR-10、CIFAR-100、TinyImageNet)和3D(ModelNet40)物体分类基准测试中优于现有方法,显著提升开放集环境下的标注效率与分类性能。
原文摘要 · Abstract (English)
Active learning (AL) has emerged as a crucial methodology for minimizing labeling costs in deep learning by selecting the most valuable samples from a pool of unlabeled data for annotation. Traditional AL operates under a closed-set assumption, where all classes in the dataset are known and consistent. However, real-world scenarios often present open-set conditions in which unlabeled data contains both known and unknown classes. In such environments, standard AL techniques struggle. They can mistakenly query samples from unknown categories, leading to inefficient use of annotation budgets. In this paper, we propose a novel dual-stage energy-based framework for open-set AL. Our method employs two specialized energy-based models (EBMs). The first, an energy-based known/unknown separator, filters out samples likely to belong to unknown classes. The second, an energy-based sample scorer, assesses the informativeness of the filtered known samples. Using the energy landscape, our models distinguish between data points from known and unknown classes in the unlabeled pool by assigning lower energy to known samples and higher energy to unknown samples, ensuring that only samples from classes of interest are selected for labeling. By integrating these components, our approach ensures efficient and targeted sample selection, maximizing learning impact in each iteration. Experiments on 2D (CIFAR-10, CIFAR-100, TinyImageNet) and 3D (ModelNet40) object classification benchmarks demonstrates that our framework outperforms existing approaches, achieving superior annotation efficiency and classification performance in open-set environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。