arrow
Return

Improved Autoencoder Model With Memory Module for Anomaly Detection

delete2024-04-15
delete3
PRE
AI
W
Wei Huang
刘振 (Zhen Liu)
J
Jin, Xiaohang
J
Jinshan Xu
X
Xin‐Wei Yao *
DOI:10.1109/JSEN.2024.3370965delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
As a commonly used model for anomaly detection, the autoencoder model for anomaly detection does not train the objective for extracted features, which is a downside of autoencoder model. In addition, it is well known that the autoencoder model has over-prominent reconstruction ability for anomalous data, leading to high false-negative rate. On the other hand, the deep support vector data description (SVDD) model first extracts features through deep neural network, and then map the extracted features into a hypersphere. However, the deep SVDD model has disadvantages such as feature information loss and feature collapse during training process, leading to a decrease in anomaly detection accuracy. To alleviate such drawbacks mentioned above, in this article, we propose a novel model, called improved autoencoder with memory module (IAEMM). Specifically, this model jointly learns deep SVDD model and autoencoder model to minimize the overall loss of deep SVDD error and reconstruction error, and add a memory module after encoder to amplify the difference of reconstruction error between normal and abnormal data. The proposed model can well identify abnormal hidden features and mitigate the problem of feature collapse. Experimental results on several datasets confirm the effectiveness and stability of our proposed method.
Keywords:
Anomaly detection
autoencoder
deep support vector data description (SVDD)
memory

Journal

IEEE Sensors Journal cover
IEEE Sensors Journal
IF:
4.5
Papers:
2.1W
Citations:
7.3W

Organization

Z
zhejiang university of technology
Scholars:
3.2W
Papers: 2.0W
Citations: 22