Detection of rail surface defects based on CNN image recognition and classification

With the rapid development of rail transit, the workload of track maintenance has increased, making the intelligent identification of rail surface defects crucial for improving detection efficiency. To address issues such as low defect detection accuracy, the loss of feature information due to single-path architecture backbones, and insufficient information interaction in existing rail defect detection methods, we propose a rail surface defect detection method based on dual-path feature fusion (DPF). This method initially employs a dual-path structure to separately extract low-level and high-level features. It then utilizes a combination of attention mechanisms and feature fusion techniques to integrate these features. By doing so, it preserves richer information and enhances detection accuracy and robustness. The experimental results demonstrate that the comprehensive performance of the proposed model is superior to mainstream algorithms.

Rail surface defect detection plays a critical role in the maintenance of the rail transportation system. Video analysis technology is a promising method to detect defects due to its low cost and effectiveness. Several attempts with hand‐craft features have been made to obtain the detection results by using traditional machine vision algorithms. However, these methods suffer from imprecise results due to challenging conditions, such as deteriorated and changeable lighting environment and various types of complex rail surface defects. Recently, classification methods with complex deep convolutional networks have become popular. Despite their high accuracy, these methods cannot meet the requirements of defects localisation and real‐time processing in practice. To solve these problems, this study proposes a novel object detection algorithm to detect rail defects. The net architecture of the proposed algorithm includes a backbone network using MobileNet and several novel detection layers with multi‐scale feature maps inspired by you only look once (YOLO) and feature pyramid networks. Two different architectures of MobileNet are used to estimate the performance of defects detection. The experimental results demonstrate the great potential of the proposed algorithm with fast inference speed and high accuracy in the industry.

The speed and load capacity of trains are improving at an accelerating rate, which raises the standards for railway services' security criteria. The track's surface may progressively reveal varying degrees of defects because of the impacts of moisture, temperature, load, and other factors; if the flaws never addressed in a prompt period, the degree of defects would expand, significantly raising the probability of train service. The development of automatic RSDD (Rail Surface Defect Detection) has substantial practical and scientific implications. In this study, a multi-crack detection technique depends on Variational Autoencoder Diffusion Model (VAEDM), The introduction of VAEDM allows for the non-destructive detection of fastener and RSD. In order to produce hidden elements from images that have various cracks, VAEDM combines the multi-layer and a rail surface image encoder was introduced. Zero Shot- Divergence Asynchronous Reinforcement Learning (ZS-DARL), Policy-Gradient is used to establish the relationship between defects and non-defects. DARL transformer encoder is introduced for taking long-range dependences for FE (Feature Extraction) from detected objects images with various patterns and dimensions. The simulation outcomes on the open Railway Track Fault Detection (RTFD) and Rail Surface Defect Datasets (RSDDs) with rail surface defects are collected from rail tracks surface defect detection. Results are measured using the metrics like recall, precision, F-measure, and accuracy.

The rapid and accurate identification of rail surface defects is critical to the maintenance and operational safety of the rail. For the problems of large-scale differences in rail surface defects and many small-scale defects, this paper proposes a rail surface defect detection algorithm, RSDNet (Rail Surface Defect Detection Net), with YOLOv8n as the baseline model. Firstly, the CDConv (Cascade Dilated Convolution) module is designed to realize multi-scale convolution by cascading the cavity convolution with different cavity rates. The CDConv is embedded into the backbone network to gather earlier defect local characteristics and contextual data. Secondly, the feature fusion method of Head is optimized based on BiFPN (Bi-directional Feature Pyramids Network) to fuse more layers of feature information and improve the utilization of original information. Finally, the EMA (Efficient Multi-Scale Attention) attention module is introduced to enhance the network's attention to defect information. The experiments are conducted on the RSDDs dataset, and the experimental results show that the RSDNet algorithm achieves a mAP of 95.4% for rail surface defect detection, which is 4.6% higher than the original YOLOv8n. This study provides an effective technical means for rail surface defect detection that has certain engineering applications.

