3 个月前

显式边界引导的半推半拉对比学习用于监督式异常检测

显式边界引导的半推半拉对比学习用于监督式异常检测

摘要

大多数异常检测(Anomaly Detection, AD)模型采用无监督方式仅基于正常样本进行训练,这可能导致决策边界模糊且判别能力不足。事实上,在现实应用场景中,往往存在少量已知的异常样本,这些已知异常所蕴含的有价值信息也应被有效利用。然而,在训练过程中引入少量已知异常样本可能引发新的问题:模型可能因过度依赖这些已知异常而产生偏差,从而难以泛化到未见过的异常样本。本文针对监督式异常检测问题,即利用少量可用的异常样本训练AD模型,目标是同时识别已见异常与未见异常。为此,我们提出一种新颖的显式边界引导的半推拉对比学习机制(Explicit Boundary Guided Semi-Push-Pull Contrastive Learning),该机制能够在提升模型判别能力的同时,有效缓解由已知异常带来的偏差问题。本方法基于两个核心设计:第一,我们显式地学习一个紧凑的分离边界,作为后续特征学习的引导。由于该边界仅依赖于正常样本的特征分布,因而可显著缓解因少量已知异常引入的偏差问题。第二,我们设计了一种边界引导的半推拉损失函数,该损失函数仅将正常特征向量拉近,同时将异常特征向量从分离边界向外推离至一定边界区域之外。通过这种方式,模型能够构建出更为清晰、具有强判别性的决策边界,从而更有效地将已知异常与未见异常与正常样本区分开来。相关代码将公开于:https://github.com/xcyao00/BGAD。

代码仓库

xcyao00/bgad
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
supervised-anomaly-detection-on-mvtec-adBGAD
Detection AUROC: 99.3
Segmentation AUROC: 99.2

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
显式边界引导的半推半拉对比学习用于监督式异常检测 | 论文 | HyperAI超神经