ADFormer: Generalizable Few-Shot Anomaly Detection With Dual CNN-Transformer Architecture
This article investigates the potential of dual CNN-Transformer architectures for generalizable few-shot anomaly detection (GFSAD), a practical yet understudied form of anomaly detection (AD). In GFSAD, a common model must be learned and shared across several categories, while simultaneously ensuring that the model is adaptable to new categories with a restricted number of normal images. While CNN-Transformer architectures obtain high success in many vision tasks, the potential of CNN-Transformer architectures in GFSAD is still to be discovered. In this article, we introduce ADFormer, a dual CNN-Transformer architecture that combines the strengths of CNNs and Transformers, to learn discriminative features that have both local and global receptive fields. We also incorporate a self-supervised bipartite matching approach in ADFormer that reconstructs query images from support images, followed by detecting anomalies based on the high loss in reconstruction. Additionally, we present a consistency-enhanced loss to enhance the spatial and semantic consistency of features, thereby reducing the dependence on a large AD dataset for training. Experimental results show that ADFormer with consistency-enhanced loss significantly improves GFSAD performance. Compared to other AD methods, ADFormer outperforms considerably the MVTec AD, MPDD, and VisA datasets.
Authors
- Zhaopeng Gu
- Guibo Zhu (ORCID: https://orcid.org/0000-0001-8293-3952)
- Bingke Zhu (ORCID: https://orcid.org/0000-0001-6429-1773)
- Jinqiao Wang (ORCID: https://orcid.org/0000-0002-9118-2780)
- Yingying Chen (ORCID: https://orcid.org/0000-0002-5049-8092)
- Ming Tang (ORCID: https://orcid.org/0000-0003-4976-3095)
Institutions
- Chinese Academy of Sciences (CN)
- Institute of Automation (CN)
Publication Details
- Journal
- IEEE Transactions on Instrumentation and Measurement
- Published
- 2024-12-25
- DOI
- https://doi.org/10.1109/tim.2024.3522624
- Citations
- 6
- Primary Topic
- Anomaly Detection Techniques and Applications
- Type
- article
- Field-Weighted Citation Impact
- 2.08
Funders
- National Natural Science Foundation of China
- National Key Research and Development Program of China