Abstract:Anomaly detection is a crucial means to detect unbalanced and multiclass abnormal data in industrial products. The existing anomaly detection model is usually suitable for products with a simple or regular background texture. However, there is a lack of a universal surface defect anomaly detection framework for railway products when facing the challenges: the complex background texture, huge intraclass variation, uneven distribution of interclass, and weak texture-featured defects. To fill this gap, an innovative generative adversarial network based on adaptive pyramid graph (APG) and variation residuals (APGVR-GAN) is proposed, aiming to improve the robustness of anomaly detection in railway products and other complex industrial supplies. First, the APG module is embedded in the encoder–decoder–encoder pattern, capturing the correlation description between neighbor regions, which is utilized to enhance the detection of abnormal defects with weak texture. Next, the variation residual module is employed to enhance the expression of various normal samples in the latent space and improve the identification ability for abnormal samples. Then, the dual-probability prototype loss is proposed to make different normal samples have more concentrated expression and more similar probability distribution centers in latent space. Finally, an adaptive focal-gate loss and a regularized log-likelihood loss are designed to overcome the imbalance problem in training samples with different background information. The effectiveness of the model is verified on three new railway datasets and three other industrial public datasets. The results show that in anomaly detection, APGVR-GAN on the three railway datasets, Northeastern University (NEU)-the rail surface defects dataset (RSDD)-1 dataset, NEU-RSDD-1 dataset, and RSDD area under curve (AUC) [that is the receiver operating characteristic curve (ROC)], can reach 0.89, 0.81, and 0.87, and its detection accuracy (ACC) reached 0.97, 0.96, and 0.92, which is much higher than other comparative anomaly detection algorithms. In the segmentation experiments on the RSDD dataset, compared with other known rail defect detection methods, the overall evaluation metric F-Score is 83.70 which is also better than other methods.

Anomaly-GAN: A data augmentation method for train surface anomaly detection

Data anomaly detection for structural health monitoring using a combination network of GANomaly and CNN

AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model

An Adaptive Pyramid Graph and Variation Residual-Based Anomaly Detection Network for Rail Surface Defects

Dual-Path GAN: A Method for Enhancing Small-scale Defect Detection on Metal Images

ADGAN: A Scalable GAN-based Architecture for Image Anomaly Detection

Anomaly Detection in Traffic Surveillance Videos with GAN-based Future Frame Prediction

GAN-based Data Augmentation Strategy for Sensor Anomaly Detection in Industrial Robots

Anomaly Detection Using GANs for Visual Inspection in Noisy Training Data

LRGAN: Visual anomaly detection using GAN with locality-preferred recoding*

Meta-learning with GANs for anomaly detection, with deployment in high-speed rail inspection system

Detecting Anomalies using Generative Adversarial Networks on Images

GAN-Based Anomaly Detection for Multivariate Time Series Using Polluted Training Set

Rail surface defect data enhancement method based on improved ACGAN

Video Anomaly Detection using GAN

Improving anomaly detection with a self-supervised task based on generative adversarial network

FEGAN: A Feature Extraction Based Approach for GAN Anomaly Detection and Localization

GANAD: A GAN-based Method for Network Anomaly Detection

DOPING: Generative Data Augmentation for Unsupervised Anomaly Detection with GAN

Metro Anomaly Detection Based on Light Strip Inductive Key Frame Extraction and MAGAN Network

CS-GANomaly: A Supervised Anomaly Detection Approach with Ancillary Classifier GANs for Chromosome Images