摘要
arXiv:2505.09263v1 异常检测类型: 横跨领域
摘要:异常检测是一项实际而又充满挑战的任务,因为在工业检查中异常样本稀缺。一些现有的异常检测方法通过添加噪声或外部数据来合成异常,解决了这一问题。然而,合成的异常与现实世界中的异常之间始终存在较大的语义差距,导致异常检测性能较弱。为了解决这一问题,我们提出了一种少样本驱动生成(AnoGen)方法,该方法仅使用少量真实的异常样本来引导扩散模型生成真实且多样的异常,从而能够提高异常检测模型的训练效果。具体而言,我们的工作分为三个阶段。在第一个阶段,我们基于给定的少量真实异常样本学习异常分布,并将学习到的知识注入到嵌入中。在第二个阶段,我们使用嵌入和给定的边界框来引导扩散模型在特定对象(或纹理)上生成真实且多样的异常。在最终阶段,我们提出了一种弱监督异常检测方法,使用生成的异常来训练一个更强大的模型。我们的方法以DRAEM和DesTSeg为基础模型,并在常用的工业异常检测数据集MVTec上进行了实验。实验结果显示,我们生成的异常能有效提高异常分类和分割任务的模型性能,例如,DRAEM和DesTSeg分别在分割任务中的AU-PR指标上分别实现了5.8%和1.5%的提升。我们的代码和生成的异常数据可在https://github.com/gaobb/AnoGen获取。