用注意力机制提升3D打印缺陷检测精度,尤其擅长捕捉微小缺陷。
GuidedFlow: An Attention-Guided Framework for Anomaly Detection in Additive Manufacturing

- 引入注意力引导的流模型,动态聚焦关键区域和时间帧。
- 在真实3D打印数据集上达到98.7%检测准确率与0.96 AUROC。
- 适合工业质检场景,尤其对小样本和微小缺陷敏感的任务。
增材制造(AM)在工业革命中至关重要,但质量控制仍具挑战性,主要源于打印缺陷或潜在的网络物理入侵。基于图像或视频的异常检测是应对该问题的关键手段。现有方法包括基于重构、嵌入和流的方法。尽管基于归一化流的方法在处理未知缺陷和泛化能力方面表现良好,但在3D打印中常见的微小/丝状缺陷检测上仍存在不足,尤其在小样本条件下限制了泛化性能。为此,我们提出 extbf{GuidedFlow}——一种新颖的注意力引导归一化流模型,用于异常检测与定位。GuidedFlow采用在领域数据集上微调的预训练ResNet模型,构建时空注意力框架,实现多尺度与多帧动态建模。空间-时间注意力网络(SAN)使流模型能够优先关注输入帧中的相关上下文线索。我们在包含真实3D打印物体图像与视频的AM3D-AD数据集上评估了GuidedFlow,同时在MVTec-AD工业图像异常检测数据集上进行对比实验。结果表明,GuidedFlow优于多数前沿模型,检测准确率与AUROC均有显著提升。
原文摘要 · Abstract (English)
Additive Manufacturing (AM) plays a vital role in the ongoing industrial revolution. However, quality control remains crucial and challenging due to printing defects or potential cyber-physical intrusions. Image or video-based anomaly detection is a key effort towards addressing these challenges. Various approaches have been explored in this domain, including reconstruction-based, embedding-based, and flow-based methods. Though normalizing flow-based methods address some of the core challenges of unforeseen defects and generalization while maintaining detection performance, existing approaches struggle with tiny/stringing defects common in 3D printing. In a small-data setting, this poses a limitation in generalization. To address these limitations, we propose \textbf{GuidedFlow}, a novel attention-guided normalizing flow model for anomaly detection and localization. GuidedFlow employs a pre-trained ResNet model, fine-tuned on the domain dataset. An attention-guided spatial and temporal flow framework models the dynamics across multiple scales and frames. A Spatio-Temporal Attention Network (SAN) enables the flow model to prioritize relevant contextual cues from input frames. We evaluate GuidedFlow on our AM3D-AD dataset, consisting of benign and anomalous real 3D printed object images and videos. We also conduct a comparative study using the MVTec-AD industrial image anomaly detection dataset. Experimental results demonstrate that GuidedFlow outperforms most of the state-of-the-art models with enhanced detection accuracy and AUROC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。