论文标题

用对抗镜像自动编码器无监督的异常检测

Unsupervised Anomaly Detection with Adversarial Mirrored AutoEncoders

论文作者

Somepalli, Gowthami, Wu, Yexin, Balaji, Yogesh, Vinzamuri, Bhanukiran, Feizi, Soheil

论文摘要

在所有机器学习应用中,检测到分布(OOD)样本至关重要。深层生成建模已成为主要的范式,以模拟没有标签的复杂数据分布。但是,先前的工作表明,与培训的数据分布相比,生成模型倾向于将更高的可能性分配给OOD样品。首先,我们提出了对抗镜像自动编码器(AMA),这是对抗自动编码器的变体,该变体使用歧视器中的镜像瓦斯坦损失来实施更好的语义级别重建。我们还提出了一个潜在空间正则化,以学习分布样品的紧凑型歧管。 AMA的使用产生更好的特征表示,以改善异常检测性能。其次,我们提出了一种替代基于重建的度量的异常评分的替代度量,该度量传统上是在基于生成模型的异常检测方法中使用的。我们的方法的表现优于几个OOD检测基准测试的当前最新方法。

Detecting out of distribution (OOD) samples is of paramount importance in all Machine Learning applications. Deep generative modeling has emerged as a dominant paradigm to model complex data distributions without labels. However, prior work has shown that generative models tend to assign higher likelihoods to OOD samples compared to the data distribution on which they were trained. First, we propose Adversarial Mirrored Autoencoder (AMA), a variant of Adversarial Autoencoder, which uses a mirrored Wasserstein loss in the discriminator to enforce better semantic-level reconstruction. We also propose a latent space regularization to learn a compact manifold for in-distribution samples. The use of AMA produces better feature representations that improve anomaly detection performance. Second, we put forward an alternative measure of anomaly score to replace the reconstruction-based metric which has been traditionally used in generative model-based anomaly detection methods. Our method outperforms the current state-of-the-art methods for anomaly detection on several OOD detection benchmarks.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源