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 .

A Few-shot Learning Approach for Anomaly Detection in Cooperative Tracking Control System

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

Learning-Based Cooperative False Data Injection Attack and Its Mitigation Techniques in Consensus-Based Distributed Estimation

An Autoencoder-Driven FDIA Detection Method for Virtual Coupling Systems of Heavy-Haul Trains

A Physics-Based Context-Aware Approach for Anomaly Detection in Teleoperated Driving Operations Under False Data Injection Attacks

A GAN-based Deep Learning Model for the Object Detection of Autonomous High-Speed Trains

A Gated Few-shot Learning Model for Anomaly Detection

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

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

Data augmentation using generative models for track intrusion detection

Distributed Tracking Control of Nonlinear Multi-agent Systems Against False Data Injection Attacks

Modeling and detection of false data injection attacks in cyber-physical distribution system with load aggregator interaction

Model-agnostic generation-enhanced technology for few-shot intrusion detection

Secure and Safe Control of Connected and Automated Vehicles Against False Data Injection Attacks

Dual Auto-Encoder GAN-Based Anomaly Detection for Industrial Control System

Siamese Neural Network Based Few-Shot Learning for Anomaly Detection in Industrial Cyber-Physical Systems

Enhanced Few-Shot Learning for Intrusion Detection in Railway Video Surveillance

Deep Learning to Improve False Data Injection Attack Detection in Cooperative Intelligent Transportation Systems

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

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

Defense of Massive False Data Injection Attack Via Sparse Attack Points Considering Uncertain Topological Changes