3 个月前

旧者为金:重新定义对抗性学习的一类分类器训练范式

旧者为金:重新定义对抗性学习的一类分类器训练范式

摘要

一种流行的异常检测方法是利用对抗网络中的生成器,基于输入数据的重建误差来构建异常得分。然而,由于异常事件本身发生频率极低,此类网络的优化过程往往十分繁琐。另一种可能的思路是同时利用生成器与判别器进行异常检测。但受限于对抗训练机制的存在,该类模型通常表现出不稳定性,导致性能在每个训练步骤间剧烈波动。在本研究中,我们提出了一种新框架,能够在一个广泛的训练步数范围内生成稳定可靠的检测结果,并有效结合对抗模型中的生成器与判别器,实现高效且鲁棒的异常检测。我们的方法对判别器的根本功能进行了重构:不再用于区分真实数据与伪造数据,而是转变为识别重建结果的优劣质量。具体而言,我们通过当前生成器生成高质量重建样本作为正例,而利用同一生成器的旧状态所产生的重建结果作为低质量样本作为负例。由此,判别器能够学习识别异常输入在重建过程中常出现的细微失真特征。在Caltech-256和MNIST图像数据集上的新奇性检测任务中,大量实验验证了本方法的优越性能。此外,在UCSD Ped2视频异常检测数据集上,我们的模型在帧级别实现了98.1%的AUC值,显著优于当前最先进的方法。

代码仓库

xaggi/OGNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-mnist-testOGNET
F1 score: 96.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
旧者为金:重新定义对抗性学习的一类分类器训练范式 | 论文 | HyperAI超神经