A Multi-Scale Traffic-State Gated Attention Network for Bus Arrival Time Prediction
Research Article  ·  Published: 29 September 2026
Issue cover
ICCK Transactions on Intelligent Systematics
Volume 3, Issue 3, 2026: 196-208
Research Article Free to Read

A Multi-Scale Traffic-State Gated Attention Network for Bus Arrival Time Prediction

1 School of Electronic Science and Technology, Huai'an University, Huai'an 223003, China
* Corresponding Author: Xiaoqi Yin, [email protected]
Volume 3, Issue 3
You have access to this article · Limited-Time Free Access

Article Information

Abstract

Accurate bus arrival time prediction supports passenger information services and transit dispatching, but irregular sampling and time-varying traffic conditions make the task strongly nonlinear. This paper proposes a Spatial Attention Multi-Scale Traffic-State Gated Long Short-Term Memory (SA-MS-TG-LSTM) network. Seven trajectory features are constructed from Global Positioning System (GPS) records, including acceleration, congestion index, and speed volatility. Scale-preserving Spatial Feature Attention (SFA) first reweights the feature dimensions, and three Traffic-State Gated Long Short-Term Memory (TG-LSTM) branches subsequently encode the latest 5, 10, and 15 time steps. Their outputs are concatenated and mapped to the remaining travel time. Experiments are conducted on 8,834 GPS records from 293 trips on a single bus route (Route 86). Under a controlled setting in which all recurrent models use a hidden size of 16, one recurrent layer, and at most 50 training epochs, SA-MS-TG-LSTM achieves the best aggregate performance, with a root mean square error (RMSE) of 1.056 min, mean absolute error (MAE) of 0.762 min, mean absolute percentage error (MAPE) of 12.06%, and coefficient of determination ($R^2$) of 0.9631. Compared with the Convolutional Neural Network--Gated Recurrent Unit (CNN-GRU), it reduces RMSE, MAE, and MAPE by 6.30%, 5.57%, and 11.26%, respectively. Trip-level analysis further confirms statistically significant MAE improvements over a standard Long Short-Term Memory (LSTM) model and TG-LSTM. These results demonstrate the effectiveness of the proposed method on the evaluated route, while broader multi-route validation remains an important direction for future work.

Graphical Abstract

A Multi-Scale Traffic-State Gated Attention Network for Bus Arrival Time Prediction

Keywords

bus arrival time prediction intelligent transportation LSTM traffic-state gating multi-scale temporal modeling feature attention

Data Availability Statement

The bus GPS data used in this study are not publicly available because they were obtained from an operational transit route. Processed data and implementation details are available from the corresponding author upon reasonable request.

Funding

This work was supported without any funding.

Conflicts of Interest

The authors declare no conflicts of interest.

AI Use Statement

The authors declare that no generative AI was used in the preparation of this manuscript.

Ethical Approval and Consent to Participate

Not applicable.

