Active anomaly detection based on deep one-class classification

被引:15
作者
Kim, Minkyung [1 ]
Kim, Junsik [2 ]
Yu, Jongmin [3 ]
Choi, Jun Kyun [1 ]
机构
[1] Korea Adv Inst Sci & Technol KAIST, Sch Elect Engn, Daejeon, South Korea
[2] Harvard Univ, Sch Engn & Appl Sci, Cambridge, MA USA
[3] Kings Coll London, Dept Engn, London, England
基金
新加坡国家研究基金会;
关键词
Deep anomaly detection; One -class classification; Deep SVDD; Active learning; Noise -contrastive estimation; SUPPORT;
D O I
10.1016/j.patrec.2022.12.009
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Active learning has been utilized as an efficient tool in building anomaly detection models by leveraging expert feedback. In an active learning framework, a model queries samples to be labeled by experts and re-trains the model with the labeled data samples. It unburdens in obtaining annotated datasets while improving anomaly detection performance. However, most of the existing studies focus on helping experts identify as many abnormal data samples as possible, which is a sub-optimal approach for one-class classification-based deep anomaly detection. In this paper, we tackle two essential problems of active learning for Deep SVDD: query strategy and semi-supervised learning method. First, rather than solely identifying anomalies, our query strategy selects uncertain samples according to an adaptive boundary. Second, we apply noise contrastive estimation in training a one-class classification model to incorporate both labeled normal and abnormal data effectively. We analyze that the proposed query strategy and semi-supervised loss individually improve an active learning process of anomaly detection and further improve when combined together on seven anomaly detection datasets.(c) 2022 Published by Elsevier B.V.
引用
收藏
页码:18 / 24
页数:7
相关论文
共 26 条
[1]  
[Anonymous], 2018, Nsl-kdd
[2]   Active Learning for One-Class Classification [J].
Barnabe-Lortie, Vincent ;
Bellinger, Colin ;
Japkowicz, Nathalie .
2015 IEEE 14TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS (ICMLA), 2015, :390-395
[3]  
Chalapathy R, 2019, Arxiv, DOI [arXiv:1802.06360, DOI 10.48550/ARXIV.1802.06360]
[4]  
Das S, 2017, Arxiv, DOI arXiv:1708.09441
[5]  
Das S, 2016, IEEE DATA MINING, P853, DOI [10.1109/ICDM.2016.0102, 10.1109/ICDM.2016.164]
[6]  
Ghasemi A., 2011, 2011 IEEE International Conference on Data Mining Workshops, P244, DOI 10.1109/ICDMW.2011.20
[7]   Toward Supervised Anomaly Detection [J].
Goernitz, Nico ;
Kloft, Marius ;
Rieck, Konrad ;
Brefeld, Ulf .
JOURNAL OF ARTIFICIAL INTELLIGENCE RESEARCH, 2013, 46 :235-262
[8]  
Gutmann M., 2010, P INT C ARTIFICIAL I, P297
[9]  
King DB, 2015, ACS SYM SER, V1214, P1, DOI 10.1021/bk-2015-1214.ch001
[10]  
Lamba H, 2019, SIAM INT C DATA MINI