Journal article
Health prediction of metro platform screen door system based on XGBoost-BiGRU-attention model
Scientific reports
26 Jun 2026
PMID: 42362812
Abstract
To address the limitations of traditional metro equipment maintenance, such as heavy reliance on manual experience, poor mobility, difficulty assessing equipment health, and the inability to meet actual operation and maintenance needs, this paper proposes a health prediction method for metro platform screen door (PSD) system based on an XGBoost-BiGRU-Attention model optimized with the Lookahead optimizer. Using the PSD system's fault records and control principles, data were collected for 14 characteristic parameters. The XGBoost algorithm was applied to rank these variables in order of importance, and the top-ranking characteristic parameters were selected as inputs to the prediction model. An attention mechanism was then integrated into a Bidirectional Gated Recurrent Unit (BiGRU) network using the Lookahead optimizer to improve prediction performance. Results show that, compared with a standard BiGRU model, the proposed XGBoost-BiGRU-Attention model reduces Root Mean Square Error (RMSE) by 52.4%, Mean Absolute Error (MAE) by 57.1%, and Mean Absolute Percentage Error (MAPE) by 57.0%, achieving a coefficient of determination of 0.972. These findings indicate that the proposed model offers strong practicality and feasibility for health management of metro equipment.
Metrics
1 Record Views
Details
- Title
- Health prediction of metro platform screen door system based on XGBoost-BiGRU-attention model
- Creators
- Zijun Zhu - Beijing Union UniversityShaohu Tang (Corresponding Author) - Beijing Union UniversityLiang Zhang - Drexel UniversityHailin Kang - Beijing Union UniversityHongKang Song - Beijing Union UniversityPengyu Li - Beijing Union University
- Publication Details
- Scientific reports
- Publisher
- Nature Publishing
- Grant note
- 2021YFB1715700 / National Key Research and Development Program of China ZT-231141703 / Theme-based Case Project of the Ministry of Education of China
- Resource Type
- Journal article
- Language
- English
- Academic Unit
- Engineering Management; School of Engineering
- Other Identifier
- 991022200097404721