English

Patch-aware Vector Quantized Codebook Learning for Unsupervised Visual Defect Detection

Computer Vision and Pattern Recognition 2025-01-17 v1 Artificial Intelligence Machine Learning

Abstract

Unsupervised visual defect detection is critical in industrial applications, requiring a representation space that captures normal data features while detecting deviations. Achieving a balance between expressiveness and compactness is challenging; an overly expressive space risks inefficiency and mode collapse, impairing detection accuracy. We propose a novel approach using an enhanced VQ-VAE framework optimized for unsupervised defect detection. Our model introduces a patch-aware dynamic code assignment scheme, enabling context-sensitive code allocation to optimize spatial representation. This strategy enhances normal-defect distinction and improves detection accuracy during inference. Experiments on MVTecAD, BTAD, and MTSD datasets show our method achieves state-of-the-art performance.

Keywords

Cite

@article{arxiv.2501.09187,
  title  = {Patch-aware Vector Quantized Codebook Learning for Unsupervised Visual Defect Detection},
  author = {Qisen Cheng and Shuhui Qu and Janghwan Lee},
  journal= {arXiv preprint arXiv:2501.09187},
  year   = {2025}
}

Comments

7 pages, Accepted to 36th IEEE ICTAI 2024