论文标题

显式边界指导的半循环对比度学习,用于监督异常检测

Explicit Boundary Guided Semi-Push-Pull Contrastive Learning for Supervised Anomaly Detection

论文作者

Yao, Xincheng, Li, Ruoqi, Zhang, Jing, Sun, Jun, Zhang, Chongyang

论文摘要

大多数异常检测(AD)模型都是仅以无监督的方式学习的,这可能导致决策界限和可区分性不足。实际上,在现实世界应用中通常可以使用一些异常样本,也应有效利用已知异常的宝贵知识。但是,在训练中利用一些已知的异常情况可能会导致另一个问题,即该模型可能会受到已知异常的偏见,并且未能概括地看不见的异常。在本文中,我们解决了监督的异常检测,即,我们使用一些可用的异常来学习AD模型,以检测可见和看不见的异常。我们提出了一种新型的显式边界指导的半孔子对比度学习机制,该机制可以增强模型的可区分性,同时减轻偏置问题。我们的方法基于两个核心设计:首先,我们找到一个明确而紧凑的分离边界作为进一步特征学习的指导。由于边界仅依赖于正常特征分布,因此可以缓解少数已知异常引起的偏差问题。其次,开发了边界引导的半循环损失,以将正常特征融合在一起,同时将异常特征推开以外的分离边界以外的区域以外的区域。通过这种方式,我们的模型可以形成更明确,更歧视的决策边界,以更有效地区分已知和看不见的异常与正常样本。代码将在https://github.com/xcyao00/bgad上找到。

Most anomaly detection (AD) models are learned using only normal samples in an unsupervised way, which may result in ambiguous decision boundary and insufficient discriminability. In fact, a few anomaly samples are often available in real-world applications, the valuable knowledge of known anomalies should also be effectively exploited. However, utilizing a few known anomalies during training may cause another issue that the model may be biased by those known anomalies and fail to generalize to unseen anomalies. In this paper, we tackle supervised anomaly detection, i.e., we learn AD models using a few available anomalies with the objective to detect both the seen and unseen anomalies. We propose a novel explicit boundary guided semi-push-pull contrastive learning mechanism, which can enhance model's discriminability while mitigating the bias issue. Our approach is based on two core designs: First, we find an explicit and compact separating boundary as the guidance for further feature learning. As the boundary only relies on the normal feature distribution, the bias problem caused by a few known anomalies can be alleviated. Second, a boundary guided semi-push-pull loss is developed to only pull the normal features together while pushing the abnormal features apart from the separating boundary beyond a certain margin region. In this way, our model can form a more explicit and discriminative decision boundary to distinguish known and also unseen anomalies from normal samples more effectively. Code will be available at https://github.com/xcyao00/BGAD.

扫码加入交流群

加入微信交流群

微信交流群二维码

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