用制造流程指导视觉模型,提升工业缺陷检测精度
Triad: Empowering LMM-based Anomaly Detection with Vision Expert-guided Visual Tokenizer and Manufacturing Process
- 引入专家标注的感兴趣区域,让大模型聚焦缺陷位置
- 结合制造流程信息,准确率优于现有方法
- 适合工业质检场景,尤其需理解缺陷成因的研究者
尽管近期研究尝试将大模型引入工业缺陷检测(IAD),但其泛化能力远低于通用任务。我们总结出两大原因:一是通用大模型缺乏对视觉缺陷的认知,难以关注缺陷区域;为此,我们修改了LLaVA的AnyRes结构,将现有IAD模型识别的潜在异常区域输入大模型。二是现有方法仅依赖缺陷模式或与正常样本对比,未能理解缺陷成因;鉴于缺陷生成与制造流程密切相关,我们提出制造驱动的IAD范式。为此设计了面向IAD的指令微调数据集InstructIAD,以及支持制造流程链式思考(CoT-M)的数据组织方式。基于上述改进,提出Triad方法,融合专家引导的感兴趣区域分词器与制造流程信息,实现更精准的工业缺陷检测。大量实验表明,Triad不仅性能媲美当前主流大模型,且在引入制造流程后进一步提升准确率。代码、训练数据及预训练模型将公开于https://github.com/tzjtatata/Triad。
原文摘要 · Abstract (English)
Although recent methods have tried to introduce large multimodal models (LMMs) into industrial anomaly detection (IAD), their generalization in the IAD field is far inferior to that for general purposes. We summarize the main reasons for this gap into two aspects. On one hand, general-purpose LMMs lack cognition of defects in the visual modality, thereby failing to sufficiently focus on defect areas. Therefore, we propose to modify the AnyRes structure of the LLaVA model, providing the potential anomalous areas identified by existing IAD models to the LMMs. On the other hand, existing methods mainly focus on identifying defects by learning defect patterns or comparing with normal samples, yet they fall short of understanding the causes of these defects. Considering that the generation of defects is closely related to the manufacturing process, we propose a manufacturing-driven IAD paradigm. An instruction-tuning dataset for IAD (InstructIAD) and a data organization approach for Chain-of-Thought with manufacturing (CoT-M) are designed to leverage the manufacturing process for IAD. Based on the above two modifications, we present Triad, a novel LMM-based method incorporating an expert-guided region-of-interest tokenizer and manufacturing process for industrial anomaly detection. Extensive experiments show that our Triad not only demonstrates competitive performance against current LMMs but also achieves further improved accuracy when equipped with manufacturing processes. Source code, training data, and pre-trained models will be publicly available at https://github.com/tzjtatata/Triad.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。