Recently, reconstruction-based deep models have gained widespread usage in unsupervised anomaly detection. However, they may overlook some anomalies owing to the over-generalization of neural networks. Several studies have incorporated memory networks to mitigate this problem. Nonetheless, some of them lack an explicit memory updating process, while others rely on data-driven updating methods that are sensitive to initial values and unsuitable for end-to-end training. Additionally, the traditional criterion for detection computed in the high-dimensional input space may collapse as the spike in the deviation score is averaged across numerous dimensions. To address these challenges, we propose MCAKE, a Memory-augmented Contrastive Autoencoder with KNN-Based Extraction. It is designed to highlight the deviation score for anomalies by reconstructing input using fixed normal prototypes recorded in the memory. We explicitly encourage the memory to be autonomously learned and effectively allocated through contrastive learning with multiple positive and multiple negative samples. Furthermore, we introduce a bivariate detection criterion that calculates anomaly scores considering both input and latent space to tackle the collapse. Extensive experiments on 50 datasets across various categories demonstrate the superiority of our approach, with a 2% relative improvement over the previous state-of-the-art models.
MCAKE: Memory-Augmented Autoencoder with Contrastive Learning for Unsupervised Anomaly Detection
Chengsen Wang,Qi Qi,Jinming Wu,Haifeng Sun,Zirui Zhuang,Yuhan Jing,Lianyuan Li,Jingyu Wang
Published 2025 in ACM Transactions on Knowledge Discovery from Data
ABSTRACT
PUBLICATION RECORD
- Publication year
2025
- Venue
ACM Transactions on Knowledge Discovery from Data
- Publication date
2025-08-12
- Fields of study
Computer Science
- Identifiers
- External record
- Source metadata
Semantic Scholar
CITATION MAP
EXTRACTION MAP
CLAIMS
- No claims are published for this paper.
CONCEPTS
- No concepts are published for this paper.
REFERENCES
Showing 1-40 of 40 references · Page 1 of 1