References

  1. Cathey, F. W., & Dailey, D. J. (2003). A prescription for transit arrival/departure prediction using automatic vehicle location data. Transportation Research Part C: Emerging Technologies, 11(3-4), 241-264.
    [CrossRef] [Google Scholar]
  2. Patnaik, J., Chien, S., & Bladikas, A. (2004). Estimation of bus arrival times using APC data. Journal of public transportation, 7(1), 1-20.
    [CrossRef] [Google Scholar]
  3. Shalaby, A., & Farhan, A. (2004). Prediction model of bus arrival and departure times using AVL and APC data. Journal of Public Transportation, 7(1), 41-61.
    [CrossRef] [Google Scholar]
  4. Mete, S., Çelik, E., & Gül, M. (2022). Predicting the time of bus arrival for public transportation by time series models. Journal of Transportation and Logistics, 7(2), 541-555.
    [CrossRef] [Google Scholar]
  5. Chien, S. I. J., Ding, Y., & Wei, C. (2002). Dynamic bus arrival time prediction with artificial neural networks. Journal of transportation engineering, 128(5), 429-438.
    [CrossRef] [Google Scholar]
  6. Jeong, R., & Rilett, L. R. (2005). Prediction model of bus arrival time for real-time applications. Transportation Research Record, 1927(1), 195-204.
    [CrossRef] [Google Scholar]
  7. Yu, B., Lam, W. H. K., & Tam, M. L. (2011). Bus arrival time prediction at bus stop with multiple routes. Transportation Research Part C: Emerging Technologies, 19(6), 1157-1170.
    [CrossRef] [Google Scholar]
  8. Bengio, Y., Simard, P., & Frasconi, P. (1994). Learning long-term dependencies with gradient descent is difficult. IEEE transactions on neural networks, 5(2), 157-166.
    [CrossRef] [Google Scholar]
  9. Hochreiter, S., & Schmidhuber, J. (1997). Long short-term memory. Neural computation, 9(8), 1735-1780.
    [CrossRef] [Google Scholar]
  10. Kumar, B. A., Singh, R., Shaji, H. E., & Vanajakshi, L. (2025). Bus arrival time prediction: A comprehensive review. IEEE Transactions on Intelligent Transportation Systems, 26(6), 7362-7379.
    [CrossRef] [Google Scholar]
  11. Jalaney, J., & Ganesh, R. S. (2023). Multiple extreme learning machines based arrival time prediction for public bus transport. Intelligent Automation & Soft Computing, 36(3), 2819-2834.
    [CrossRef] [Google Scholar]
  12. Cho, K., Van Merriënboer, B., Gulçehre, Ç., Bahdanau, D., Bougares, F., Schwenk, H., & Bengio, Y. (2014, October). Learning phrase representations using RNN encoder–decoder for statistical machine translation. In Proceedings of the 2014 conference on empirical methods in natural language processing (EMNLP) (pp. 1724-1734).
    [CrossRef] [Google Scholar]
  13. Petersen, N. C., Rodrigues, F., & Pereira, F. C. (2019). Multi-output bus travel time prediction with convolutional LSTM neural network. Expert Systems with Applications, 120, 426-435.
    [CrossRef] [Google Scholar]
  14. Bahdanau, D., Cho, K., & Bengio, Y. (2014). Neural machine translation by jointly learning to align and translate. arXiv preprint arXiv:1409.0473.
    [CrossRef] [Google Scholar]
  15. Zhang, B., Tang, L., Zhou, D., Liu, K., & Xue, Y. (2024). Bus Arrival Time Prediction Based on the Optimized Long Short‐Term Memory Neural Network Model With the Improved Whale Algorithm. Journal of Advanced Transportation, 2024(1), 6997338.
    [CrossRef] [Google Scholar]
  16. Li, Z. (2024). DA-RNN-based bus arrival time prediction model. International Journal of Intelligent Transportation Systems Research, 22(3), 660-674.
    [CrossRef] [Google Scholar]
  17. Yu, C., Chong, Y. W., Budi, A. S., Setiawan, E., & Keoh, S. L. (2024, August). Bus Arrival Time Prediction Using Hybrid LightGBM-LSTM. In 2024 International Conference on Platform Technology and Service (PlatCon) (pp. 44-49). IEEE.
    [CrossRef] [Google Scholar]
  18. Lee, C., & Yoon, Y. (2022). A novel bus arrival time prediction method based on spatio-temporal flow centrality analysis and deep learning. Electronics, 11(12), 1875.
    [CrossRef] [Google Scholar]
  19. Sun, H., Li, F., & Liu, L. (2024, August). A spatial-temporal neural network model based on meta learning for bus arrival time prediction. In 2024 10th International Conference on Big Data Computing and Communications (BigCom) (pp. 57-64). IEEE.
    [CrossRef] [Google Scholar]
  20. Qiu, T., Lam, C. T., Liu, B., Ng, B. K., Yuan, X., & Im, S. K. (2025). FEN-MRMGCN: A frontend-enhanced network based on multi-relational modeling GCN for bus arrival time prediction. IEEE Access, 13, 5296-5307.
    [CrossRef] [Google Scholar]
  21. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, L., & Polosukhin, I. (2017). Attention is all you need. Advances in Neural Information Processing Systems, 30.
    [Google Scholar]
  22. Jeong, S., Oh, C., & Jeong, J. (2024). BAT-transformer: Prediction of bus arrival time with transformer encoder for smart public transportation system. Applied Sciences, 14(20), 9488.
    [CrossRef] [Google Scholar]
  23. Srivastava, N., Hinton, G., Krizhevsky, A., Sutskever, I., & Salakhutdinov, R. (2014). Dropout: A simple way to prevent neural networks from overfitting. Journal of Machine Learning Research, 15(56), 1929-1958. https://jmlr.org/papers/v15/srivastava14a.html
    [Google Scholar]
  24. Kingma, D. P., & Ba, J. (2014). Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980.
    [CrossRef] [Google Scholar]
  25. Loshchilov, I., & Hutter, F. (2017). Decoupled weight decay regularization. arXiv preprint arXiv:1711.05101.
    [CrossRef] [Google Scholar]

Cite This Article

