Semi-supervised rail defect detection from imbalanced image data
Siamak Hajizadeh (TU Delft - Railway Engineering)
Alfredo Nunez Vicencio (TU Delft - Railway Engineering)
David Tax (TU Delft - Pattern Recognition and Bioinformatics)
More Info
expand_more
Other than for strictly personal use, it is not permitted to download, forward or distribute the text or part of it, without the consent of the author(s) and/or copyright holder(s), unless the work is under an open content license such as Creative Commons.
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 specic 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.