A Lightweight Rail Tread Extraction Framework for Ballastless Track LiDAR Point Clouds Using Multi-Stage Filtering and Curvature-Guided Region Growing
Abstract
1. Introduction
- A curvature-ranked seed selection strategy is proposed to replace conventional random initialization in region growing, thereby improving clustering stability and suppressing boundary leakage near rail-side transition areas.
- A dual-constrained region growing mechanism integrating normal vector consistency and curvature thresholds is developed to improve segmentation homogeneity while reducing excessive growth caused by geometrically similar structures.
- A lightweight and training-free extraction framework is established for ballastless metro environments. The proposed method does not depend on RGB imagery, trajectory priors, or template matching, making it suitable for practical engineering deployment in complex rail transit scenarios.
2. Related Work
2.1. Methods Based on Point Attributes and Image Processing
2.2. Methods Based on Deep Learning
2.3. Traditional Geometry-Driven Rail Extraction Methods
3. Case Study
4. Methodology
4.1. Intensity Threshold Filtering



4.2. Cloth Simulation Filtering
4.3. Calculation of Normal Vector and Curvature of Point Cloud
- Establish a KD-tree to build a topological relationship for disordered and scattered point clouds. The KD-tree can divide the data in the k-dimensional space and then retrieve it in the divided interval.
- Based on the KD-tree local neighborhood search, a point set P is constructed with each data point as the center. The point set P is shown in Equation (1). In the formula, n represents the total number of points, and are the three-dimensional coordinates of the point .Based on each point , a point set M is constructed with a neighborhood radius of r, where the distance between any point and is less than r. The point set M is defined in Equation (2), where r denotes the neighborhood radius.where n represents the total number of points in the dataset, and is the three-dimensional coordinate vector of the i-th point.
- According to the principle of least squares of distance, the Locally fitted plane K of the data set M is calculated. The plane K is shown in the function in Equation (3). In the formula, e is the normal vector of the plane K, d is the distance from the sampling point to the plane K, and is the variable value of the minimum value of the function.In the process of fitting the local plane, the center of mass of k near-neighbor points is located on the plane K, and e is the unit vector, satisfying . Therefore, the solution point cloud normal vector problem is converted into the covariance matrix eigenvalue decomposition problem of k near neighbors. The eigenvector corresponding to the minimum eigenvalue of the obtained covariance matrix is the normal vector of the sampled point. Where the covariance matrix is shown in Formulas (4) and (5), k in the formula is the total number of neighbor points. Number, is the center of mass of k nearest neighbors, is the eigenvalue of the covariance matrix, and is the eigenvector corresponding to .At each sampling point in the point cloud, there is a curved surface approaching the neighborhood point cloud, and the curvature at a point can be represented by the local surface curvature of the point and its neighborhood point. At present, the more common method is to find the curvature of a certain point by fitting a quadratic surface, but since the least square method has been used above to find the normal vector of the sampling point, in which the eigenvalue represents the degree of change of the sampling point in three directions, so the curvature of the point is close. The likelihood estimation method is shown in Equation (6):
4.4. Rail Tread Extraction Based on Regional Growth Algorithm
- The KD-tree paired with Principal Component Analysis (PCA) is utilized to estimate the local surface normal vector for each point, from which the corresponding curvature is subsequently derived. As illustrated in Figure 8, the central region of the rail tread exhibits low curvature values, whereas the transition zone toward the rail edge is characterized by high curvature. Consequently, all points are sorted in ascending order according to their curvature magnitudes. The unassigned point with the minimum curvature is selected as the initial seed point, ensuring each point is evaluated as a seed point candidate only once.
- The k-nearest neighbors (KNN) algorithm is employed to retrieve the neighborhood of the current seed point. The orientation deviation is evaluated by calculating the angle between the normal vector of each neighboring point and that of the seed point. If (where denotes the smoothing angle threshold), the neighboring point is deemed geometrically compliant with the seed point and is assigned to the same region.
- As demonstrated in Figure 8, the central surface of the rail tread is relatively planar with a minimal curvature profile, whereas the geometric transition at the rail edge consistently reaches and above. To prevent the growing boundary from crossing the sharp edge into the rail side—which would trigger severe over-segmentation and leakage—a stringent curvature threshold is required. Through an empirical sensitivity analysis, the curvature constraint was established at for seed point propagation. A higher threshold (e.g., >0.02) risks introducing high-frequency geometric noise from the concrete base into the seed queue, whereas a lower threshold restricts growth prematurely, leaving the rail tread fragmented. Therefore, only points satisfying within the identical region are appended to the seed queue as new seeds.
- If the seed point queue is not empty, a new seed point is sequentially retrieved from it to repeat steps (2) through (4).
- Once the current seed queue is exhausted, the localized cluster sharing homogenous geometric attributes is finalized and assigned a distinct random color. Steps (1) to (5) are iteratively executed until all points in the workspace have been given regional attributes, concluding the region-growing process.
- Owing to structural complexity or spatial discontinuity in the subway scene, non-target background objects typically form heavily fragmented clusters with limited point counts. Consequently, a minimum cluster size threshold M is implemented to efficiently filter out these residual outliers containing fewer than M points, thereby successfully isolating the continuous, intact rail tracks.

