arXiv:2603.22840cs.CVcs.AI2026-03中稿 · IEEE TCSVT被引 9

通过显式还原异常为正常形态,提升无监督缺陷检测精度

URA-Net: Uncertainty-Integrated Anomaly Perception and Restoration Attention Network for Unsupervised Anomaly Detection

  • 用预训练网络提取多级语义特征作为重建目标
  • 合成人工异常样本并引入不确定性感知模块
  • 基于全局正常语义修复异常区域,适合工业与医疗图像

无监督异常检测在工业缺陷检测和医学图像分析中至关重要,现有方法多依赖重建框架,但易过度泛化,导致异常被良好重构而检测性能下降。为此,本文提出不确定性集成的异常感知与恢复注意力网络(URA-Net),明确将异常模式还原为对应正常状态。首先,不采用传统重建方式,改用预训练卷积神经网络提取多层级语义特征作为重建目标;为辅助模型学习恢复异常,设计新型特征级人工异常合成模块生成训练用异常样本。其次,引入基于贝叶斯神经网络的不确定性集成异常感知模块,学习异常与正常特征分布,实现对异常区域及模糊边界的估计。随后,提出新颖的恢复注意力机制,利用全局正常语义信息修复检测到的异常区域,获得无缺陷的恢复特征。最后,通过输入特征与恢复特征间的残差图进行异常检测与定位。在两个工业数据集MVTec AD、BTAD及一个医学图像数据集OCT-2017上的实验结果充分证明该方法的有效性与优越性。

原文摘要 · Abstract (English)

Unsupervised anomaly detection plays a pivotal role in industrial defect inspection and medical image analysis, with most methods relying on the reconstruction framework. However, these methods may suffer from over-generalization, enabling them to reconstruct anomalies well, which leads to poor detection performance. To address this issue, instead of focusing solely on normality reconstruction, we propose an innovative Uncertainty-Integrated Anomaly Perception and Restoration Attention Network (URA-Net), which explicitly restores abnormal patterns to their corresponding normality. First, unlike traditional image reconstruction methods, we utilize a pre-trained convolutional neural network to extract multi-level semantic features as the reconstruction target. To assist the URA-Net learning to restore anomalies, we introduce a novel feature-level artificial anomaly synthesis module to generate anomalous samples for training. Subsequently, a novel uncertainty-integrated anomaly perception module based on Bayesian neural networks is introduced to learn the distributions of anomalous and normal features. This facilitates the estimation of anomalous regions and ambiguous boundaries, laying the foundation for subsequent anomaly restoration. Then, we propose a novel restoration attention mechanism that leverages global normal semantic information to restore detected anomalous regions, thereby obtaining defect-free restored features. Finally, we employ residual maps between input features and restored features for anomaly detection and localization. The comprehensive experimental results on two industrial datasets, MVTec AD and BTAD, along with a medical image dataset, OCT-2017, unequivocally demonstrate the effectiveness and superiority of the proposed method.

异常检测无监督学习图像修复贝叶斯网络

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