Anomaly classification based on self-supervised learning and its application

Abstract

BackgroundAnomaly classification remains a challenging task in computer vision applications across diverse practical fields such as industrial detection and security check. The purpose of this study is to develop a new anomaly classification method based on self-supervised learning (Self-ACM), which is expected to enhance anomaly classification accuracy significantly. MethodsThe feature maps of images were abstracted using VGG16 model pre-trained on ImageNet, which were then fed into the innovative model to capture the normality distribution within the dataset domain. Then, a self-supervised adversarial anomaly classification learning framework was proposed to facilitate the acquisition of a higher-level semantic representations for improved anomaly detection. Thirdly, we collected and constructed a novel terahertz (THZ) dataset, which serves as a pioneering resource for benchmarking anomaly classification tasks in the field. ResultsThrough a series of rigorous experiments, our findings unequivocally demonstrate the following key insights: Firstly, harnessing feature maps as input data yielded a significant enhancement in anomaly detection performance, underscoring the effectiveness of this approach. Secondly, the integration of self-supervision enriched the dataset with invaluable information, empowering both the discriminator and generator to acquire superior feature representations. The culmination of these advancements is our novel method achieving unparalleled state-of-the-art performance across multiple benchmark datasets. This breakthrough underscores the transformative impact of our approach on anomaly detection methodologies, solidifying its position as a pioneering solution in the field. ConclusionThe Self-ACM strategy not only advances anomaly detection methodologies but also offers a remarkable contribution to dataset creation, setting a new standard for anomaly classification research.