To address the challenges of accurately extracting features from elongated scratches, irregular defects, and small-scale surface flaws on high-precision linear guide rails, this paper proposes a novel instance segmentation algorithm tailored for guide rail surface defect detection. The algorithm integrates the YOLOv8 instance segmentation framework with deformable convolutional networks and multi-scale feature fusion to enhance defect feature extraction and segmentation performance. A dedicated guide rail surface Defect (GSD) segmentation dataset is constructed to support model training and evaluation. In the backbone, the DCNv3 module is incorporated to strengthen the extraction of elongated and irregular defect features while simultaneously reducing model parameters. In the feature fusion network, a multi-scale feature fusion module and a triple-feature encoding module are introduced to jointly capture global contextual information and preserve fine-grained local defect details. Furthermore, a Channel and Position Attention Module (CPAM) is employed to integrate global and local features, improving the model’s sensitivity to channel and positional cues of small-target defects and thereby enhancing segmentation accuracy. Experimental results show that, compared with the original YOLOv8n-Seg, the proposed method achieves improvements of 3.9% and 3.8% in Box and Mask mAP50, while maintaining a real-time inference speed of 148 FPS. Additional evaluations on the public MSD dataset further demonstrate the model’s strong versatility and robustness.

The application of computer vision technology in defect detection of industrial products is a popular research direction in recent years. This article presents the pyramid feature convolutional neural network (CNN) for defect detection of rail surfaces. First, multi-scale feature maps are extracted based on the characteristics of defects and backgrounds by the pyramid feature extraction module (PFEM). Then the feature maps are input to a lightweight network consisting of a small number of parameters. The network is trained with only 40% data of the dataset using binary cross-entropy loss function and the intersection of union (IOU) loss function. In the experiment, the performance of the proposed method is evaluated using the rail surface defect dataset (RSDD) dataset by comparing it with other methods. The experimental results show that the segmentation performance and real-time performance of the proposed method are better than those of other methods.

The detection of rail surface defects is an important tool to ensure the safe operation of rail transit. Due to the complex diversity of track surface defect features and the small size of the defect area, it is difficult to obtain satisfying detection results by traditional machine vision methods. The existing deep learning-based methods have the problems of large model sizes, excessive parameters, low accuracy and slow speed. Therefore, this paper proposes a new method based on an improved YOLOv4 (You Only Look Once, YOLO) for railway surface defect detection. In this method, MobileNetv3 is used as the backbone network of YOLOv4 to extract image features, and at the same time, deep separable convolution is applied on the PANet layer in YOLOv4, which realizes the lightweight network and real-time detection of the railway surface. The test results show that, compared with YOLOv4, the study can reduce the amount of the parameters by 78.04%, speed up the detection by 10.36 frames per second and decrease the model volume by 78%. Compared with other methods, the proposed method can achieve a higher detection accuracy, making it suitable for the fast and accurate detection of railway surface defects.

This paper proposes a multi-source data fusion algorithm for rail surface defect detection in both camera-based rail inspection images and ultrasound B-scan images. First, we design a rail surface segmentation algorithm based on image bilateral filtering, Sobel edge detection, and rail surface edge detection to extract the rail surface area. Second, we build a camera and ultrasound data fusion (CUFuse) model for rail surface defect detection, including two main networks: multi-source data feature extraction and multi-scale feature fusion networks. The multi-source data feature extraction network consists of two BoTNet 50 networks as feature extraction networks to extract five stages of features in camera-based images and ultrasound B-scan images. The multi-scale feature fusion network consists of five feature fusion modules to fuse the feature information output by the multi-source data feature extraction network. Finally, we use the CUFuse model to detect the rail surface defect dataset, and output five rail surface state types, including Light, Moderate, Severe, Normal, and Joint. The results show that the accuracy of the CUFuse model is 96.97%, which can accomplish the task of rail surface defect detection on railway sites.

As track components deteriorate, their interaction dynamic response will alter accelerations on the train. Measuring acceleration on train components is a method to conduct condition monitoring of track defects. The measured signal can be used for the detection of rail surface defects which implement impacts on measured accelerations. In this paper, the axle-box acceleration (ABA) is measured in a subway as a case study. Fourier transform (FFT), empirical mode decomposition (EMD) and ensemble EMD (EEMD) methods are used to study accelerations relative to the track. Wheel frequencies are calculated using finite element method (FEM) to determine frequency couplings. Velocity-dependent/independent components and source of excitation of the measured signal are distinguished. Results indicated that the FFT approach can be applied for both velocity-dependent and velocity-independent vibration components for frequencies up to 680 Hz. Also, the EEMD method can be used to distinguish the impact component of the measured signal.

Internal defects detection and classification in hollow cylindrical surfaces using single shot detection and MobileNet

