提出新方法动态判断新样本归入旧类或创建新类
DP-BOA: Dirichlet-Process Birth-or-Assign for On-the-Fly Category Discovery

- 基于狄利克雷过程的后验预测框架,显式比较新旧类别证据
- 在标准基准上新类发现性能显著优于基线,旧类准确率也领先
- 适合需要实时新增类别的场景,如持续学习和开放世界识别
在线类别发现需判断每个测试样本应归属已有类别还是生成新类别。现有方法多采用半径或哈希等匹配启发式规则,通常将类别创建视为匹配失败后的备选方案,而非基于独立统计证据的主动决策。为此,本文提出DP-BOA,一种基于在线狄利克雷过程高斯混合模型与正态逆威沙特先验的后验预测决策框架。训练时,利用标注数据校准类别高斯分布的共享NIW先验,并初始化已知类别的后验;测试时,对每个输入样本,比较其分配至已有类别的后验预测证据与由狄利克雷过程先验诱导的创建新类的证据,并在决策后在线更新类别统计量。该方法能捕捉异向类别几何结构,并随证据积累自然调整决策置信度。在多个标准OCD基准上,DP-BOA持续优于强基线,在新类发现任务中表现尤为突出,同时保持优异的已知类识别精度。
原文摘要 · Abstract (English)
On-the-fly category discovery requires deciding for each incoming test sample whether to assign it to an existing category or spawn a new one. Existing methods typically implement this decision through matching-based heuristics, such as radius- or hash-based rules. While effective in practice, these methods usually treat category birth implicitly as a fallback when no existing category matches confidently, rather than as an explicit alternative supported by its own statistical evidence. To address this, we propose DP-BOA, a posterior-predictive decision framework based on an online Dirichlet-process Gaussian mixture model with a Normal-Inverse-Wishart prior. During training, we use labeled data to calibrate a shared NIW prior over category Gaussians and warm-start the known-category posteriors. At test time, for each incoming sample, DP-BOA compares the posterior predictive evidence for assignment to existing categories against the evidence for spawning a new category induced by the DP prior, and then updates category statistics online after the decision. The method captures anisotropic category geometry and naturally adapts decision confidence as evidence accumulates. Across standard OCD benchmarks, DP-BOA consistently outperforms strong baselines and delivers particularly strong novel-class discovery performance while maintaining competitive known-class accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。