arXiv:2412.03058cs.CV2024-12

用简单变换生成的异常数据提升模型识别分布外样本的能力

Revisiting Energy-Based Model for Out-of-Distribution Detection

  • 用简单数据变换生成边缘分布数据替代人工构造异常样本
  • 提出能量屏障损失,使分布内/外样本能量差更明显
  • 在多个基准上优于或相当当前最优方法,理论基础扎实

分布外(OOD)检测是提升深度学习模型鲁棒性的关键方法,可识别训练分布之外的输入。现有方法通常依赖人工设计的数据,如特定异常数据集或复杂数据增强,但此类数据与真实分布外数据常存在不匹配,限制了模型的泛化能力。为此,我们提出通过简单变换生成的异常暴露(OEST)框架,利用“边缘分布”(PD)数据来增强检测性能。PD数据由简单数据变换生成,无需人工标注。我们采用能量基础模型(EBMs)研究这些数据,并发现“能量屏障”概念——即分布内与分布外样本间的能量差异有助于检测。在训练中引入PD数据以建立能量屏障。进一步地,该概念启发了理论驱动的能量屏障损失,替代传统能量约束损失,形成改进的OEST*范式,在分布内与分布外样本间实现更有效、更合理的分离。我们在多个基准上进行了实证验证,大量实验表明,OEST*在性能上优于或相当于当前最优方法。

原文摘要 · Abstract (English)

Out-of-distribution (OOD) detection is an essential approach to robustifying deep learning models, enabling them to identify inputs that fall outside of their trained distribution. Existing OOD detection methods usually depend on crafted data, such as specific outlier datasets or elaborate data augmentations. While this is reasonable, the frequent mismatch between crafted data and OOD data limits model robustness and generalizability. In response to this issue, we introduce Outlier Exposure by Simple Transformations (OEST), a framework that enhances OOD detection by leveraging "peripheral-distribution" (PD) data. Specifically, PD data are samples generated through simple data transformations, thus providing an efficient alternative to manually curated outliers. We adopt energy-based models (EBMs) to study PD data. We recognize the "energy barrier" in OOD detection, which characterizes the energy difference between in-distribution (ID) and OOD samples and eases detection. PD data are introduced to establish the energy barrier during training. Furthermore, this energy barrier concept motivates a theoretically grounded energy-barrier loss to replace the classical energy-bounded loss, leading to an improved paradigm, OEST*, which achieves a more effective and theoretically sound separation between ID and OOD samples. We perform empirical validation of our proposal, and extensive experiments across various benchmarks demonstrate that OEST* achieves better or similar accuracy compared with state-of-the-art methods.

OOD检测能量模型异常检测

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