用贝叶斯方法提升工业图像异常检测在少样本下的鲁棒性
BayPrAnoMeta: Bayesian Proto-MAML for Few-Shot Industrial Image Anomaly Detection
- 用概率模型替代传统原型,实现不确定性感知的异常评分
- 在少样本下相比基线方法平均提升1.8% AUROC,最高达4.2%
- 适合工业界部署,尤其适用于缺陷样本稀缺的场景
工业图像异常检测因类别极度不平衡和标注缺陷样本稀少而极具挑战,尤其在少样本设置下。本文提出 BayPrAnoMeta,一种面向少样本工业图像异常检测的贝叶斯原型元学习方法。与现有基于确定性原型和距离度量的 Proto-MAML 方法不同,BayPrAnoMeta 将原型替换为任务相关的概率正常态模型,并通过贝叶斯后验预测似然进行内循环优化。我们采用 Normal-Inverse-Wishart (NIW) 先验建模正常支持嵌入,生成 Student-t 预测分布,实现具有不确定性的重尾异常打分,在极端少样本条件下显著增强鲁棒性。此外,我们将 BayPrAnoMeta 扩展至联邦元学习框架,引入监督对比正则化以应对异构工业客户端,并证明了非凸目标函数收敛至稳定点。在 MVTec AD 基准测试中,该方法在少样本异常检测设置下持续且显著优于 MAML、Proto-MAML 和 PatchCore 方法,平均 AUROC 提升 1.8%,最高达 4.2%。
原文摘要 · Abstract (English)
Industrial image anomaly detection is a challenging problem owing to extreme class imbalance and the scarcity of labeled defective samples, particularly in few-shot settings. We propose BayPrAnoMeta, a Bayesian generalization of Proto-MAML for few-shot industrial image anomaly detection. Unlike existing Proto-MAML approaches that rely on deterministic class prototypes and distance-based adaptation, BayPrAnoMeta replaces prototypes with task-specific probabilistic normality models and performs inner-loop adaptation via a Bayesian posterior predictive likelihood. We model normal support embeddings with a Normal-Inverse-Wishart (NIW) prior, producing a Student-$t$ predictive distribution that enables uncertainty-aware, heavy-tailed anomaly scoring and is essential for robustness in extreme few-shot settings. We further extend BayPrAnoMeta to a federated meta-learning framework with supervised contrastive regularization for heterogeneous industrial clients and prove convergence to stationary points of the resulting nonconvex objective. Experiments on the MVTec AD benchmark demonstrate consistent and significant AUROC improvements over MAML, Proto-MAML, and PatchCore-based methods in few-shot anomaly detection settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。