Deep learning has achieved great success in the field of computer vision, and the precision in image classification and image detection has surpassed humans. Therefore, this paper combines deep learning and medical image segmentation, focusing on how to improve the accuracy and speed of segmentation algorithm of medical exercise rehabilitation image. Aiming at the shortcomings of traditional medical image recognition methods, a medical exercise rehabilitation image segmentation algorithm based on hierarchical features of convolutional neural networks is proposed, this paper calls it as hierarchical features of convolutional neural networks (HFCNN). The algorithm firstly samples the convolution output of multiple layers in the convolutional neural network to a unified scale and combines them to construct a hierarchical feature. This hierarchical feature combines the structural information of objects contained in the shallow layer of the network with the semantic information of objects contained in the deep layers of the network, so it has a strong ability to express. Secondly, the image can be segmented into multiple super pixels by the super pixel segmentation algorithm. The classifier is trained using the hierarchical features of the super pixel, and then the classification result is mapped back to the pixel. Finally, a fully connected conditional random field algorithm including one-potential potential energy and paired potential energy is constructed. The corresponding energy function is used to smooth the classification result of the pixel, and the regional consistency and continuity of the pixel mark are improved. Compared with many classical convolutional neural network algorithms, this algorithm not only accelerates the network convergence speed, shortens the training time, but also significantly improves the accuracy of segmentation algorithm of medical exercise rehabilitation image, showing good practical value.

Defect detection method for rail surface based on line-structured light

Surface defect detection aims to classify and locate a certain defect that exists in the target surface area. It is an important part of industrial quality inspection. Most of the research on surface defect detection are currently based on convolutional neural networks (CNNs), which are more concerned with local information and lack global perception. Thus, CNNs are unable to effectively extract the defect features. In this paper, a defect detection method based on the Swin transformer is proposed. The structure of the Swin transformer has been fine-tuned so that it has five scales of output, making it more suitable for defect detection tasks with large variations in target size. A bi-directional feature pyramid network is used as the feature fusion part to efficiently fuse to the extracted features. The focal loss is used as a loss function to weight the hard- and easy-to-distinguish samples, potentially making the model fit the surface defect data better. To reduce the number of parameters in the model, a shared detection head was chosen for result prediction. Experiments were conducted on the flange surface defect dataset and the steel surface defect dataset, respectively. Compared with the classical CNNs target detection algorithm, our method improves the mean average precision (mAP) by about 15.4%, while the model volume and detection speed are essentially the same as those of the CNNs-based method. The experimental results show that our proposed method is more competitive compared with CNNs-based methods and has some generality for different types of defects.

As an important component of the railway system, the surface damage that occurs on the rails due to daily operations can pose significant safety hazards. This paper proposes a simple yet effective rail surface defect detection model, FS-RSDD, for rail surface condition monitoring, which also aims to address the issue of insufficient defect samples faced by previous detection models. The model utilizes a pre-trained model to extract deep features of both normal rail samples and defect samples. Subsequently, an unsupervised learning method is employed to learn feature distributions and obtain a feature prototype memory bank. Using prototype learning techniques, FS-RSDD estimates the probability of a test sample belonging to a defect at each pixel based on the prototype memory bank. This approach overcomes the limitations of deep learning algorithms based on supervised learning techniques, which often suffer from insufficient training samples and low credibility in validation. FS-RSDD achieves high accuracy in defect detection and localization with only a small number of defect samples used for training. Surpassing benchmarked few-shot industrial defect detection algorithms, FS-RSDD achieves an ROC of 95.2% and 99.1% on RSDDS Type-I and Type-II rail defect data, respectively, and is on par with state-of-the-art unsupervised anomaly detection algorithms.

The report presents an algorithm for constructing a convolutional hierarchical neural network classifier, which is a modification of the algorithm for constructing hierarchical neural network classifiers suggested before. The original algorithm was designed to exploit intrinsic class hierarchy to build a class tree with a neural network in each node classifying groups of initial classes (in a non-terminal node) or a subset of original classes (in a terminal node). The convolutional modification utilizes convolutional neural networks instead of regular fully connected networks in order to apply the model to image classification tasks. Use of class hierarchy for image classification should reduce the number of adjusted neural network parameters compared to deep convolutional neural networks, and therefore it should reduce training and inference time. In this context the algorithm may be compared with some pruning techniques. The convolutional hierarchical neural network classifier inherits some hyperparameters of a conventional hierarchical neural network classifier, like the activation threshold and the threshold by the share of voting patterns. The goal of this study was to explore different strategies of choosing these hyperparameters. To test these strategies, we used the CIFAR-10 dataset. Also, for demonstration purposes we apply the convolutional hierarchical neural network classifier to the CIFAR-100 dataset.