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

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.

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

    Open on Semantic Scholar

  • 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