11 天前

基于单类嵌入的逆蒸馏异常检测

Hanqiu Deng, Xingyu Li
基于单类嵌入的逆蒸馏异常检测
摘要

知识蒸馏(Knowledge Distillation, KD)在具有挑战性的无监督异常检测(Anomaly Detection, AD)任务中取得了令人瞩目的成果。在教师-学生(Teacher-Student, T-S)模型中,异常样本在特征表示上的差异为异常检测提供了关键证据。然而,以往研究通常采用相似或相同的网络架构构建教师与学生模型,这限制了异常表征的多样性。为解决这一问题,本文提出一种新型T-S架构,其由教师编码器与学生解码器构成,并相应引入一种简单而有效的“反向蒸馏”(reverse distillation)范式。与传统方法不同,学生网络不再直接接收原始图像作为输入,而是以教师模型生成的一类嵌入(one-class embedding)为输入,目标是恢复教师模型在多尺度下的特征表示。该方法本质上实现了从抽象的高层语义表征向低层特征的知识迁移。此外,本文在T-S模型中引入了一个可学习的一类瓶颈嵌入(One-Class Bottleneck Embedding, OCBE)模块。该模块生成的紧凑嵌入能够有效保留正常模式的核心信息,同时抑制异常扰动。在多个异常检测与一类新奇性检测(one-class novelty detection)基准数据集上的大量实验表明,所提方法显著超越了现有最先进(SOTA)性能,充分验证了所提方法的有效性与良好的泛化能力。

基于单类嵌入的逆蒸馏异常检测 | 最新论文 | HyperAI超神经