5. Results
5.1. Parameter Analysis

5.1.1. Selection of Intensity Threshold
5.1.2. Cloth Simulation Algorithm Filter
5.1.3. Regional Growth Algorithm
5.2. Quantitative Evaluation
5.3. Comparative Experiments
5.3.1. Comparison Methods and Baseline Setup
- Euclidean distance-based clustering (Method A): Segmentation using only spatial distance thresholds, representing pure geometric position-based segmentation.
- Traditional region growing algorithm (Method B): Region growing with random seed points based on normal vectors and curvature (the standard PCL implementation), representing conventional unimproved geometric segmentation.
5.3.2. Quantitative Evaluation Metrics
5.3.3. Quantitative Result Analysis Table
5.3.4. Qualitative Results Visualization Comparison
| Method | Precision (%) | Recall (%) | F1-Score (%) | Point Count |
|---|---|---|---|---|
| Euclidean Clustering | 78.5 | 45.2 | 57.4 | 32,410 |
| Standard Region Growing | 72.3 | 95.1 | 82.1 | 74,280 |
| Ours () | 95.3 | 92.2 | 93.7 | 65,821 |

6. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Ahrén, J.; Olofsson, T. Future urban rail transit systems: Trends and challenges for smart and sustainable cities. Sustainability 2023, 15, 1882. [Google Scholar]
- Ahn, K.; Jang, H.; Song, Y. Economic impacts of being close to subway networks: A case study of Korean metropolitan areas. Res. Transp. Econ. 2020, 83, 100900. [Google Scholar] [CrossRef] [Scilit]
- Song, J.; Abuduwayiti, A.; Gou, Z. The role of subway network in urban spatial structure optimization—Wuhan city as an example. Tunn. Undergr. Space Technol. 2023, 131, 104842. [Google Scholar] [CrossRef] [Scilit]
- Arastounia, M.; Oude Elberink, S. Application of template matching for improving classification of urban railroad point clouds. Sensors 2016, 16, 2112. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Qi, Z.; Tian, Y.; Shi, Y. Efficient railway tracks detection and turnouts recognition method using HOG features. Neural Comput. Appl. 2013, 23, 245–254. [Google Scholar]
- Lou, Y.; Zhang, T.; Tang, J.; Song, W.; Zhang, Y.; Chen, L. A fast algorithm for rail extraction using mobile laser scanning data. Remote Sens. 2018, 10, 1998. [Google Scholar] [CrossRef] [Scilit]
- Kumar, S.S.J.; Titus, T.J.; Ganesh, V.; Devi, V.S. Automotive crack detection for railway track using ultrasonic sensors. Int. J. Eng. Technol. Comput. Res. 2016, 4, 34–37. [Google Scholar]
- Soilán, M.; Sánchez-Rodríguez, A.; del Río-Barral, P.; Perez-Collazo, C.; Arias, P.; Riveiro, B. Review of laser scanning technologies and their applications for road and railway infrastructure monitoring. Infrastructures 2019, 4, 58. [Google Scholar] [CrossRef] [Scilit]
- Che, E.; Jung, J.; Olsen, M.J. Object recognition, segmentation, and classification of mobile laser scanning point clouds: A state of the art review. Sensors 2019, 19, 810. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Trilla, A.; Bob-Manuel, J.; Lamoureux, B.; Vilasis-Cardona, X. Integrated Multiple-Defect Detection and Evaluation of Rail Wheel Tread Images using Convolutional Neural Networks. Int. J. Prognit. Health Manag. 2021, 12, 1–10. [Google Scholar] [CrossRef] [Scilit]
- Beger, R.; Gedrange, C.; Hecht, R.; Neubert, M. Data fusion of extremely high resolution aerial imagery and LiDAR data for automated railroad centre line reconstruction. ISPRS J. Photogramm. Remote Sens. 2011, 66, S40–S51. [Google Scholar] [CrossRef] [Scilit]
- Zhu, L.; Hyyppä, J. The use of airborne and mobile laser scanning for modeling railway environments in 3D. Remote Sens. 2014, 6, 3075–3100. [Google Scholar] [CrossRef] [Scilit]
- Zhong, J.; Liu, Z.; Han, Z.; Han, Y.; Zhang, W. A CNN-based defect inspection method for catenary split pins in high-speed railway. IEEE Trans. Instrum. Meas. 2018, 68, 2849–2860. [Google Scholar]
- Li, J.; Chen, H.; Liu, Z. Automated Rail Extraction Using ntensity and Raster Image Processing. IEEE J. Sel. Top. Appl. Earth Obs. Remote Sens. 2024, 17, 8921–8935. [Google Scholar]
- Xu, L.; Zheng, S.; Na, J.; Yang, Y.; Mu, C.; Shi, D. A Vehicle-Borne Mobile Mapping System Based Framework for Semantic Segmentation and Modeling on Overhead Catenary System Using Deep Learning. Remote Sens. 2021, 13, 4939. [Google Scholar] [CrossRef] [Scilit]
- Grandio, J.; Riveiro, B.; Soilán, M.; Arias, P. Point cloud semantic segmentation of complex railway environments using deep learning. Autom. Constr. 2022, 141, 104425. [Google Scholar] [CrossRef] [Scilit]
- Grandio, J.; Riveiro, B.; Lamas, D.; Arias, P. Multimodal deep learning for point cloud panoptic segmentation of railway environments. Autom. Constr. 2023, 150, 104854. [Google Scholar] [CrossRef] [Scilit]
- Zhang, R.; Ding, Q.; Ng, A.H.-M.; Wang, D.; Deng, J.; Xu, M.; Hou, Y. TripletA-Net: A deep learning model for automatic railway track extraction from airborne LiDAR point clouds. IEEE J. Sel. Top. Appl. Earth Obs. Remote Sens. 2025, 18, 9195–9210. [Google Scholar] [CrossRef] [Scilit]
- Sun, N.; Li, K.; Liu, J.; Chai, L.; Wu, C. Learning and fusing the rasterized and serialized point cloud for 3D semantic segmentation in railway scenes. Expert Syst. Appl. 2025, 295, 128897. [Google Scholar] [CrossRef] [Scilit]
- Yang, B.; Fang, L. Automated Extraction of 3-D Railway Tracks from Mobile Laser Scanning Point Clouds. IEEE J. Sel. Top. Appl. Earth Obs. Remote Sens. 2014, 7, 4750–4761. [Google Scholar] [CrossRef] [Scilit]
- Arastounia, M. Automated recognition of railroad infrastructure in rural areas from LiDAR data. Remote Sens. 2015, 7, 14916–14938. [Google Scholar] [CrossRef] [Scilit]
- Ren, Y.; Lu, P.; Ai, C. Hybrid Local-Feature-Based Approach for Automated Rail Extraction from Lidar Data. Transp. Res. Rec. 2025, 2679, 53–71. [Google Scholar] [CrossRef] [Scilit]
- Wang, H.; Li, M.; Zhang, Q. Rail Point Cloud Extraction via Elevation Slice and Image Morphology. Remote Sens. 2023, 15, 3012. [Google Scholar]
- Zhao, L.; Chen, G.; Wang, H. Grid-Based Classification of Rails Using Height and Intensity Features. Remote Sens. 2021, 13, 3876. [Google Scholar]
- Chen, Y.; Zhang, X.; Li, C. Intensity-Guided RANSAC and Image Projection for Rail Extraction. ISPRS Ann. Photogramm. Remote Sens. Spat. Inf. Sci. 2022, 5, 345–352. [Google Scholar]
- Zou, R.; Fan, X.; Qian, C.; Ye, W.; Zhao, P.; Tang, J.; Liu, H. An efficient and accurate method for different configurations railway extraction based on mobile laser scanning. Remote Sens. 2019, 11, 2929. [Google Scholar] [CrossRef] [Scilit]
- Amodio, P.; De Giosa, M.; Iavernaro, F.; La Scala, R.; Labianca, A.; Lazzo, M.; Mazzia, F.; Pisani, L. Detection of anomalies in the proximity of a railway line: A case study. J. Comput. Math. Data Sci. 2022, 4, 100052. [Google Scholar] [CrossRef] [Scilit]
- Stein, D.; Spindler, M.; Lauer, M. Model-based rail detection in mobile laser scanning data. In Proceedings of the 2016 IEEE Intelligent Vehicles Symposium, Gothenberg, Sweden, 19–22 June 2016; IEEE: New York, NY, USA, 2016; pp. 654–661. [Google Scholar]
- Cheng, Y.-J.; Qiu, W.-G.; Duan, D.-Y. Automatic creation of as-is building information model from single-track railway tunnel point clouds. Autom. Constr. 2019, 106, 102911. [Google Scholar] [CrossRef] [Scilit]
- Lamas, D.; Soilán, M.; Grandío, J.; Riveiro, B. Automatic point cloud semantic segmentation of complex railway environments. Remote Sens. 2021, 13, 2332. [Google Scholar] [CrossRef] [Scilit]
- Pastucha, E. Catenary system detection, localization and classification using mobile scanning data. Remote Sens. 2016, 8, 801. [Google Scholar] [CrossRef] [Scilit]
- Song, L.; Liu, H.; Xu, L.; Hou, J.; Yu, Z. Dynamic performance of CRTS III ballastless track structure under train load and temperature. In Structures; Elsevier: Amsterdam, The Netherlands, 2023; Volume 53, pp. 408–420. [Google Scholar]
- Xu, S.; Wang, H.; Tian, X.; Wang, T.; Tanikawa, H. From efficiency to equity: Changing patterns of China’s regional transportation systems from an in-use steel stocks perspective. J. Ind. Ecol. 2022, 26, 548–561. [Google Scholar]
- Kaasalainen, S.; Jaakkola, A.; Kaasalainen, M.; Krooks, A.; Kukko, A. Analysis of incidence angle and distance effects on terrestrial laser scanner intensity: Search for correction methods. Remote Sens. 2011, 3, 2207–2221. [Google Scholar] [CrossRef] [Scilit]
- Soudarissanane, S.; Lindenbergh, R.; Menenti, M.; Teunissen, P. Scanning geometry: Influencing factor on the quality of terrestrial laser scanning points. ISPRS J. Photogramm. Remote Sens. 2011, 66, 389–399. [Google Scholar] [CrossRef] [Scilit]
- Song, J.H.; Han, S.H.; Yu, K.Y.; Kim, Y.I. Assessing the possibility of land-cover classification using LiDAR intensity data. Int. Arch. Photogramm. Remote Sens. Spat. Inf. Sci. 2002, 34, 259–262. [Google Scholar]
- Zhang, W.; Qi, J.; Wan, P.; Wang, H.; Xie, D.; Wang, X.; Yan, G. An easy-to-use airborne LiDAR data filtering method based on cloth simulation. Remote Sens. 2016, 8, 501. [Google Scholar] [CrossRef] [Scilit]
- Cai, S.; Zhang, W.; Qi, J.; Wan, P.; Shao, J.; Shen, A. Applicability analysis of cloth simulation filtering algorithm for mobile LiDAR point cloud. Int. Arch. Photogramm. Remote Sens. Spat. Inf. Sci. 2018, 42, 107–111. [Google Scholar] [CrossRef] [Scilit]
- Hoppe, H.; DeRose, T.; Duchamp, T.; McDonald, J.; Stuetzle, W. Surface reconstruction from unorganized points. In Proceedings of the 19th Annual Conference on Computer Graphics and Interactive Techniques, Chicago, IL, USA, 26–31 July 1992; Association for Computing Machinery (ACM): New York, NY, USA, 1992; pp. 71–78. [Google Scholar]
- Zhang, J.; Cao, J.; Liu, X.; Wang, J.; Liu, J.; Shi, X. Point cloud normal estimation via low-rank subspace clustering. Comput. Graph. 2013, 37, 697–706. [Google Scholar] [CrossRef] [Scilit]











