Unifying Reconstruction and Density Estimation via Invertible Contraction Mapping in One-Class Classification
Xiaolei Wang, Tianhong Dai, Huihui Bai, Yao Zhao, Jimin Xiao
Abstract
Due to the difficulty in collecting all unexpected abnormal patterns, One-Class Classification (OCC) has become the most popular approach to anomaly detection (AD). Reconstruction-based AD method relies on the discrepancy between inputs and reconstructed results to identify unobserved anomalies. However, recent methods trained only on normal samples may generalize to certain abnormal inputs, leading to well-reconstructed anomalies and degraded performance. To address this, we constrain reconstructions to remain on the normal manifold using a novel AD framework based on contraction mapping. This mapping guarantees that any input converges to a fixed point through iterations of this mapping. Based on this property, training the contraction mapping using only normal data ensures that its fixed point lies within the normal manifold. As a result, abnormal inputs are iteratively transformed toward the normal manifold, increasing the reconstruction error. In addition, the inherent invertibility of contraction mapping enables flow-based density estimation, where a prior distribution learned from the previous reconstruction is used to estimate the input likelihood for anomaly detection, further improving the performance. Using both mechanisms, we propose a bidirectional structure with forward reconstruction and backward density estimation. Extensive experiments on tabular data, natural image, and industrial image data demonstrate the effectiveness of our method. Code is available at URD.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext de9f5ea5-73f3-463a-a5d0-73947fa66b2eCited by top-tier papers2
- TINA: Text-Free Inversion Attack for Unlearned Text-to-Image Diffusion ModelsQianlong Xiang, Miao Zhang, Haoyu Zhang, Kun Wang et al.CVPR 2026 · 2 citations
- Hunting Normality from Query Sample via Residual Learning for Generalist Anomaly DetectionXiaolei Wang, Yuexin Wang, Tianhong Dai, Huihui Bai et al.CVPR 2026
Builds on31
- Towards Total Recall in Industrial Anomaly DetectionKarsten Roth, Latha Pemula, Joaquin Zepeda, Bernhard Schölkopf et al.CVPR 2022 · 1,301 citations
- CSI: Novelty Detection via Contrastive Learning on Distributionally Shifted InstancesJihoon Tack, Sangwoo Mo, Jongheon Jeong, Jinwoo ShinNeurIPS 2020 · 755 citations
- Anomaly Detection via Reverse Distillation from One-Class EmbeddingHanqiu Deng, Xingyu LiCVPR 2022 · 701 citations
- A Unified Model for Multi-class Anomaly DetectionZhiyuan You, Lei Cui, Yujun Shen, Kai Yang et al.NeurIPS 2022 · 585 citations
- Classification-Based Anomaly Detection for General DataLiron Bergman, Yedid HoshenICLR 2020 · 412 citations
Related papers
- DecAD: Decoupling Anomalies in Latent Space for Multi-Class Unsupervised Anomaly DetectionXiaolei Wang, Xiaoyang Wang, Huihui Bai, Eng Gee Lim et al.ICCV 2025 · 3 citations
- Diversity-Measurable Anomaly DetectionWenrui Liu, Hong Chang, Bingpeng Ma, Shiguang Shan et al.CVPR 2023
- Anomaly Detection using Score-based Perturbation ResilienceWoosang Shin, Jonghyeon Lee, Taehan Lee, Sangmoon Lee et al.ICCV 2023 · 13 citations
- ReContrast: Domain-Specific Anomaly Detection via Contrastive ReconstructionJia Guo, Shuai Lu, Lize Jia, Weihang Zhang et al.NeurIPS 2023 · 124 citations
- DRL: Decomposed Representation Learning for Tabular Anomaly DetectionHangting Ye, He Zhao, Wei Fan, Mingyuan Zhou et al.ICLR 2025
