Enhanced industrial anomaly detection via CutMask data augmentation: a self-supervised approach
摘要
Deep learning techniques have revolutionized industrial anomaly detection. However, their reliance on substantial labeled data poses challenges in scenarios where anomalous samples are scarce. This paper presents a practical self-supervised anomaly detection framework, CutMask-based Anomaly Detection (CMAD), designed for industrial inspection using normal target-domain training images together with independently sourced defect-shape priors. The main methodological contribution is CutMask, a defect-shape-aware data augmentation strategy that, unlike generic augmentations such as CutMix, MixUp, and regular CutPaste-style rectangular patch pasting, explicitly leverages obtainable prior knowledge of defect morphology to generate more realistic and morphology-consistent simulated defect samples. Furthermore, CMAD uses a lightweight ResNet-18 as the backbone for anomaly-sensitive feature extraction, integrates a Self-Supervised Predictive Convolutional Attentive Block (SSPCAB) module to refine feature modeling, and optimizes the network with the proposed Cyclical Auxiliary Focal Loss (CAFL) objective to improve discrimination between normal and CutMask-generated pseudo-anomalous samples. Experimental results on the MVTec AD dataset, the VisA dataset, and a practical presswork dataset show that CMAD achieves strong image-level anomaly detection performance under limited anomalous data conditions, including image-level area under the receiver operating characteristic curve (AUROC) scores of 97.93% on MVTec AD and 93.88% on VisA. Overall, the proposed framework provides a practical and defect-shape-aware extension of augmentation-driven industrial anomaly detection for scenarios in which real anomalous samples are scarce.