arXiv:2508.09178cs.CVcs.AI2025-08AAAI被引 21

用两阶段训练让视觉语言模型更准地识别工业缺陷。

IAD-R1: Reinforcing Consistent Reasoning in Industrial Anomaly Detection

  • 构建思维链数据集,提升模型看懂异常的能力。
  • 在DAGM数据集上平均准确率提高43.3%。
  • 小模型零样本性能超商用大模型,适合工业落地。

工业异常检测是现代制造的关键环节,但缺陷样本稀缺限制了传统方法的应用。尽管视觉语言模型(VLMs)具备较强的泛化能力,其在工业异常检测中的表现仍受限。为此,我们提出IAD-R1,一种适用于不同架构和参数规模VLM的通用后训练框架,显著提升其异常检测能力。IAD-R1采用两阶段训练策略:第一阶段感知激活监督微调(PA-SFT)利用精心构建的高质量思维链数据集(Expert-AD)进行训练,增强异常感知能力并建立推理与答案间的关联;第二阶段结构化控制组相对策略优化(SC-GRPO)通过设计的奖励函数,实现从“异常感知”到“异常解释”的能力跃升。实验表明,IAD-R1在7种VLM上均取得显著提升,最大提升出现在DAGM数据集,平均准确率比0.5B基线高43.3%。值得注意的是,使用IAD-R1训练的0.5B模型在零样本设置下超越GPT-4.1和Claude-Sonnet-4等商用模型,验证了IAD-R1的有效性与优越性。数据集、代码及所有模型权重将公开于https://github.com/Yanhui-Lee/IAD-R1。

原文摘要 · Abstract (English)

Industrial anomaly detection is a critical component of modern manufacturing, yet the scarcity of defective samples restricts traditional detection methods to scenario-specific applications. Although Vision-Language Models (VLMs) demonstrate significant advantages in generalization capabilities, their performance in industrial anomaly detection remains limited. To address this challenge, we propose IAD-R1, a universal post-training framework applicable to VLMs of different architectures and parameter scales, which substantially enhances their anomaly detection capabilities. IAD-R1 employs a two-stage training strategy: the Perception Activation Supervised Fine-Tuning (PA-SFT) stage utilizes a meticulously constructed high-quality Chain-of-Thought dataset (Expert-AD) for training, enhancing anomaly perception capabilities and establishing reasoning-to-answer correlations; the Structured Control Group Relative Policy Optimization (SC-GRPO) stage employs carefully designed reward functions to achieve a capability leap from "Anomaly Perception" to "Anomaly Interpretation". Experimental results demonstrate that IAD-R1 achieves significant improvements across 7 VLMs, the largest improvement was on the DAGM dataset, with average accuracy 43.3% higher than the 0.5B baseline. Notably, the 0.5B parameter model trained with IAD-R1 surpasses commercial models including GPT-4.1 and Claude-Sonnet-4 in zero-shot settings, demonstrating the effectiveness and superiority of IAD-R1. The dataset, code, and all model weights will be publicly available at https://github.com/Yanhui-Lee/IAD-R1.

异常检测视觉语言模型工业质检后训练

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