VAE-based Deep SVDD for anomaly detection

被引:73
作者
Zhou, Yu [1 ]
Liang, Xiaomin [1 ]
Zhang, Wei [1 ]
Zhang, Linrang [1 ]
Song, Xing [1 ]
机构
[1] Xidian Univ, Natl Key Lab Radar Signal Proc, Xian 710071, Peoples R China
基金
中国国家自然科学基金;
关键词
Anomaly detection; Variational autoencoder; Deep support vector data description; SUPPORT;
D O I
10.1016/j.neucom.2021.04.089
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Anomaly detection is an essential task for different fields in the real world. The imbalanced data and lack of labels make the task challenging. Deep learning models based on autoencoder (AE) have been applied to address the above difficulties successfully. However, in these AE-based deep methods, the AE-based model's optimization and the anomaly detector design are separated. Therefore, the latent representations in AE are less relevant for the anomaly detection task, which reduces the accuracy of anomaly detec-tion. A deep support vector data description based on variational autoencoder (Deep SVDD-VAE) is proposed in this paper to solve this problem. In the proposed model, VAE is used to reconstruct the input instances, while a spherical discriminative boundary is learned with the latent representations simulta-neously based on SVDD. Unlike existing AE-based methods, we seek the model parameters via the joint optimization of VAE and SVDD, which ensures the separability of the latent representations. Experimental results on MNIST, CIFAR-10, and GTSRB datasets show the effectiveness of Deep SVDD-VAE. (C) 2021 Elsevier B.V. All rights reserved.
引用
收藏
页码:131 / 140
页数:10
相关论文
共 35 条
[1]   Using an autoencoder in the design of an anomaly detector for smart manufacturing [J].
Alfeo, Antonio L. ;
Cimino, Mario G. C. A. ;
Manco, Giuseppe ;
Ritacco, Ettore ;
Vaglini, Gigliola .
PATTERN RECOGNITION LETTERS, 2020, 136 :272-278
[2]  
[Anonymous], 2014, ICLR
[3]  
[Anonymous], 2015, Technical Report
[4]  
Borghesi A, 2019, AAAI CONF ARTIF INTE, P9428
[5]  
Chalapathy R., ARXIV PREPRINT ARXIV
[6]   Anomaly detection in surveillance video based on bidirectional prediction [J].
Chen, Dongyue ;
Wang, Pengtao ;
Yue, Lingyi ;
Zhang, Yuxin ;
Jia, Tong .
IMAGE AND VISION COMPUTING, 2020, 98 (98)
[7]  
Choras M., 2019, 13 INT C COMPUTATION
[8]   Anomaly detection of defects on concrete structures with the convolutional autoencoder [J].
Chow, J. K. ;
Su, Z. ;
Wu, J. ;
Tan, P. S. ;
Mao, X. ;
Wang, Y. H. .
ADVANCED ENGINEERING INFORMATICS, 2020, 45 (45)
[9]  
Cortes C, MNIST Handwritten Digit Database
[10]  
Ghrib Z, 2020, 2020 INT JOINT C NEU, P1, DOI 10.1109/IJCNN48605.2020.9207013