arXiv:2603.15650cs.LGcs.CV2026-03

让模型自动增减原型,更好识别分布外数据。

How to Achieve Prototypical Birth and Death for OOD Detection?

  • 基于生物细胞的生灭机制,动态调整原型数量。
  • 在CIFAR-100上FPR95指标达到当前最优水平。
  • 适合需要高可靠性OOD检测的部署场景。

分布外(OOD)检测对机器学习模型的安全部署至关重要,原型方法是主流策略之一。现有方法通常采用固定数量的原型,难以适应不同类别间固有的复杂度差异。目前尚缺乏根据数据复杂度自适应调整原型数的机制。受生物学中细胞出生与死亡过程启发,本文提出新型方法PID(原型生灭),在训练过程中通过原型生与死两个动态机制实现自适应调整。出生机制通过识别现有原型的过载程度,在表示不足的数据区域生成新原型,以精细捕捉类内子结构;死亡机制则通过评估原型的可分性,剔除边界模糊的原型,强化分类边界。该方法使原型数量随数据复杂度动态变化,从而学习到更紧凑、分离度更高的分布内嵌入,显著提升对分布外样本的检测能力。实验表明,所提方法在CIFAR-100等基准上显著优于现有方法,尤其在FPR95指标上达到当前最优(SOTA)性能。

原文摘要 · Abstract (English)

Out-of-Distribution (OOD) detection is crucial for the secure deployment of machine learning models, and prototype-based learning methods are among the mainstream strategies for achieving OOD detection. Existing prototype-based learning methods generally rely on a fixed number of prototypes. This static assumption fails to adapt to the inherent complexity differences across various categories. Currently, there is still a lack of a mechanism that can adaptively adjust the number of prototypes based on data complexity. Inspired by the processes of cell birth and death in biology, we propose a novel method named PID (Prototype bIrth and Death) to adaptively adjust the prototype count based on data complexity. This method relies on two dynamic mechanisms during the training process: prototype birth and prototype death. The birth mechanism instantiates new prototypes in data regions with insufficient representation by identifying the overload level of existing prototypes, thereby meticulously capturing intra-class substructures. Conversely, the death mechanism reinforces the decision boundary by pruning prototypes with ambiguous class boundaries through evaluating their discriminability. Through birth and death, the number of prototypes can be dynamically adjusted according to the data complexity, leading to the learning of more compact and better-separated In-Distribution (ID) embeddings, which significantly enhances the capability to detect OOD samples. Experiments demonstrate that our dynamic method, PID, significantly outperforms existing methods on benchmarks such as CIFAR-100, achieving State-of-the-Art (SOTA) performance, especially on the FPR95 metric.

OOD检测原型学习自适应动态机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。