用小波变换优化工业图像异常检测,提升定位精度
Wavelet-Enhanced PaDiM for Industrial Anomaly Detection
- 将小波变换嵌入多层特征,按频率选择关键信息
- 在MVTec数据集上达99.32%图像AUC、92.10%像素AUC
- 结果可解释,适合工业质检场景部署
工业图像中的异常检测与定位对自动化质量检查至关重要。PaDiM方法通过预训练卷积神经网络提取正常图像特征,但采用随机通道选择降维,可能丢失结构化信息。本文提出小波增强型PaDiM(WE-PaDiM),将离散小波变换(DWT)与多层CNN特征以结构化方式结合。WE-PaDiM对多个主干网络层的特征图应用二维DWT,选取特定频带(如LL、LH、HL),进行空间对齐后通道拼接,再使用PaDiM的多元高斯框架建模。该‘小波前拼接’策略基于频率内容实现有原则的特征选择,利用多尺度小波信息替代随机选择。我们在包含多个骨干网络(ResNet-18和EfficientNet B0-B6)的MVTec AD数据集上评估,方法在15个类别上实现平均99.32%图像AUC和92.10%像素AUC。分析表明,小波类型影响性能权衡:简单小波(如Haar)搭配细节子带(HL或LH/HL/HH)常提升定位效果,而近似子带(LL)改善图像级检测。WE-PaDiM为PaDiM中随机特征选择提供了高效且可解释的替代方案,具备工业检测所需的鲁棒性与相近效率。
原文摘要 · Abstract (English)
Anomaly detection and localization in industrial images are essential for automated quality inspection. PaDiM, a prominent method, models the distribution of normal image features extracted by pre-trained Convolutional Neural Networks (CNNs) but reduces dimensionality through random channel selection, potentially discarding structured information. We propose Wavelet-Enhanced PaDiM (WE-PaDiM), which integrates Discrete Wavelet Transform (DWT) analysis with multi-layer CNN features in a structured manner. WE-PaDiM applies 2D DWT to feature maps from multiple backbone layers, selects specific frequency subbands (e.g., LL, LH, HL), spatially aligns them, and concatenates them channel-wise before modeling with PaDiM's multivariate Gaussian framework. This DWT-before-concatenation strategy provides a principled method for feature selection based on frequency content relevant to anomalies, leveraging multi-scale wavelet information as an alternative to random selection. We evaluate WE-PaDiM on the challenging MVTec AD dataset with multiple backbones (ResNet-18 and EfficientNet B0-B6). The method achieves strong performance in anomaly detection and localization, yielding average results of 99.32% Image-AUC and 92.10% Pixel-AUC across 15 categories with per-class optimized configurations. Our analysis shows that wavelet choices affect performance trade-offs: simpler wavelets (e.g., Haar) with detail subbands (HL or LH/HL/HH) often enhance localization, while approximation bands (LL) improve image-level detection. WE-PaDiM thus offers a competitive and interpretable alternative to random feature selection in PaDiM, achieving robust results suitable for industrial inspection with comparable efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。