GANomaly
一,简介
异常检测旨在只使用正常样本建模从而能够区分OK样本与NG样本,所面对的场景的数据分布极不平衡, 通常OK样本非常多,而NG样本非常少。
自编码器是异常检测算法中比较经典的模型,它利用大量OK训练一个自编码网络,然后通过原图与重构图像之间的重构误差来检测NG样本,但该方法非常容易受噪声影响,其对NG样本也能够重建,导致所谓的重构误差“崩塌”。
GANomaly 采用编码器-解码器-编码器的模型结构, 同时对“原图-》重建图” 及“原图的高维特征编码->重建图的高维特征编码”进行重构误差约束。另外引入生成对抗网络的对抗训练思想, Encoder-Decoder-Encoder结构当作生成网络G-Net, 又定义了一个判别网络D-Net。
推理接断,用于推断异常的不是原图和重建图的差异,而是第一部分编码器产生的隐空间特征(原图的编码)和第二部分编码器产生的隐空间特征(重建图的编码)的差异。这种方法更关注图片实质内容的差异,对图片中的微小变化不敏感,因而能解决自编码器中易受噪声影响的问题,鲁棒性更好。