arXiv:2512.15319cs.CV2025-12被引 28

用原型学习提升少样本异常检测的特征表达,更准识别细微异常。

Prototypical Learning Guided Context-Aware Segmentation Network for Few-Shot Anomaly Detection

  • 基于原型特征自适应,让正常样本聚类更紧密、与异常分离
  • 8样本下图像级AUROC达94.9%(MVTec)和80.2%(MPDD)
  • 适合工业质检等标注样本极少的真实场景

少样本异常检测(FSAD)指在仅有限正常样本条件下识别目标类别中的异常。现有方法多依赖预训练特征表示,但常忽视预训练特征与目标场景间的领域差异。本文提出原型学习引导的上下文感知分割网络(PCSNet),以缓解领域差异,提升目标场景中特征的描述能力,从而增强检测性能。PCSNet由原型特征适配(PFA)子网络和上下文感知分割(CAS)子网络构成。PFA提取原型特征作为指导,确保正常数据特征紧凑且与异常明显分离;设计像素级差异分类损失,增强对细微异常的区分能力。随后引入CAS子网络实现像素级异常定位,利用伪异常辅助训练。在MVTec和MPDD数据集上的实验表明,PCSNet在8样本场景下图像级AUROC分别达到94.9%和80.2%。真实汽车塑料件质检应用也验证了其在少量训练样本下的优异表现。代码已开源。

原文摘要 · Abstract (English)

Few-shot anomaly detection (FSAD) denotes the identification of anomalies within a target category with a limited number of normal samples. Existing FSAD methods largely rely on pre-trained feature representations to detect anomalies, but the inherent domain gap between pre-trained representations and target FSAD scenarios is often overlooked. This study proposes a Prototypical Learning Guided Context-Aware Segmentation Network (PCSNet) to address the domain gap, thereby improving feature descriptiveness in target scenarios and enhancing FSAD performance. In particular, PCSNet comprises a prototypical feature adaption (PFA) sub-network and a context-aware segmentation (CAS) sub-network. PFA extracts prototypical features as guidance to ensure better feature compactness for normal data while distinct separation from anomalies. A pixel-level disparity classification loss is also designed to make subtle anomalies more distinguishable. Then a CAS sub-network is introduced for pixel-level anomaly localization, where pseudo anomalies are exploited to facilitate the training process. Experimental results on MVTec and MPDD demonstrate the superior FSAD performance of PCSNet, with 94.9% and 80.2% image-level AUROC in an 8-shot scenario, respectively. Real-world applications on automotive plastic part inspection further demonstrate that PCSNet can achieve promising results with limited training samples. Code is available at https://github.com/yuxin-jiang/PCSNet.

异常检测少样本学习原型学习工业质检

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