APA Style
Liu, R., Yin, X., Zhou, J., Ma, C., & Li, C. (2026). A Multi-Scale Traffic-State Gated Attention Network for Bus Arrival Time Prediction. ICCK Transactions on Intelligent Systematics, 3(3), 196-208. https://doi.org/10.62762/TIS.2026.906270
Export Citation
RIS Format
Compatible with EndNote, Zotero, Mendeley, and other reference managers
TY  - JOUR
AU  - Liu, Ruping
AU  - Yin, Xiaoqi
AU  - Zhou, Jing
AU  - Ma, Chenxi
AU  - Li, Chen
PY  - 2026
DA  - 2026/09/29
TI  - A Multi-Scale Traffic-State Gated Attention Network for Bus Arrival Time Prediction
JO  - ICCK Transactions on Intelligent Systematics
T2  - ICCK Transactions on Intelligent Systematics
JF  - ICCK Transactions on Intelligent Systematics
VL  - 3
IS  - 3
SP  - 196
EP  - 208
DO  - 10.62762/TIS.2026.906270
UR  - https://www.icck.org/article/abs/TIS.2026.906270
KW  - bus arrival time prediction
KW  - intelligent transportation
KW  - LSTM
KW  - traffic-state gating
KW  - multi-scale temporal modeling
KW  - feature attention
AB  - Accurate bus arrival time prediction supports passenger information services and transit dispatching, but irregular sampling and time-varying traffic conditions make the task strongly nonlinear. This paper proposes a Spatial Attention Multi-Scale Traffic-State Gated Long Short-Term Memory (SA-MS-TG-LSTM) network. Seven trajectory features are constructed from Global Positioning System (GPS) records, including acceleration, congestion index, and speed volatility. Scale-preserving Spatial Feature Attention (SFA) first reweights the feature dimensions, and three Traffic-State Gated Long Short-Term Memory (TG-LSTM) branches subsequently encode the latest 5, 10, and 15 time steps. Their outputs are concatenated and mapped to the remaining travel time. Experiments are conducted on 8,834 GPS records from 293 trips on a single bus route (Route 86). Under a controlled setting in which all recurrent models use a hidden size of 16, one recurrent layer, and at most 50 training epochs, SA-MS-TG-LSTM achieves the best aggregate performance, with a root mean square error (RMSE) of 1.056 min, mean absolute error (MAE) of 0.762 min, mean absolute percentage error (MAPE) of 12.06%, and coefficient of determination ($R^2$) of 0.9631. Compared with the Convolutional Neural Network--Gated Recurrent Unit (CNN-GRU), it reduces RMSE, MAE, and MAPE by 6.30%, 5.57%, and 11.26%, respectively. Trip-level analysis further confirms statistically significant MAE improvements over a standard Long Short-Term Memory (LSTM) model and TG-LSTM. These results demonstrate the effectiveness of the proposed method on the evaluated route, while broader multi-route validation remains an important direction for future work.
SN  - 3068-5079
PB  - Institute of Central Computation and Knowledge
LA  - English
ER  - 
BibTeX Format
Compatible with LaTeX, BibTeX, and other reference managers
@article{Liu2026A,
  author = {Ruping Liu and Xiaoqi Yin and Jing Zhou and Chenxi Ma and Chen Li},
  title = {A Multi-Scale Traffic-State Gated Attention Network for Bus Arrival Time Prediction},
  journal = {ICCK Transactions on Intelligent Systematics},
  year = {2026},
  volume = {3},
  number = {3},
  pages = {196-208},
  doi = {10.62762/TIS.2026.906270},
  url = {https://www.icck.org/article/abs/TIS.2026.906270},
  abstract = {Accurate bus arrival time prediction supports passenger information services and transit dispatching, but irregular sampling and time-varying traffic conditions make the task strongly nonlinear. This paper proposes a Spatial Attention Multi-Scale Traffic-State Gated Long Short-Term Memory (SA-MS-TG-LSTM) network. Seven trajectory features are constructed from Global Positioning System (GPS) records, including acceleration, congestion index, and speed volatility. Scale-preserving Spatial Feature Attention (SFA) first reweights the feature dimensions, and three Traffic-State Gated Long Short-Term Memory (TG-LSTM) branches subsequently encode the latest 5, 10, and 15 time steps. Their outputs are concatenated and mapped to the remaining travel time. Experiments are conducted on 8,834 GPS records from 293 trips on a single bus route (Route 86). Under a controlled setting in which all recurrent models use a hidden size of 16, one recurrent layer, and at most 50 training epochs, SA-MS-TG-LSTM achieves the best aggregate performance, with a root mean square error (RMSE) of 1.056 min, mean absolute error (MAE) of 0.762 min, mean absolute percentage error (MAPE) of 12.06\%, and coefficient of determination (\$R^2\$) of 0.9631. Compared with the Convolutional Neural Network--Gated Recurrent Unit (CNN-GRU), it reduces RMSE, MAE, and MAPE by 6.30\%, 5.57\%, and 11.26\%, respectively. Trip-level analysis further confirms statistically significant MAE improvements over a standard Long Short-Term Memory (LSTM) model and TG-LSTM. These results demonstrate the effectiveness of the proposed method on the evaluated route, while broader multi-route validation remains an important direction for future work.},
  keywords = {bus arrival time prediction, intelligent transportation, LSTM, traffic-state gating, multi-scale temporal modeling, feature attention},
  issn = {3068-5079},
  publisher = {Institute of Central Computation and Knowledge}
}

Article Metrics

Citations
Crossref
0
Scopus
0
Views
19
PDF Downloads
5

Publisher's Note

ICCK stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and Permissions

Institute of Central Computation and Knowledge (ICCK) or its licensor holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
ICCK Transactions on Intelligent Systematics
ICCK Transactions on Intelligent Systematics
ISSN: 3068-5079 (Online) | ISSN: 3069-003X (Print)
Portico
Preserved at
Portico