Abstract:Train surface anomaly detection is an essential task in vision-based railway safety inspection. Although existing deep learning methods show great potential, their anomaly detection accuracy is affected by the lack of abnormal images. The number of abnormal images on the train surface is far less than that of normal images. One effective way to solve this problem is to expand the abnormal sample. However, most of train surface anomaly image generation methods faces difficulties in producing images with high quality and rich diversity at the same time. In addition, generating small-area anomalies is not ideal. An Anomaly-GAN based on mask pool, abnormal aware loss, and local versus global discriminators is thus proposed in this paper to solve these problems. The mask pool consists of prior-knowledge-based masks and expert-experience-based masks that guide model in generating anomalies with different shapes, rotation angles, spatial locations, and part numbers. The anomaly aware loss focuses on small-area anomalies, thereby promoting the generated anomalies with more detailed textures and richer semantics. The local versus global consistency discriminators combine local and global feature expressions that lead to the generation of more realistic and natural abnormal samples. The experiments show that, compared with other advanced data augmentation algorithms, the images generated by Anomaly-GAN achieve the best FID and LPIPS scores in all anomaly categories. In addition, compared with the case without data augmentaion, the proposed data enhancement method improves the performance of CNN on mAP and mIOU by 25.6% and 24.2%, respectively. Test code is available in https://github.com/AI-Dream-Chaser/Anomaly-GAN .

Make the Rocket Intelligent at IoT Edge: Stepwise GAN for Anomaly Detection of LRE With Multisource Fusion

Unsupervised Multimodal Anomaly Detection With Missing Sources for Liquid Rocket Engine

Fault detection and diagnosis for liquid rocket engines with sample imbalance based on Wasserstein generative adversarial nets and multilayer perceptron

Accurate Anomaly Detection Leveraging Knowledge-enhanced GAT

Knowledge distillation-optimized two-stage anomaly detection for liquid rocket engine with missing multimodal data

A Graph Embedded In Graph Framework With Dual-sequence Input For Efficient Anomaly Detection of Complex Equipment Under Insufficient Samples

Retentive Multimodal Scale-Variable Anomaly Detection Framework with Limited Data Groups for Liquid Rocket Engine

Intelligent Intrusion Detection for Internet of Things Security: A Deep Convolutional Generative Adversarial Network-enabled Approach

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

Anomaly-Based Intrusion on IoT Networks Using AIGAN-a Generative Adversarial Network

IoT-Enabled Few-Shot Image Generation for Power Scene Defect Detection Based on Self-Attention and Global–Local Fusion

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

Enhancing Cybersecurity in IoT Networks: A Deep Learning Approach to Anomaly Detection

DL-ADS: Improved Grey Wolf Optimization Enabled AE-LSTM Technique for Efficient Network Anomaly Detection in Internet of Thing Edge Computing

Adversarially Learned Anomaly Detection

ADRIoT: an Edge-Assisted Anomaly Detection Framework Against IoT-Based Network Attacks

Deep Learning-Based Cyber–Physical Feature Fusion for Anomaly Detection in Industrial Control Systems

Mitigate noisy data for smart IoT via GAN based machine unlearning

Anomaly detection by using a combination of generative adversarial networks and convolutional autoencoders

An Asymmetrical Graph Siamese Network for One-classAnomaly Detection of Engine Equipment with Multi-source Fusion

GAAD: GAN-enabled Autoencoder for Real-Time Sensor Anomaly Detection and Recovery in Autonomous Driving