Semi-supervised Rail Defect Detection from Imbalanced Image Data

Abstract

Rail defect detection by video cameras has recently gained much attention in both academia and industry. Rail image data has two properties. It is highly imbalanced towards the non-defective class and it has a large number of unlabeled data samples available for semisupervised learning techniques. In this paper we investigate if positive defective candidates selected from the unlabeled data can help improve the balance between the two classes and gain performance on detecting a specific type of defects called Squats. We compare data sampling techniques as well and conclude that the semi-supervised techniques are a reasonable alternative for improving performance on applications such as rail track Squat detection from image data.

A Novel Approach for Continuous Health Status Monitoring and Automatic Detection of Infection Incidences in People With Type 1 Diabetes Using Machine Learning Algorithms (Part 2): A Personalized Digital Infectious Disease Detection Mechanism

“…Class imbalance was mitigated by oversampling of the nontarget data sets through random sampling [ 44 ]. Performance was measured using the area under the receiver operating characteristic (ROC) curve (AUC), specificity, and F1-score [ 45 - 48 ]. The AUC, specificity, and F1-score were reported as the average (SD) of twenty times five-fold stratified cross-validation rounds.…”

“…AUC is the result of integration (summation) of the ROC curve over a range of possible classification thresholds [ 49 ]. It is regarded as robust (insensitive) when it comes to the presence of data imbalance; however, it is impractical for real-world implementation because it is independent of a single threshold [ 48 ]. Specificity measures the ratio of correctly classified negative samples from the total number of available negative samples [ 50 ].…”

Application of Deep Learning Convolutional Neural Networks for Internal Tablet Defect Detection: High Accuracy, Throughput, and Adaptability

“…9,18 In addition, automatic inspection and defect detection based on deep learning CNNs have been widely adopted in many fields. 19 CNN methods have been shown to achieve high throughput quality control during the manufacture of metallic rails 20 and steel surfaces, 21 demonstrating their widespread adoption.…”

Automatic detection and classification of manufacturing defects in metal boxes using deep neural networks

“…It is used for high throughput quality control in production systems such as the detection of flaws on manufactured surfaces, e.g. metallic rails [ 1 ] or steel surfaces [ 2 ]. The idea is to design autonomous devices that automatically detect and examine specific visual patterns from images and videos in order to overcome the limitations and improve the performance of the traditional inspection systems that depend heavily on human inspectors.…”

“…Since the first use of classical CNNs in inspection task [ 8 , 14 ], they have shown a good classification accuracy for many applications such as the detection of defects on photometric stereo images of rail surfaces [ 1 ]. Another application of CNNs is the classification of steel images and used Pyramid of Histograms of Orientation Gradients (HOG) as feature extractor and Max-Pooling Convolutional Neural Networks.…”