| Region | Object | Intensity |
|---|---|---|
| Track subgrade | Concrete subgrade | 7–14 |
| Rail tread | 1–4 | |
| Rail side | 4–7 | |
| Rail fastener | 1–10 | |
| Noise barriers | Metal noise barriers | 1–9 |
| Sheet iron | 10–60 | |
| Transparent plastic board | 0 | |
| Railway catenary system | Transmission line | 1–7 |
| Overhead power cable | 0–6 | |
| Mast | 10–40 | |
| Cantilevers | 0–40 | |
| Sign | Mileage sign | 254–255 |
| Sensitivity (%) | Precision (%) | |
|---|---|---|
| 3 | 86.89 | 98.29 |
| 4 | 91.53 | 96.49 |
| 5 | 92.23 | 95.32 |
| 6 | 92.81 | 82.76 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
He, G.; Zhang, R.; Zhong, Y. A Lightweight Rail Tread Extraction Framework for Ballastless Track LiDAR Point Clouds Using Multi-Stage Filtering and Curvature-Guided Region Growing. Appl. Sci. 2026, 16, 8791. https://doi.org/10.3390/app16178791
He G, Zhang R, Zhong Y. A Lightweight Rail Tread Extraction Framework for Ballastless Track LiDAR Point Clouds Using Multi-Stage Filtering and Curvature-Guided Region Growing. Applied Sciences. 2026; 16(17):8791. https://doi.org/10.3390/app16178791
Chicago/Turabian StyleHe, Guizhen, Rui Zhang, and Yuxin Zhong. 2026. "A Lightweight Rail Tread Extraction Framework for Ballastless Track LiDAR Point Clouds Using Multi-Stage Filtering and Curvature-Guided Region Growing" Applied Sciences 16, no. 17: 8791. https://doi.org/10.3390/app16178791
APA StyleHe, G., Zhang, R., & Zhong, Y. (2026). A Lightweight Rail Tread Extraction Framework for Ballastless Track LiDAR Point Clouds Using Multi-Stage Filtering and Curvature-Guided Region Growing. Applied Sciences, 16(17), 8791. https://doi.org/10.3390/app16178791
